hihocoder #1136 : Professor Q's Software 数组解法

topic link:hihocoder#1136

这题在拓扑排序的基础上加了一点变化，那就是在结点（即题中的module）之间增加了一个信号，用信号将有向图连接起来，增大了构造图的难度，我为了加快速度采用了数组存储图结构。

主要维护以下数组：
1.module
1.sig:sig[i]表示接受信号为i的module链表的表头。我把所有signal的序号加1，这样空出序号0作为结束条件。
2.ind: ind[i]表示module[i]的入度(in-dgree,即有多少个信号指向它)。
3.init:表示professor的初始化信号。

在理解这几个数组的意思后就可以看代码了，好了，废话不多说，上代码：

#include<iostream>
#include<queue>
#include<memory.h>
using namespace std;
#define N 100005
#define modulo 142857 
int zeros[3] = { 0, 0, 0 };
int T,n, m,tot, sig[N], ind[N], init[N];
struct module{
    int next, cnt, e[3];
    module(int a = 0, int *c = zeros) :next(a), cnt(0){
        int i = 3;
        while (i--)e[i] = *(c + i);
    }
};
void add(module *mod,int s, int* e){
    for (int i = 0; i < 3; i++)e[i]++;
    mod[++tot] = module(sig[s + 1], e), sig[s + 1] = tot;
}
void solve(module *mod){
    queue<int> q;
    for (int i = 1; i <= n; i++){
        for (int j = 0; j < 3; j++){
            for (int k = sig[mod[i].e[j]]; k; k = mod[k].next){
                ind[k]++;
            }
        }
    }
    for (int i = 0; i < m; i++){
        for (int j = sig[init[i]]; j; j = mod[j].next)
            mod[j].cnt++;
    }
    for (int i = 1; i <= n; i++)if(!ind[i])q.push(i);
    while (!q.empty()){
        int i = q.front();
        q.pop();
        for (int j = 0; j < 3; j++){
            for (int k = sig[mod[i].e[j]]; k; k = mod[k].next){
                mod[k].cnt = (mod[k].cnt + mod[i].cnt) % modulo;
                ind[k]--;
                if (!ind[k])q.push(k);
            }
        }
    }
    for (int i = 1; i <= n; i++)cout << mod[i].cnt << ' ';
    cout << endl;
}
int main(){
    ios::sync_with_stdio(false);
    cin >> T;
    while (T--){
        cin >> n >> m;
        module *mod = new module[n+1];
        memset(sig, 0, sizeof(sig));
        memset(ind, 0, sizeof(ind));
        memset(init, 0, sizeof(init));
        tot = 0;
        for (int i = 0; i < m; i++){
            cin >> init[i];
            init[i]++;
        }
        for (int i = 0; i < n; i++){
            int s, k, e[3] = { -1,-1,-1 };
            cin >> s >> k;
            for (int j = 0; j < k; j++)cin >> e[j];
            add(mod,s, e);
        }
        solve(mod);
    }
    return 0;
}