Matlab——资源分配算法-注水算法

算法实现
其中: Num_subc子载波数 Pt功率 gain_subc子载波增益 noise_var噪声功率 gap信噪比间隔
B带宽 K最大分配子载波数 Rt数据速率
%---------Water filling algo--------------
function [bit_alloc,power_alloc,bit_theory]=waterfill(Num_subc,Pt,gain_subc,noise_var,gap,B,K,Rt)
snr_subc=SNR_subc(gap,gain_subc,noise_var);
%------------remove the subcarrier with zero gain-------------------
% [gain_subc,index]=sort(gain_subc,‘descend’);
% num_zero_gain=length(find(gain_subc0));
% num_use=Num_subc-num_zero_gain;
%------------------water filling allocation--------------
power_alloc=(Pt+sum(1./snr_subc))/Num_subc-1./snr_subc;
while(length(find(power_alloc<0))~=0)
neg_index=find(power_alloc<=0);
pos_index=find(power_alloc>0);
power_alloc(neg_index)=0;
num_use=length(pos_index);
snr_use=snr_subc(pos_index);
power_temp=(Pt+sum(1./snr_use))/num_use-1./snr_use