R生成固定长度的固定和的随机二进制向量

import random

def generate_random_binary_vector(n, k):
    vector = [0] * n
    current_sum = 0

    for i in range(n):
        while True:
            random_bit = random.randint(0, 1)
            if current_sum + random_bit <= k:
                vector[i] = random_bit
                current_sum += random_bit
                break

    return vector

# 示例用法
n = 10  # 向量长度
k = 5   # 目标和
result = generate_random_binary_vector(n, k)
print(result)