Ensure x is a probability vector, that is all 0<=x_j<=1 and sum(x)=1.使用MATLAB编程

以下是MATLAB代码实现：

function x = ensure_probability_vector(x)
% x: 输入的向量
% 确保x是一个概率向量，即所有元素在[0,1]之间且总和为1

if any(x < 0) || any(x > 1)
    error('输入向量的元素必须在[0,1]之间');
end

if abs(sum(x) - 1) > 1e-6
    x = x / sum(x);
end

使用方法：

x = [0.2, 0.3, 0.5];
x = ensure_probability_vector(x);

输出结果：

x =

    0.2000    0.3000    0.5000