AcWing 刷题计划:0x60 图论——AcWing 349. 黑暗城堡

1.题目描述:
在这里插入图片描述
2.题意:
略。

3.思路:
最短路径树。 这题和我周训那题类似,这不过这次是把边权改成了w,我们跑一遍迪杰斯特拉算法,然后再去找每个节点有多少个前驱可以选择。然后乘起来就搞定了。

4.代码:

//AcWing 349. 黑暗城堡
//#include<bits/stdc++.h>
//#pragma GCC optimize(3,"Ofast","inline")
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
//#include<random>
#include<cstdlib>
#include<ctime>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<limits.h>
#define FAST ios::sync_with_stdio(false)
#define DEV_RND ((int)rand()*RAND_MAX+rand())
#define RND(L,R) (DEV_RND%((R)-(L)+1)+(L))
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<n;++i)
#define per(i,n,a) for(int i=n-1;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define li inline
#define re register
using namespace std;
//typedef uniform_int_distribution<int> RNDI;
typedef pair<int,int> PII;
typedef vector<int> VI;
typedef double db;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int Hash = 131;//13331
const int maxn = 1e3+5;
const int maxm = 5e6+5;
const int mod = 1e9+7;
const int inf = 0x3f3f3f3f;
const double eps = 1e-7;
const double pi = acos(-1);
//int dir[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
//li int f(int x){return x==par[x]?par[x]:par[x]=f(par[x]);}
//mt19937 eng(time(0));
//li int RND(int L,int R){RNDI rnd(L,R);return rnd(eng);}
li ll lowbit(ll x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
li ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res%MOD;}
li ll qmul(ll a,ll b,ll MOD=mod){return (a*b-(ll)((long double)a/MOD*b)*MOD+MOD)%MOD;}
li ll inv(ll x,ll p){return qpow(x,p-2,p);}
li ll jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
db f(db x){return x;}
li db sim(db l,db r){return (f(l)+4.*f((l+r)/2.)+f(r))*(r-l)/6.;}
db asr(db l,db r,db ans,db eps){db m=l+(r-l)/2.,L=sim(l,m),R=sim(m,r);return fabs(L+R-ans)<=15.*eps?L+R+(L+R-ans)/15.:asr(l,m,L,eps/2)+asr(m,r,R,eps/2);}
db asr(db l,db r,db eps){return asr(l,r,sim(l,r),eps);}
namespace IO
{
	li ll read()
	{
		ll x=0,sign=1;char c=getchar();
		while(c>'9'||c<'0') {if(c=='-') sign=-1;c=getchar();}
		while('0'<=c&&c<='9') x=x*10+c-'0',c=getchar();
		return x*sign;
	}
	template<typename T>
	li void write(T x,char t='\n')
	{
		if(x<0){x=-x;putchar('-');};
		static int sta[25];int top=0;
		do{sta[top++]=x%10;}while(x/=10);
		while(top) putchar(sta[--top]+'0');
		putchar(t);
	}
}
using namespace IO;
/*-------------head-------------*/
int head[maxn],cnt;
struct Edge{int v,w,nx;}edge[maxm<<1];
li void add(int u,int v,int w){edge[++cnt]={v,w,head[u]};head[u]=cnt;}
//
int n,m;
int dis[maxn];
bool vis[maxn];
void dij(int s)
{
	rep(i,1,n+1) dis[i]=inf,vis[i]=0;
	dis[s]=0;
	priority_queue<PII> q;
	q.push(mp(dis[s],s));
	while(!q.empty())
	{
		auto t=q.top();q.pop();
		int u=t.se;
		if(vis[u]) continue;vis[u]=true;
		for(int i=head[u];i;i=edge[i].nx)
		{
			int v=edge[i].v;
			int w=edge[i].w;
			if(dis[v]>dis[u]+w)
			{
				dis[v]=dis[u]+w;
				q.push(mp(dis[v],v));
			}
		}
	}
}
int pre[maxn];
li void solve()
{
	m=read();
	rep(i,0,m)
	{
		int u=read(),v=read(),w=read();
		add(u,v,w),add(v,u,w);
	}
	dij(1);
	ll res=1;
	rep(u,1,n+1)
		for(int i=head[u];i;i=edge[i].nx)
		{
			int v=edge[i].v;
			int w=edge[i].w;
			if(dis[v]==dis[u]+w) pre[v]++;
		}
	cout<<inf<<endl;
	rep(i,1,n+1)
	    if(pre[i]) res=res*pre[i]%inf;
	write(res);
	//puts("");
}
int main()
{
	//srand(time(0));
	//freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
	//for(int QwQ=read();QwQ;QwQ--) solve();
	while(~scanf("%d",&n)) solve();
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值