PAT 1076. Forwards on Weibo (30) Dijkstra应用

/*************************
题意:
此题一半时间花在理解题意上。。。
每个人都会关注(follow)一些人。
当他关注的人发了新微博时，他也一定会转发。

题目里给的user_list[i]是i所关注的人，而不是i的粉丝。
然后问转发量会有多少， 并把转发量限定在L级之内（不是L级，是L级及L级以下）
************************/
/***********************
解题思路：
	可以理解为有向图的最短路问题。
	A关注了B，那么可表示为  A<-B， 其距离为1
	即B发的东西，可以传到A那里去。
	那么对于给定的图，对于要求的点，求一次Dijkstra
	然后统计距离在L或L之内的点的个数（不能包括自己）

注意点：
	1.询问次数K的数量是不确定的，可能会反复询问一个用户，所以要弄一个数组保存结果
	2.记住下标是1-N,而不是0-N-1，  很多次错都因为这个。
*************************/
/***********************
笔记：
*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 1005
#define INF 0x7ffffff

int dis[M][M];
int ans[M];
int n, L;
int dij(int s){
	int vis[M],d[M];
	int i,j;

	if(ans[s] != -1)
		return ans[s];

	for(i = 1;i <= n;i++){
		vis[i]=0;
		d[i]=INF;
	}
	d[s] = 0;
	int dmin,select;
	for(i = 1;i <= n; i++){
		dmin = INF;
		for(j = 1;j <= n; j++){
			if(!vis[j] && d[j] < dmin){
				dmin = d[j];
				select = j;
			}
		}
		if(dmin == INF)
			break;
		vis[select] = 1;
		for(j = 1;j <= n;j++){
			if(!vis[j] && d[select] + dis[select][j] < d[j]){
				d[j] = d[select] + dis[select][j];	
			}
		}
	}

	int sum = 0;
	for(i = 1;i <= n;i++){
		if(d[i] <= L && i != s)
			sum++;
	}
	ans[s] = sum;
	return sum;
}

int main(){
	int i,k,follow,j;
	scanf("%d%d",&n,&L);

	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++){
			if(i!=j)
				dis[i][j] = INF;
			else dis[i][j] = 0;
		}

	for(i=1;i <= n;i++){
		scanf("%d",&k);
		while(k--){
			scanf("%d",&follow);
			dis[follow][i] = 1;
		}
	}

	int s;
	for(i = 1;i <= n;i++){
		ans[i] = -1;
	}
	scanf("%d",&k);
	while(k--){
		scanf("%d",&s);
		printf("%d\n",dij(s));
	}
	return 0;
}