数据结构 判断有向图中是否有有向回路

//算法思想：
//对每个顶点v都进行一次深搜，如果深搜过程中又能走回顶点v，则不在进行搜索，将回路标志flag置为1
#include<stdio.h>
int a[6][6]={0};
int flag=0;//回路标记
//a[1][2]=1;a[1][5]=1;a[2][3]=1;a[3][1]=1;
int main(){
    int i;
    a[1][2]=1;a[1][5]=1;a[2][3]=1;a[3][1]=1;//初始化图
    for(i=1;i<=5;i++)
        {
            dfs(i,i);//前一个i是路径的头，如果还能回到i说明有回路，后一个i是当前走到的节点
            if(flag==1)
            {
                    printf("有回路！\n");
                    break;
                    }
        }
    if(i>5)printf("无回路!\n");
    return 0;
}

int dfs(int v,int n )
{
        int i;
        for(i=1;i<=5;i++)
            {
                if(flag==1)break;//已经判断出有回路，不用再搜索
                if(a[n][i]==1)//i是n的邻接顶点
                {
                    if(i==v)//如果i是头顶点，说明有回路
                        flag=1;
                
                    else dfs(v,i);//否则，从i顶点继续进行深搜
                }
            }
}