Improve SPAM （ 拓扑排序 ）

Improve SPAM （ 拓扑排序 ）

这里有n个地址，其中m个地址是服务器地址，编号为1～m，剩下的n-m个地址为用户地址。每个服务器地址都包含若干其他地址，当某个服务器地址收到一封邮件，它会把这封邮件转发给它包含的所有地址。

例如：1、2和3是邮件列表，而4和5是客户电子邮件。
此外，邮件列表1包含邮件列表2和3，邮件列表2包含客户电子邮件
4和5，而邮件列表3包含客户电子邮件4和邮件列表2。现在假设
邮件被发送到邮件列表1。这意味着该列表已按照上述说明进行处理，并且
然后还会处理邮件列表2和3。 处理邮件列表2后，该消息
发送到客户电子邮件4和5。处理邮件列表3后，第二条消息发送到
客户电子邮件4，并再次处理邮件列表2，这会产生发送给客户的第三条消息
电子邮件4和第二条消息发送到客户电子邮件5。因此，总共有5条消息发送到客户电子邮件。

但是这样4和5都收到了重复的邮件，你需要维修这个系统，使得每个用户不会收到重复邮件。

一开始给1号地址发送一封邮件，要求分别输出维修前和维修后所有用户收到的邮件总数。

 

思路：把每个地址当作是图中的点，1号地址包含2,3，就让节点1和2,3节点相连。这样先从1跑一遍bfs （ 注意加via标记 ），如果到了用户地址（ 编号>m ）ans2 ++， 这样就可以统计不重复的送达邮件能送多少封。我们早bfs的时候记录bfs的路径，新建一张图（这样就舍弃掉了无用边），在跑一边拓扑排序，期间更新每个节点应该被访问的次数，ci[to] = ci[to] + ci[from] 。

代码：

#include <bits/stdc++.h>

using namespace std;

const int mod = 1e9+7;
const int maxn = 2e3+10;
long long ans1,ans2;
vector<int> G[maxn];
int n,m,cnt;
int via[maxn];
int head[maxn];
int in[maxn];
int ci[maxn];
struct node {
    int to,nxt;
}e[maxn*maxn];

void addage( int u, int v )
{
    e[cnt].to = v;
    e[cnt].nxt = head[u];
    head[u] = cnt++;
    in[v] ++;
}

void bfs1()
{
    ans2 = 0;
    memset(via,0,sizeof(via));
    queue <int> Q;
    Q.push(1);
    via[1] = 1;
    while ( !Q.empty() ) {
        int t = Q.front(); Q.pop();
        if ( t>m ) ans2++;
        for ( int i=0; i<G[t].size(); i++ ) {
            int x = G[t][i];
            addage(t,x); // 新建一张图，只连接有用的边
            if ( via[x]==0 ) {
                via[x] = 1;
                Q.push(x);
            }
        }
    }
}

void topu()
{
    memset(ci,0,sizeof(ci));
    queue <int> Q;
    Q.push(1);
    ci[1] = 1;
    while ( !Q.empty() ) {
        int t = Q.front(); Q.pop(); // 当前需要删掉 t 点
        for ( int i=head[t]; i!=-1; i=e[i].nxt ) {
            int to = e[i].to;  // t点连接to点
            ci[to] = (ci[t]+ci[to])%mod; // 便利到to点的次数
            in[to] --;   // to点的入度减一
            if ( in[to]==0 ) {
                Q.push(to);
            }
        }
    }
    ans1 = 0;
    for ( int i=m+1; i<=n; i++ ) {
        ans1 = (ans1+ci[i])%mod;
    }
}

int main()
{
    cnt = 0;
    memset(head,-1,sizeof(head));
    int i,j,k,x;
    cin >> n >> m;
    for ( i=1; i<=m; i++ ) {
        cin >> k;
        for ( j=1; j<=k; j++ ) {
            scanf("%d",&x);
            G[i].push_back(x);
        }
    }
    bfs1(); // bfs，找到种类数
    topu(); // 拓扑排序，顺便找到次数关系
    printf("%lld %lld\n",ans1,ans2);

    return 0;
}