关闭

浙大PAT甲级 1076

39人阅读 评论(0) 收藏 举报
分类:

广度优先搜索。用queue来实现,且用mark来表示userid是否出现。

AC代码:

#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 999999999
using namespace std;
vector<int> v[1001];
int n,m;
int mark[1001];
struct node
{
    int lay;
    int data;
};
queue<node> q;
void bfs(int x)
{
    node w;
    w.lay=0;
    w.data=x;
    q.push(w);
    while(!q.empty())
    {
        node tmp=q.front();
        q.pop();
        for(int i=0;i<v[tmp.data].size();i++)
        {
           node gg;
           gg.data=v[tmp.data][i];
           if(mark[gg.data]==1)
            continue;
           gg.lay=tmp.lay+1;
           if(gg.lay!=m)
           {
               q.push(gg);
           }
           mark[gg.data]=1;
        }
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int d;
        scanf("%d",&d);
        for(int j=1;j<=d;j++)
        {
            int t;
            scanf("%d",&t);
            v[t].push_back(i);
        }
    }
    int d;
    scanf("%d",&d);
    for(int i=0;i<d;i++)
    {
        memset(mark,0,sizeof(mark));
        int t;
        scanf("%d",&t);
        bfs(t);
        int ans=0;
        for(int j=1;j<=n;j++)
        {
            if(j==t) continue;
            if(mark[j]==1)
                ans++;
        }
        printf("%d\n",ans);
    }
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:7827次
    • 积分:940
    • 等级:
    • 排名:千里之外
    • 原创:91篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章分类
    文章存档