loj黑暗城堡

黑暗城堡

題目描述
你知道黑暗城堡有\(N\)個房間，M 條能夠製造的雙向通道，以及每條通道的長度。
城堡是樹形的而且知足下面的條件：
設\(D_i\)爲若是全部的通道都被修建，第i號房間與第1號房間的最短路徑長度；
而\(S_i\)爲實際修建的樹形城堡中第i號房間與第1號房間的路徑長度；
要求對於全部整數\(i(1<= i <= N)\)，有\(S_i = D_i\)成立。
你想知道有多少種不一樣的城堡修建方案。固然，你只須要輸出答案對\(2 ^ {31} - 1\)取模以後的結果就好了。
輸入格式
第一行爲兩個由空格隔開的整數 ;
第二行到第n + 1行爲3個由空格隔開的整數x,y,l：表示x號房間與y號房間之間的通道長度爲l。
輸出格式
一個整數：不一樣的城堡修建方案數對\(2 ^ {31} - 1\)取模以後的結果。

題解：首先dij，求出1號點到其餘點的最短距離。
而後掃描全部邊，若是dis[v]==dis[u]+e[i].w，說明讓v點的距離爲給最短距離的路徑又多了一條，因此，cnt[v]++;
最終結果就是全部點的cnt[]相乘。

#include <cstdio>
#include <iostream>
#include <queue> 
#define orz cout << "AK IOI"

using namespace std;
const int maxn = 1010;
const int maxm = 500010;
const int mod = 2e31 - 1;

inline int read()
{
	int x = 0, f = 1;
	char ch = getchar();
	while (ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
	while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0';ch = getchar();}
	return x * f;
}
int n, m, dis[maxn], vis[maxn], cnt[maxn];
long long ans = 1;
struct edge{
	int u, v, w, nxt;
}e[maxm << 1];
int head[maxn], js;
void add(int u, int v, int w)
{
	e[++js].u = u;
	e[js].v = v;
	e[js].w = w;
	e[js].nxt = head[u];
	head[u] = js;
}
struct node{
	int w, now;
	bool operator < (const node &x)const
	{
		return w > x.w;
	}
};
priority_queue<node> q;
void dijkstra(int s)
{
	for(int i = 1; i <= n; i++)
	dis[i] = 0x7fffffff;
	dis[s] = 0;
	q.push((node{0, s}));
	while(!q.empty())
	{
		node t = q.top();
		q.pop();
		int u = t.now;
		if(vis[u]) continue;
		vis[u] = 1;
		for(int i = head[u]; i; i = e[i].nxt)
	    {
	    	int v = e[i].v;
	    	if(dis[v] > dis[u] + e[i].w)
	    	{
	    		dis[v] = dis[u] + e[i].w;
				q.push((node{dis[v], v})); 
			}
		}
	}
}
int main()
{
    n = read(), m = read();
    for(int i = 1; i <= m; i++)
    {
    	int u = read(), v = read(), w = read();
    	add(u, v, w);
    	add(v, u, w);
	}
	dijkstra(1);
	/*for(int i = 1; i <= n; i++)
	printf("%d ",dis[i]);*/
	for(int i = 1; i <= n; i++)
	   for(int j = head[i]; j; j = e[j].nxt)
	   {
	   	  int v = e[j].v;
	  	  if(dis[v] == dis[i] + e[j].w) 
		  cnt[v]++;
	   } 
	for(int i = 2; i <= n; i++)
	ans = ans * cnt[i] % mod;
	printf("%lld",ans);
	return 0;
}