HNCPC 2016 湖南省程序设计竞赛 有向无环图

题目链接

http://acm.csu.edu.cn/csuoj/problemset/problem?pid=1804

题意

中文题：给个含有n个点的有向无环图，计算ans：

long long ans=0;for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)ans+=count(i,j)*a[i]*b[j];

count(i,j) 代表结点I到j可走路的条数。

思路

拓扑排序，因为是有向无环图，所以应该要找到入度为0的点，沿着这点所到的点，入度为0的点都能到达，将入度为0的点的a[i] ”传递“到它能直接到达的点，累加过去。

#include<cstdio>#include<queue>#include<iostream>#include<vector>#include<map>#include<cstring>#include<string>#include<set>#include<stack>#include<algorithm>#define cle(a) memset(a,0,sizeof(a))#define inf(a) memset(a,0x3f,sizeof(a))#define ll long long#define Rep(i,a,n) for(int i=a;i<=n;i++)using namespace std;#define INF2 9223372036854775807llconst int INF = ( 2e9 ) + 2;const ll maxn = 1e5+10;int d[maxn],a[maxn],b[maxn],In[maxn];vector<int> g[maxn];const int mod = 1e9+7;ll topsort(int n){    ll ans=0;    priority_queue<int> q;    for(int i=1;i<=n;i++)    {        if(In[i]==0)        {            q.push(i);        }    }    while(!q.empty())    {        int u=q.top();q.pop();        d[u]=(d[u]+a[u])%mod;        for(int i=0,L=g[u].size();i<L;i++)        {            int v=g[u][i];            In[v]--;            d[v] = (d[u]+d[v])%mod;            if(In[v]==0)            q.push(v);            ans = (ans+1LL*d[u]*b[v])%mod;        }    }    return ans%mod;}int main(){    int n,m;    while(~scanf("%d%d",&n,&m))    {        for(int i=1;i<=n;i++)g[i].clear();        memset(In,0,sizeof(In));        memset(d,0,sizeof(d));        for(int i=1;i<=n;i++)        scanf("%d%d",&a[i],&b[i]);        for(int i=0;i<m;i++)        {            int u,v;            scanf("%d%d",&u,&v);            g[u].push_back(v);            In[v]++;        }        ll ans=topsort(n);        cout<<ans<<endl;    }}