1179: [Apio2009]Atm

6000多MS,差点以为要TLE了。

果然SPFA不稳定性太强了(好像有个随机双端队列的方法防数据卡spfa的,不过我懒。。。。。)

因为一个路口可以经过多次,所以显然裸的spfa是不可取的。

考虑tarjan缩点,然后另建一个图(DAG),spfa乱搞就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
using namespace std;
const int N=500000+10;
struct Edge{int to,next;}e[N],g[N];
int head[N],cnt,last[N],sz;
int val[N],d[N],v[N];
int sccno[N],scc_cnt,lowlink[N],pre[N],dfs_clock;
stack<int>s;
bool ispub[N],inq[N];
void ins(int u,int v){
	e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
}
void insert(int u,int v){
	g[++sz].to=v;g[sz].next=last[u];last[u]=sz;
}
void tarjan(int u){
	pre[u]=lowlink[u]=++dfs_clock;
	s.push(u);
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(!pre[v]){
			tarjan(v);
			lowlink[u]=min(lowlink[u],lowlink[v]);
		}else if(!sccno[v])lowlink[u]=min(lowlink[u],pre[v]);
	}
	if(pre[u]==lowlink[u]){
		scc_cnt++;
		while(true){
			int x=s.top();s.pop();
			sccno[x]=scc_cnt;
			if(x==u)break;
		}
	}
}
void spfa(int S){
	memset(d,-1,sizeof(d));
	d[S]=v[S];queue<int>q;q.push(S);
	while(!q.empty()){
		int u=q.front();q.pop();inq[u]=false;
		for(int i=last[u];i;i=g[i].next){
			int t=g[i].to;
			if(d[t]<d[u]+v[t]){
				d[t]=d[u]+v[t];
				if(!inq[t]){inq[t]=true;q.push(t);}
			}
		}
	}
}
int main(){
	int n,m;scanf("%d%d",&n,&m);
	int u,t;
	for(int i=1;i<=m;i++){
		scanf("%d%d",&u,&t);
		ins(u,t);
	}
	for(int i=1;i<=n;i++)
	scanf("%d",&val[i]);
	int S,P;scanf("%d%d",&S,&P);
	for(int i=1;i<=P;i++){
		scanf("%d",&u);ispub[u]=true;
	}
	for(int i=1;i<=n;i++)if(!pre[i])tarjan(i);
	for(int i=1;i<=n;i++)v[sccno[i]]+=val[i];
	for(int u=1;u<=n;u++)
	for(int i=head[u];i;i=e[i].next)
	if(sccno[e[i].to]!=sccno[u])insert(sccno[u],sccno[e[i].to]);
	spfa(sccno[S]);
	int ans=0;
	for(int i=1;i<=n;i++)
	if(ispub[i])ans=max(ans,d[sccno[i]]);
	printf("%d",ans);
	return 0;
}


你可能感兴趣的:(1179: [Apio2009]Atm)