251722026-02-18 10:56:48KevinBarátnők (50 pont)cpp17Time limit exceeded 2/501.1s9384 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> 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();
        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();
        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;
            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;
                bad%=MOD;
            }
        }
    }
    cout << (total-bad+MOD)%MOD;
}
SubtaskSumTestVerdictTimeMemory
base2/50
1Accepted0/01ms316 KiB
2Time limit exceeded0/01.085s8880 KiB
3Wrong answer0/26ms316 KiB
4Wrong answer0/210ms316 KiB
5Wrong answer0/23ms316 KiB
6Wrong answer0/23ms316 KiB
7Wrong answer0/24ms548 KiB
8Accepted1/148ms5684 KiB
9Wrong answer0/286ms9176 KiB
10Time limit exceeded0/21.1s8972 KiB
11Time limit exceeded0/21.08s8756 KiB
12Wrong answer0/2100ms9384 KiB
13Time limit exceeded0/21.087s8884 KiB
14Time limit exceeded0/21.09s8884 KiB
15Time limit exceeded0/21.085s8884 KiB
16Time limit exceeded0/11.088s8632 KiB
17Time limit exceeded0/11.083s8500 KiB
18Accepted1/148ms5684 KiB
19Time limit exceeded0/21.09s8816 KiB
20Time limit exceeded0/21.1s8624 KiB
21Time limit exceeded0/21.087s8880 KiB
22Time limit exceeded0/21.087s8372 KiB
23Time limit exceeded0/21.09s8884 KiB
24Time limit exceeded0/21.087s7532 KiB
25Time limit exceeded0/21.1s7220 KiB
26Time limit exceeded0/21.082s8372 KiB
27Time limit exceeded0/21.083s8824 KiB
28Time limit exceeded0/21.1s8880 KiB
29Time limit exceeded0/21.1s8196 KiB