【Henu ACM Round#14 F】 President and Roads

【链接】 我是链接,点我呀:)
【题意】

在这里输入题意

【题解】

处理出起点到任意点的最短路以及最短路条数=>dis[0][i],cnt[0][i] 然后 把所有的边反向 处理出在反图上终点到任意点的最短路以及最短路条数=>dis[1][i],cnt[1][i] dis数组的初值为-1,表示无穷大 设起点到终点的最短路为mini 起点到终点的最短路条数为minicnt 对于每一条边(x,y,z) 如果dis[0][x]==-1 || dis[1][y]==-1代表x或者y不能到达起点或终点 则直接输出NO 否则 如果dis[0][x]+dis[1][y]==mini 且 cnt[0][x]*cnt[1][y]==minicnt的话 输出YES.这条边肯定在最短路上 (这种情况是把重边也考虑进去了的,因为如果有重边那么minicnt肯定不等于cnt[0][x]*cnt[1][y],肯定还要乘上重边的个数的 否则 让z减小到满足dis[0][x]+dis[1][y]+z求最短路条数要用dijkstra算法
用spfa会错
这道题某个点卡1e9+7模数
改成987654321就对了

【代码】

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 1e5;
const ll MOD = 987654321;

int n,m,s,t;
ll cnt[2][N+10];
ll dis[2][N+10];
vector<pair<int,int> > g[2][N+10];
pair<int,pair<int,int> > bian[N+10];
queue<int> dl;
map<pair<int,pair<int,int> >,int>dic;

void get_dis(int s,int k){
    for (int i = 1;i <= n;i++) dis[k][i] = -1;
    cnt[k][s] = 1;
    dis[k][s] = 0;
    priority_queue<pair<ll,int> ,vector<pair<ll,int> >,greater<pair<ll,int> > > pq;
    pq.push(make_pair(0,s));
    while (!pq.empty()){
        auto temp1 = pq.top();
        pq.pop();
        int x = temp1.second;ll dis0 = temp1.first;
        if (dis[k][x]!=dis0) continue;
        for (auto temp:g[k][x]){
            int y = temp.first;ll z = temp.second;
            if (dis[k][y]==-1 || dis[k][y]>dis0+z){
                dis[k][y] = dis0 + z;
                cnt[k][y] = cnt[k][x];
                pq.push({dis[k][y],y});
            }else if (dis[k][y]==dis0+z) cnt[k][y]=(cnt[k][y]+cnt[k][x])%MOD;
        }
    }
}

int main()
{
    #ifdef LOCAL_DEFINE
        freopen("rush.txt","r",stdin);
    #endif // LOCAL_DEFINE
    ios::sync_with_stdio(0),cin.tie(0);
    cin >> n >> m >>s>>t;
    for (int i = 1;i <= m;i++){
        int x,y,z;
        cin >> x >> y >>z;
        dic[{x,{y,z}}]++;
        bian[i] = {x,{y,z}};
        g[0][x].push_back({y,z});
        g[1][y].push_back({x,z});
    }
    get_dis(s,0);
    get_dis(t,1);
    for (int i = 1;i <= m;i++){
        int x,y;
        x = bian[i].first;y = bian[i].second.first;
        if (dis[0][x]<0 || dis[1][y]<0) {
            cout<<"NO"<<endl;
            continue;
        }
        ll z = bian[i].second.second;
        if (dis[0][x]+dis[1][y]+z==dis[0][t] && (cnt[0][x]*cnt[1][y]%MOD)==cnt[0][t]){
                cout<<"YES"<<endl;
        }else{
            ll need = dis[0][x]+dis[1][y]+z-dis[0][t]+1;
            if ((z-need)>=1){
                cout<<"CAN "<<need<<endl;
            }else{
                cout<<"NO"<<endl;
            }
        }
    }
    return 0;
}