Adjacent Node Sum (图的邻接表表示 C语言)

Adjacent Node Sum

题目内容：

給定一個節點有權重的圖，
請你計算與一個節點相鄰的所有節點的權重和。
節點編號為1~N。
每個節點的編號即為他的權重。


输入格式:
只有一組測資。
第一行有三個數字，N、M、Q。
N表示這張圖有多少節點，
M表示這張圖有多少邊，
Q表示會有多少個詢問。
之後M行，每行有兩個數字a、b，
代表a與b是相鄰的。
之後Q行，每行有一個數字x，
代表詢問的點編號。
給定的編不會重複，
而且不會有自己和自己相連的情況。
測資範圍：
0 < N < 1000
0 < M < 20000
0 < Q < 2000


输出格式：
對於每一個詢問，計算與該節點相鄰的所以節點權重和。
將所有詢問的答案加總後再輸出。


输入样例：
10 8 2
1 2
1 3
2 5
2 6
3 8
10 9
8 6
3 5
1
3


输出样例：
19


时间限制：100ms内存限制：16000kb

coding using c

#include <stdio.h>
#include <stdlib.h>


struct _node{
	int value;
	struct _node* next;
};

typedef struct _node Node;

void insert_node(Node* array[], int idx, int value)
{
	Node* p;
	p = (Node*)malloc(sizeof(Node));
	p->value = value;
	if(array[idx]==NULL)
	{
		p->next = NULL;
		array[idx] = p;
	}
	else
	{
		p->next = array[idx];
		array[idx] = p;
	}
}

int count_sum(Node* head)
{
	int sum=0;
	while(head)
	{
		sum+=head->value;
		head=head->next;
	}
	return sum;
}
/* run this program using the console pauser or add your own getch, system("pause") or input loop */

int main(int argc, char *argv[]) {
	int n = 10, edges = 8, query = 2, idx, value,sum;
	scanf("%d %d %d",&n,&edges,&query);
	Node * array[n+1];
	int i=0,j=0;
	for(i=0;i<n+1;i++)
	{
		array[i]=NULL;
	}
	for(i=0;i<edges;i++)
	{
		scanf("%d %d", &idx, &value);		
		insert_node(array, idx, value);
		insert_node(array, value, idx);
	}
	sum=0;
	for(i=0;i<query;i++)
	{
		scanf("%d",&idx);
		sum+=count_sum(array[idx]);
	}
	printf("%d",sum);
	
	return 0;
}