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 Rust (1.15.1) (judge verdict: Wrong Answer)

use std::cmp::Ordering;
use std::collections::BinaryHeap;
use std::collections::BTreeSet;
use std::collections::BTreeMap;
use std::usize;
use std::cmp;

const MOD: usize = 1000000007;

#[derive(Copy, Clone, Eq, PartialEq)]
struct State {
    cost: usize,
    position: usize,
}

impl Ord for State {
    fn cmp(&self, other: &State) -> Ordering {
        other.cost.cmp(&self.cost)
    }
}

impl PartialOrd for State {
    fn partial_cmp(&self, other: &State) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

fn main() {
    let (n, m) = {
        let v = read_values::<usize>();
        (v[0], v[1])
    };
    let mut graph = (0..n).map(|_| Vec::new()).collect::<Vec<_>>();
    let (s, t) = {
        let v = read_values::<usize>();
        (v[0] - 1, v[1] - 1)
    };
    for _ in 0..m {
        let v = read_values::<usize>();
        let from = (v[0] - 1);
        let to = (v[1] - 1);
        let d = v[2];
        graph[from].push((to, d));
        graph[to].push((from, d));
    }

    let (count_s, dist_s) = first_dijkstra(&graph, s, t);
    let (count_t, dist_t) = first_dijkstra(&graph, t, s);

    let shortest_dist = dist_s[t];

    let mut edge_set = BTreeSet::new();
    for from in 0..n {
        for e in &graph[from] {
            let (to, c) = *e;

            if dist_s[from] + c + dist_t[to] == shortest_dist && dist_s[from] * 2 < shortest_dist && dist_t[to] * 2 < shortest_dist {
                edge_set.insert((from, to));
            }
        }
    }

    let mut ans = count_s[t] * count_t[s];
    ans %= MOD;
    for e in &edge_set {
        let (from, to) = *e;
        let (from, to) = if dist_s[from] < dist_s[to] {
            (from, to)
        } else {
            (to, from)
        };

        let mut c = (count_s[from] * count_s[from]) % MOD;
        c = (c * count_t[to]) % MOD;
        c = (c * count_t[to]) % MOD;
        ans = (ans + MOD - c) % MOD;
    }

    for v in 0..n {
        if dist_s[v] * 2 == shortest_dist {
            let mut c = (count_s[v] * count_s[v]) % MOD;
            c = (c * count_t[v]) % MOD;
            c = (c * count_t[v]) % MOD;
            ans = (ans + MOD - c) % MOD;
        }
    }
    println!("{}", ans);
}

fn first_dijkstra(graph: &Vec<Vec<(usize, usize)>>, s: usize, t: usize) -> (Vec<usize>, Vec<usize>) {
    let n = graph.len();
    let mut count = vec![0; n];
    count[s] = 1;
    let mut dist = vec![usize::MAX; n];
    dist[s] = 0;
    let mut heap = BinaryHeap::new();
    heap.push(State { cost: 0, position: s });
    while let Some(State { cost, position }) = heap.pop() {
        if cost > dist[t] {
            break;
        }
        for e in &graph[position] {
            let (to, d) = *e;
            if dist[to] < dist[position] + d {
                continue;
            } else if dist[to] > dist[position] + d {
                dist[to] = dist[position] + d;
                count[to] = 0;
                heap.push(State { cost: dist[to], position: to });
            }

            if dist[to] == dist[position] + d {
                count[to] += count[position];
                if count[to] >= MOD {
                    count[to] -= MOD;
                }
            }
        }
    }
    (count, dist)
}

fn read_line() -> String {
    let stdin = std::io::stdin();
    let mut buf = String::new();
    stdin.read_line(&mut buf).unwrap();
    buf
}

fn read_values<T>() -> Vec<T> where T: std::str::FromStr, T::Err: std::fmt::Debug {
    read_line()
        .split(' ')
        .map(|a| a.trim().parse().unwrap())
        .collect()
}