[HAOI2012]ROAD(最短路+拓扑排序)

做法

先跑出每个点的最短路 并且标记哪些边 在最短路上
在拓扑排序 记录一下到 某一个的方案和 从某一个点到 终点的方案
乱搞一下就行

dij比spfa能快一点（水题练练最短路

#include <bits/stdc++.h>
using namespace  std;
//#define  int long long
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
#define fi first
#define se second
#define pb  push_back
#define inf 1<<62
#define endl "\n"
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define de_bug(x) cerr << #x << "=" << x << endl
#define all(a) a.begin(),a.end()
#define IOS   std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define  fer(i,a,b)  for(int i=a;i<=b;i++)
#define  der(i,a,b)  for(int i=a;i>=b;i--)
const int mod = 1e9 + 7;
const int N=1600;
const int M=5010;
#define inf 0x3f3f3f3f
int h[M],e[M],ne[M],w[M];
int cnt;
int d[M];
int vis[M];
int n,m;
int e1[M];
int f[M];
int ans[M];
int cnt1[M];
int cnt2[M];
int du[M];

void add(int a,int b,int c)
{
    e1[cnt]=a, e[cnt]=b,w[cnt]=c,ne[cnt]=h[a],h[a]=cnt++;
}
void spfa(int s)
{
    memset(vis,0,sizeof(vis));
    memset(d,inf,sizeof(d));
    queue<int>q;
    q.push(s);
    d[s]=0;
    vis[s]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=h[u];~i;i=ne[i])
        {
            int v=e[i];
            if(d[v]>d[u]+w[i])
            {
                d[v]=d[u]+w[i];
                if(!vis[v])
                {
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}
void dij(int s){
    memset(vis,0,sizeof(vis));
    memset(d,inf,sizeof(d));
    d[s]=0;
    priority_queue<pii,vector<pii>,greater<pii>>q;
    q.push({d[s],s});
    while(!q.empty()){
        int u=q.top().se;
        q.pop();
        if(vis[u])continue;
        vis[u]=1;
        for(int i=h[u];~i;i=ne[i]){
            int v=e[i];
            if(d[v]>d[u]+w[i]){
                d[v]=d[u]+w[i];
                q.push({d[v],v});
            }
        }
    }
}

void topo(int x)
{
    memset(cnt1,0,sizeof(cnt1));
    memset(cnt2,0,sizeof(cnt2));
    vector<int>path;
    queue<int>q;
     q.push(x);
    for(int i=0;i<cnt;i++)
    {
        if(f[i])du[e[i]]++;
    }
    cnt1[x]=1;
    while(!q.empty())
    {
        int t=q.front();
        path.pb(t);
        q.pop();
        for(int i=h[t];~i;i=ne[i])
        {
            if(!f[i])continue;
            int v=e[i];
            du[v]--;
            if(!du[v])q.push(v);
            cnt1[v]=(cnt1[v] +cnt1[t])%mod;
        }

    }
    for(int i=(int)path.size()-1;i>=0;i--)
    {
        int t=path[i];
        cnt2[t]++;
        for(int j=h[t];~j;j=ne[j])
        {
            if(!f[j])continue;
            int v=e[j];
            cnt2[t]=(cnt2[v] +cnt2[t])%mod;
        }
    }
    for(int i=0;i<cnt;i++)
    {
        int v=e[i];
        int u=e1[i];
        if(f[i])
            (ans[i]+= (1ll)*cnt1[u]*cnt2[v])%=mod ;
    }
}

signed main()
{
    memset(h,-1,sizeof(h));
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    for(int i=1;i<=n;i++)
    {
       dij(i);
        // spfa(i);
        memset(f,0,sizeof(f));
        for(int j=0;j<cnt;j++)
        {
            int a=e1[j];
            int b=e[j];
            int val=w[j];
            if(d[a]+val==d[b])
            {
                f[j]=1;
            }
        }
        topo(i);
    }
    for(int i=0;i<cnt;i++)
    {
        cout<<ans[i]<<endl;
    }
    return 0;

}