【gmoj】 【tarjan】 【拓扑】 【并查集】 QYQ在艾泽拉斯

【gmoj】 【tarjan】 【拓扑】【并查集】 QYQ在艾泽拉斯

题目

---

解题思路

因为可能出现环用tarjan缩点
再建一个新图
用按拓扑序跑一边DP
求出一个连通块中从哪个点跑出来的价值最大
k次闪现可以看作可以跑k+1个连通块
排完序后，取出前k+1大求和即为答案

---

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int N=100010;
queue<int> qd;
struct lzf{
	int x,to,next;
}f[10*N],h[10*N];
int t,n,m,x,y,k,t2,da,tot,cnt;
int rd[N],a[N],p[N],d[N],q[N],zc[N],bj[N],fa[N];
int head[N],head2[N],dfn[N],low[N],ans[N];

void add(int x,int y)
{
	 f[++t].to=y;
	 f[t].x=x;
	 f[t].next=head[x];
	 head[x]=t;
}

void add2(int x,int y)
{
	 h[++t2].to=y;
	 h[t2].next=head2[x];
	 head2[x]=t2;
}

void tarjan(int x)
{
	 dfn[x]=low[x]=++tot; 
	 q[++t]=x;
	 p[x]=1;
	 for (int i=head[x];i;i=f[i].next)
	     if (!dfn[f[i].to])
	     {
	     	tarjan(f[i].to);
	     	low[x]=min(low[x],low[f[i].to]);
		 }
		 else if (p[f[i].to]) low[x]=min(low[x],dfn[f[i].to]);
	 if (dfn[x]==low[x])
	 {
	 	   cnt++; 
	 	   do{
	 	   	   y=q[t--];
	 	   	   p[y]=0;
	 	   	   zc[cnt]+=a[y]; 
	 	   	   bj[y]=cnt;
			}while (y!=x); 
	 }
}

int find(int x)
{
	if (fa[x]==x) return x;
	   else return fa[x]=find(fa[x]);
}

void solve()  
{	
	while (!qd.empty())
	{  
	       int x=qd.front();
	       qd.pop();
	       int fx=find(x);
	       d[x]+=zc[x];
	       ans[fx]=max(ans[fx],d[x]);
		   for (int j=head2[x];j;j=h[j].next)
		   {
		      	rd[h[j].to]--;
		      	d[h[j].to]=max(d[x],d[h[j].to]);
		      	if (!rd[h[j].to]) qd.push(h[j].to);
			}
	}
}

int main()
{
	freopen("azeroth.in","r",stdin);
	freopen("azeroth.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);
	}
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	scanf("%d",&k);
	
	t=0;
	for (int i=1;i<=n;i++)
	    if (!dfn[i])
	       tarjan(i);  //缩点
	for (int i=1;i<=cnt;i++) fa[i]=i; 
	for (int i=1;i<=m;i++)
	{
		x=bj[f[i].x],y=bj[f[i].to];
		if (x==y) continue;
	    int fx=find(fa[x]),fy=find(fa[y]);  //用并查集判断是否在同一个连通块
	    if (fx!=fy) fa[fx]=fy;
	    add2(x,y);  //建新图
	    rd[y]++;  //统计入度
	}
	
	for (int i=1;i<=cnt;i++)
	    if (!rd[i]) qd.push(i);  
	solve();  //拓扑跑DP
	
	sort(ans+1,ans+cnt+1); 
	for (int i=cnt;i>=max(cnt-k,1);i--)
	    da+=ans[i];
	printf("%d",da);
	return 0;
}