%======歡迎關注公眾號：電力系統(tǒng)與算法之美=======%
function spreadingBySir()
    A=load('test.in');
    % node number
    N=size(A,1);   
    %感染概率
    irate=0.3;
    %恢復概率
    rrate=1;
    %初始時節(jié)點的狀態(tài)表,初始時只有節(jié)點1為感染狀態(tài)，其他的都為易感染狀態(tài)  
    start_node=1;
    %按照圖的廣度優(yōu)先的原則來進行病毒傳播
    BFSspreading(A,N,start_node,irate,rrate);
end
 
function BFSspreading(A,N,start_node,irate,rrate)
%隊列頭
head=1;            
%隊列尾，開始隊列為空，tail==head
tail=1;            
%向頭中加入感染源節(jié)點
queue(head)=start_node;      
%隊列擴展
head=head+1;  
 
%感染節(jié)點列表 
infection=start_node;  
%恢復節(jié)點列表  
recover=[];
%易感染節(jié)點列表
for i=1:N
    %初始時，start_node為感染狀態(tài)
    if i==start_node
        %-1表示該節(jié)點已經(jīng)從列表中刪除
        susceptible(i)=-1;
    end
    %初始時，除了start_node為感染狀態(tài)外，其他節(jié)點都處于易感染狀態(tài)
     susceptible(i)=i;
end
 
%開始按照廣度優(yōu)先搜索順序向鄰居節(jié)點傳播
%判斷隊列是否為空
while tail~=head   
    %取隊尾節(jié)點 
    i=queue(tail);  
    %如果該節(jié)點不在移除列表之中
    if isempty(find(recover==i,1))
            for j=1:N
             %如果節(jié)點j與當前節(jié)點i相連并且節(jié)點j不在感染列表中
            if A(i,j)==1 && isempty(find(infection==j,1))   
                 infection_random=rand(1);
                 if infection_random < irate
                    %新節(jié)點入列
                    queue(head)=j;  
                    %擴展隊列
                    head=head+1;   
                    %將新節(jié)點j加入感染列表
                    infection=[infection j]; 
                    
                    %從易感染節(jié)點列表中刪除該節(jié)點,設置為-1
                    [row,col,v] = find(susceptible==j) ;
                    susceptible(col)=-1;
                    susceptible(find(susceptible==-1))=[];                    
                 end
            end
        end
        %將感染的節(jié)點按概率加入恢復節(jié)點列表  
        recover_random=rand(1);
        if infection_random < rrate
            %恢復
            recover=[recover i];  
            %從感染列表中刪除
            [row,col,v] = find(infection==i) ;
            infection(col)=-1;
            infection(find(infection==-1))=[];
        end
        tail=tail+1; 
        
    end %end if  isempty(find(recover==i,1)
end %end while
 
%分別現(xiàn)實最后節(jié)點的狀態(tài)
infection
susceptible
recover
end