Problem:
Score : 700 points

Problem Statement

We have a graph with N vertices and M edges, and there are two people on the graph: Takahashi and Aoki.

The i-th edge connects Vertex U_i and Vertex V_i.
The time it takes to traverse this edge is D_i minutes, regardless of direction and who traverses the edge (Takahashi or Aoki).

Takahashi departs Vertex S and Aoki departs Vertex T at the same time. Takahashi travels to Vertex T and Aoki travels to Vertex S, both in the shortest time possible.
Find the number of the pairs of ways for Takahashi and Aoki to choose their shortest paths such that they never meet (at a vertex or on an edge) during the travel, modulo 10^9 + 7.

Constraints

1 \leq N \leq 100 000

1 \leq M \leq 200 000

1 \leq S, T \leq N

S \neq T

1 \leq U_i, V_i \leq N (1 \leq i \leq M)

1 \leq D_i \leq 10^9 (1 \leq i \leq M)

If i \neq j, then (U_i, V_i) \neq (U_j, V_j) and (U_i, V_i) \neq (V_j, U_j).

U_i \neq V_i (1 \leq i \leq M)

D_i are integers.

The given graph is connected.

Input

Input is given from Standard Input in the following format:

N M
S T
U_1 V_1 D_1
U_2 V_2 D_2
:
U_M V_M D_M

Output

Print the answer.

Sample Input 1

4 4
1 3
1 2 1
2 3 1
3 4 1
4 1 1

Sample Output 1

2

There are two ways to choose shortest paths that satisfies the condition:

Takahashi chooses the path 1 \rightarrow 2 \rightarrow 3, and Aoki chooses the path 3 \rightarrow 4 \rightarrow 1.

Takahashi chooses the path 1 \rightarrow 4 \rightarrow 3, and Aoki chooses the path 3 \rightarrow 2 \rightarrow 1.

Sample Input 2

3 3
1 3
1 2 1
2 3 1
3 1 2

Sample Output 2

2

Sample Input 3

3 3
1 3
1 2 1
2 3 1
3 1 2

Sample Output 3

2

Sample Input 4

8 13
4 2
7 3 9
6 2 3
1 6 4
7 6 9
3 8 9
1 2 2
2 8 12
8 6 9
2 5 5
4 2 18
5 3 7
5 1 515371567
4 8 6

Sample Output 4

6

Solution in C++14 (GCC 5.4.1) (judge verdict: Wrong Answer)


#include <bits/stdc++.h>

#define FOR(i,b,e) for(int i=(b); i <= (e); ++i)
#define FORD(i,b,e) for(int i=(b); i >= (e); --i)
#define SIZE(c) (int) (c).size()
#define FORE(i,c) FOR(i,0,SIZE(c)-1)
#define FORDE(i,c) FORD(i,SIZE(c)-1,0)

#define pb push_back
#define mp make_pair
#define st first
#define nd second


using namespace std;

typedef long long ll;
typedef pair <int,int> pii;
typedef pair <ll,ll> pll;

typedef vector <int> VI;
typedef vector <bool> VB;
typedef vector <pii> VP;
typedef vector <ll> VL;
typedef vector <pll> VPL;

typedef vector <VI> VVI;
typedef vector <VL> VVL;
typedef vector <VB> VVB;
typedef vector <VP> VVP;

const int MOD = 1000000007;

typedef ll dist_t;
const dist_t INF = 1LL << 60;

vector <dist_t> dijkstra(int src, VVP &g) {
    int n = SIZE(g);
    vector <dist_t> dist(n, INF);

    dist[src] = 0;
    
    priority_queue <pair<dist_t,int>> q;
    q.push({0, src});
    
    while (!q.empty()) {
        auto e = q.top(); q.pop();
        int v = e.nd;
        
        if (e.st != -dist[v]) continue;
        
        for (auto &e : g[v]) {
            int u = e.st;
            
            if (dist[u] > dist[v] + e.nd) {
                dist[u] = dist[v] + e.nd;
                q.push({-dist[u], u});
            }
        }
    }
    
    return dist;
}

int cnt(int v, VVI &gRev, VI &ans) {
    if (ans[v] == -1) {
        ans[v] = 0;
        for (int u : gRev[v]) {
            ans[v] = (ans[v] + cnt(u, gRev, ans)) % MOD;
        }
    }
    
    return ans[v];
}

VI countPaths(int src, VVI &gRev) {
    VI ans(gRev.size(), -1);
    ans[src] = 1;
    
    FORE(i,gRev) {
        cnt(i, gRev, ans);
    }
    
    return ans;
}

/*************************************************************************/

int main() {
    ios_base::sync_with_stdio(0);
    
    int n, m;
    cin >> n >> m;
    
    int s, t;
    cin >> s >> t; s--; t--;
    
    VVP g(n);
    while (m--) {
        int u, v, len;
        cin >> u >> v >> len; u--; v--;
        
        g[u].pb({v, len});
        g[v].pb({u, len});
    }
    
    auto dist = dijkstra(s, g);
    
    VVI sDag(n), tDag(n);
    FOR(i,0,n-1) for (auto &e : g[i]) {
        int j, len;
        tie(j, len) = e;
        
        if (dist[j] == dist[i] + len) {
            sDag[i].pb(j);
            tDag[j].pb(i);
        }
    }
    
    auto sPaths = countPaths(s, tDag);
    auto tPaths = countPaths(t, sDag);
    
    ll ans = ((ll) sPaths[t] * sPaths[t]) % MOD;
    ll whole = dist[t];
    
    if (whole % 2 == 0) {
        ll mid = whole / 2;
        
        FOR(i,0,n-1) if (dist[i] == mid) {
            ll left = ((ll) sPaths[i] * sPaths[i]) % MOD;
            ll right = ((ll) tPaths[i] * tPaths[i]) % MOD;
            ll here = (left * right) % MOD;
            
            ans = (ans + MOD - here) % MOD;
        }
        
        FOR(i,0,n-1) for (int j : sDag[i]) {
            if (dist[i] < mid && dist[j] > mid) {
                ll left = ((ll) sPaths[i] * sPaths[i]) % MOD;
                ll right = ((ll) tPaths[j] * tPaths[j]) % MOD;
                ll here = (left * right) % MOD;
                
                ans = (ans + MOD - here) % MOD;
            }
        }
    }
    
    cout << ans;

    return 0;
}

/*************************************************************************/