cogs 2509. 森林大礼包

Ÿ 2509. 森林大礼包

★   输入文件：three_squirrels.in   输出文件：three_squirrels.out   简单对比
时间限制：1 s   内存限制：256 MB

【题目描述】

   有一天紫萱学姐心情不好，学长就买了一个自己最喜欢吃的森林大礼包送给学姐，因为反正最后都是自己吃。

   森林大礼包里一共可能有n种包装袋，编号为0~n-1，0号包装袋里有一颗坚果，所有1~n-1号包装袋里都装着若干包其他包装袋，但是不会出现一个包装袋的某个子包装袋里包含这这个包装袋本身的灵异现象。

  最终森林大礼包里也会装着若干个包装袋，为了方便表示，我们将森林大礼包视为n号包装袋。请你帮助紫萱学姐计算这个大礼包里一共有多少颗坚果，由于答案可能很大，输出这个数对10^9+7取余的结果。

【输入格式】

  输入数据第一行为一个整数n，意义如题目所示。

  接下来n行，第i行一个整数ki，代表i号包装袋中含有ki个其他包装袋，接下来ki个整数aij代表这ki个包装袋的编号。

【输出格式】

  一个整数，意义如题目所示。

【样例输入】

7
1 0
2 0 1
2 1 2
2 2 3
2 3 4
2 4 5
2 5 6

【样例输出】

21

【提示】

样例中7个包装袋中的坚果数组成了一个斐波那契数列。

对于40%的数据，n≤10。

对于70%的数据，aij＜i。

对于100%的数据，n≤100000，0＜ki≤10，0≤aij＜n且≠i，对于1≤x＜y≤ki，aix可能与aiy相等，即一个包装袋内可能装有多个相同编号的包装袋。

拓扑：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>

using namespace std;
const int N = 5e5 + 10;

#define lgj 1000000007
#define LL long long

LL n, now = 1, head[N], in[N], Ans[N];
struct Node{
    int u, v, nxt;
}E[N << 1];
queue <int> Q;

inline int read(){
    int x = 0; char c = getchar();
    while(c < '0' || c > '9') c  =getchar();
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x;
}

inline void add(int u, int v){
    E[now].v = v;
    E[now].nxt = head[u];
    head[u] = now ++;
}

inline void toposort(){
    while(!Q.empty())
    {
        int topp = Q.front();
        Q.pop();
        for(int i = head[topp]; ~ i; i = E[i].nxt)
        {
            int v = E[i].v;
            in[v] --;
            Ans[v] = (Ans[v] % lgj + Ans[topp] % lgj) % lgj;
            if(!in[v]) Q.push(v);
        }
    }
}

int main()
{
    freopen("three_squirrels.in", "r", stdin);
    freopen("three_squirrels.out", "w", stdout);
    n = read();
    for(int i = 0; i <= n; i ++) head[i] = -1;
    for(int i = 1; i <= n; i ++)
    {
        in[i] = read();
        for(int j = 1; j <= in[i]; j ++) add(read(), i);
    }
    for(int i = head[0]; ~ i; i = E[i].nxt)
    {
        int v = E[i].v;
        in[v] --;
        Ans[v] ++;
        if(!in[v]) Q.push(v);
    }
    toposort();
    printf("%lld", Ans[n]);
    return 0;
}