图的基本存储的基本方式四

一直不知道这个题为什么用链表可以A，但是结构体就会WA，如果有童鞋们知道，希望能留下想法，一起学习，一起进步

Problem Description

解决图论问题，首先就要思考用什么样的方式存储图。但是小鑫却怎么也弄不明白如何存图才能有利于解决问题。你能帮他解决这个问题么？

Input

 多组输入，到文件结尾。

每一组第一行有一个数n表示n个点。接下来给出一个n*n的矩阵 表示一个由邻接矩阵方式存的图。
矩阵a中的元素aij如果为0表示i不可直接到j，1表示可直接到达。
之后有一个正整数q，表示询问次数。
接下来q行每行有一个询问，输入两个数为a，b。

注意：点的编号为0~n-1，2<=n<=5000，0<=q<=100,0 <= a,b < n。
保证可达边的数量不超过1000

Output

 对于每一条询问，输出一行。若a到b可以直接连通输出Yes，否则输出No。

Sample Input

2
0 1
0 0
2
0 1
1 0

Sample Output

Yes
No

Hint

Source

LeiQ

 

 

 

 

 

#include <stdio.h>
#include <stdlib.h>>
#include <string.h>
struct node
{
    int data;
    struct node *next;
}*a[5010];
void add(int i,int j)
{
    struct node *q;
    q = (struct node *)malloc(sizeof(struct node));
    q->data = j;
    q->next = NULL;
    if(a[i] == NULL)
        a[i] = q;
    else
    {
        q->next = a[i]->next;
        a[i]->next = q;
    }
}
int main()
{
    int n,i,j,m,u,v;
    while(scanf("%d",&n) != EOF)
    {
        struct node *p;
        for(i = 0;i < n;i++)
            a[i] = NULL;
        for(i = 0;i <n;i++)
        {
            for(j = 0;j < n;j++)
            {
                scanf("%d",&m);
                if(m == 1)
                    add(i,j);
            }
        }
        scanf("%d",&m);
        for(i = 0;i < m;i++)
        {
            scanf("%d%d",&u,&v);
            int flag= 0;
            p = a[u];
            while(p){
                if(p->data == v){
                    flag = 1;
                    break;}
                p=p->next;
            }
            if(flag)
                printf("Yes\n");
            else{
                printf("No\n");
            }
        }
    }
    return 0;
}