251742026-02-18 11:02:22KevinBarátnők (50 pont)cpp17Hibás válasz 13/50123ms9044 KiB
#include <bits/stdc++.h>
#include <climits>
#include <queue>
using namespace std;
using ll=long long;
using pll=pair<ll, ll>;
const ll MOD=1e9+7;
const ll INF=LONG_LONG_MAX;

int main() {
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    ll n, m; cin >> n >> m;
    vector<vector<pll>> graph(n, vector<pll> ());
    for (ll i=0; i<m; i++){
        ll a, b, c; cin >> a >> b >> c; a--; b--;
        graph[a].push_back({b, c});
        graph[b].push_back({a, c});
    }
    ll a, b; cin >> a >> b; a--; b--;
    vector<ll> disA(n, INF);
    vector<ll> disB(n, INF);
    vector<ll> waysA(n, 0);
    vector<ll> waysB(n, 0);
    priority_queue<pll, vector<pll>, greater<pll>> pq;
    disA[a]=0;
    waysA[a]=1;
    pq.push({0, a});
    while (pq.size()>0){
        ll disCurr=pq.top().first, idParent=pq.top().second;
        pq.pop();
        if (disCurr!=disA[idParent]) continue;
        for (auto& [idChild, road]:graph[idParent]){
            if (disA[idChild]==disCurr+road)
                waysA[idChild]+=waysA[idParent];
            if (disA[idChild]>disCurr+road){
                disA[idChild]=disCurr+road;
                waysA[idChild]=waysA[idParent];
                pq.push({disA[idChild], idChild});
            }
        }
    }
    disB[b]=0;
    waysB[b]=1;
    pq.push({0, b});
    while (pq.size()>0){
        ll disCurr=pq.top().first, idParent=pq.top().second;
        pq.pop();
        if (disCurr!=disB[idParent]) continue;
        for (auto& [idChild, road]:graph[idParent]){
            if (disB[idChild]==disCurr+road)
                waysB[idChild]+=waysB[idParent];
            if (disB[idChild]>disCurr+road){
                disB[idChild]=disCurr+road;
                waysB[idChild]=waysB[idParent];
                pq.push({disB[idChild], idChild});
            }
        }
    }
    ll total=waysA[b]*waysA[b]%MOD;
    ll bad=0;
    for (ll i=0; i<n; i++){
        if (disA[i]*2==disA[b] && disA[i]==disB[i]){
            bad+=waysA[i]*waysB[i]%MOD*waysA[i]*waysB[i]%MOD;
            bad%=MOD;
        }
    }
    for (ll i=0; i<n; i++){
        for (auto& [id, road]:graph[i]){
            if (disA[i]+road==disA[id] && disA[id]+disB[id]==disA[b] && disA[i]*2<disA[b] && disB[id]*2<disA[b]){
                bad+=waysA[i]*waysB[id]%MOD*waysA[i]*waysB[id]%MOD;
                bad%=MOD;
            }
        }
    }
    cout << (total-bad+MOD)%MOD;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base13/50
1Elfogadva0/01ms316 KiB
2Hibás válasz0/0120ms8624 KiB
3Hibás válasz0/22ms316 KiB
4Hibás válasz0/22ms316 KiB
5Elfogadva2/21ms316 KiB
6Hibás válasz0/22ms316 KiB
7Hibás válasz0/22ms316 KiB
8Elfogadva1/150ms5736 KiB
9Elfogadva2/276ms9044 KiB
10Hibás válasz0/2103ms8308 KiB
11Hibás válasz0/287ms8280 KiB
12Elfogadva2/287ms8580 KiB
13Hibás válasz0/2104ms8244 KiB
14Hibás válasz0/296ms8260 KiB
15Hibás válasz0/281ms8244 KiB
16Elfogadva1/1108ms8632 KiB
17Hibás válasz0/1123ms8988 KiB
18Elfogadva1/154ms5684 KiB
19Hibás válasz0/2107ms8888 KiB
20Hibás válasz0/2103ms8304 KiB
21Hibás válasz0/2119ms8624 KiB
22Elfogadva2/2101ms8372 KiB
23Hibás válasz0/2108ms8884 KiB
24Hibás válasz0/282ms8168 KiB
25Hibás válasz0/278ms7536 KiB
26Hibás válasz0/296ms8372 KiB
27Hibás válasz0/2107ms8648 KiB
28Elfogadva2/2114ms8888 KiB
29Hibás válasz0/2108ms8200 KiB