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; }