PAT1076 Forwards on Weibo

题目链接：

http://www.nowcoder.com/pat/5/problem/4306

题目描述：

根据微博的关注链，求一条微博可能被转发的最大次数。
x在转发L层的范围内微博可以被转发多少次。

输入数据:
7 3
3 2 3 4
0
2 5 6
2 3 1
2 3 4
1 4
1 5
2 2 6

共有7个用户，L=3
3 2 3 4 代表用户1关注3个人2、3、4，换句话说2，3，4的微博1会转。
以此类推。
求在3层内的2，6的微博可以最大被转发多少次。

题目分析：

根据关注链建立有向图，其实就是记录层的BFS的改进。用了略挫的办法，用两个queue搞定分层的问题。

代码：

#include<iostream>
#include<queue>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
using namespace std;
bool map[1005][1005];
bool vis[1005];
int N, L;
int query(int start)
{
    memset(vis,false,sizeof(vis));
    queue<int> q1;
    queue<int> q2;
    int level = 0;
    int cnt = 0;
    q1.push(start);
    vis[start] = true;
    while (level <= L){
        if (!q1.empty()){
            cnt += q1.size();
            while (!q1.empty()){
                int user = q1.front();
                q1.pop();
                for (int i = 1; i <= N; i++){
                    if (!vis[i] && map[user][i]){
                        q2.push(i);
                        vis[i] = true;
                    }
                }
            }
        }
        else if (!q2.empty()){
            cnt += q2.size();
            while (!q2.empty()){
                int user = q2.front();
                q2.pop();
                for (int i = 1; i <= N; i++){
                    if (!vis[i] && map[user][i]){
                        q1.push(i);
                        vis[i] = true;
                    }
                }
            }
        }
        level++;
    }
    return cnt-1;
}
int main()
{
    scanf("%d%d",&N,&L);
    memset(map,false,sizeof(map));
    for (int i = 1; i <= N; i++){
        int follows;
        int user;
        scanf("%d",&follows);
        while (follows--){
            scanf("%d",&user);
            map[user][i] = true;
        }
    }
    int k;
    scanf("%d",&k);
    for (int i = 0; i < k; i++){
        int start;
        scanf("%d",&start);
        printf("%d\n",query(start));
    }
    return 0;
}