60 pts WA 求调,玄一关
查看原帖
60 pts WA 求调,玄一关
807774
_Wind_Leaves_ShaDow_楼主2024/3/2 15:40

rt,从早上调到现在,评测记录

#include <bits/stdc++.h>
#define fi first
#define se second
#define pb push_back

using namespace std;
typedef pair<int,int> pii;
const int N=2e5,Inf=1e9;

int T,n,m,K,mod,ex[N+5],ey[N+5],ew[N+5],dp[N+5][60];
int dis[N+5][2],ind[N+5],tpp[N+5];
bool vis[N+5];
struct Nodj{
	int ds,ps;
	inline bool operator <(const Nodj &stmp)const{
		return ds>stmp.ds;
	} 
};
vector<pii>v[N+5];
priority_queue<Nodj>dj;
queue<int>tp;

inline void Dijk(int sps,bool op){
	for(int i=1;i<=n;i++){
		dis[i][op]=((i^sps)?Inf:0);
		vis[i]=0;
	}
	dj.push({0,sps});
	while(!dj.empty()){
		int fq=dj.top().ps;
		dj.pop();
		if(vis[fq])continue;
		vis[fq]=1;
		for(pii i:v[fq]){
			if(dis[i.fi][op]<=dis[fq][op]+i.se)continue;
			dis[i.fi][op]=dis[fq][op]+i.se;
			if(!vis[i.fi])dj.push({dis[i.fi][op],i.fi});
		}
	}
	return;
}

inline void Topo(){
	int ppt=0;
	tp.push(1);
	while(!tp.empty()){
		int fq=tp.front();
		tp.pop();
		tpp[++ppt]=fq;
		for(pii i:v[fq]){
			ind[i.fi]--;
			if(!ind[i.fi])tp.push(i.fi);
		}
	}
	return;
}

inline void solve(){
	cin>>n>>m>>K>>mod;
	for(int i=1;i<=n;i++)v[i].resize(0);
	for(int i=1;i<=m;i++){
		cin>>ex[i]>>ey[i]>>ew[i];
		v[ex[i]].pb({ey[i],ew[i]});
	}
	Dijk(1,0);
	for(int i=1;i<=n;i++)v[i].resize(0);
	for(int i=1;i<=m;i++)v[ey[i]].pb({ex[i],ew[i]});
	Dijk(n,1);
	for(int i=1;i<=n;i++){
		v[i].resize(0);
		ind[i]=0;
	}
	for(int i=1;i<=m;i++){
		if(dis[ex[i]][0]+ew[i]==dis[ey[i]][0]&&dis[ex[i]][0]+ew[i]+dis[ey[i]][1]<=dis[n][0]+K){
			v[ex[i]].pb({ey[i],0});
			ind[ey[i]]++;
		}
		ew[i]=dis[ex[i]][0]+ew[i]-dis[ey[i]][0];
	}
	Topo();
	for(int i=1;i<=n;i++){
		if(!ind[i])continue;
		cout<<"-1\n";
		return;
	}
	for(int i=1;i<=n;i++)for(int j=0;j<=K;j++)dp[i][j]=(i==1&&!j);
	for(int i=0;i<=K;i++){
		for(int j=1;j<=n;j++)for(pii k:v[tpp[j]])dp[k.fi][i]=(dp[k.fi][i]+dp[tpp[j]][i])%mod;
		for(int j=1;j<=m;j++)if(i+ew[j]<=K&&ew[j])dp[ey[j]][i+ew[j]]=(dp[ey[j]][i+ew[j]]+dp[ex[j]][i])%mod;
	}
	int AnS=0;
	for(int i=0;i<=K;i++)AnS=(AnS+dp[n][i])%mod;
	cout<<AnS<<'\n';
	return;
}

signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>T;
	while(T--)solve();
	return 0;
} 

关注用小号给。

2024/3/2 15:40
加载中...