【ybtoj 高效进阶 3.4】【强连通分量】 有向图缩点

【ybtoj 高效进阶 3.4】【强连通分量】 有向图缩点

题目

---

解题思路

先缩点然后DP求出最大点权之和

---

代码

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
queue<int>q;
struct lzf{
	int to,next;
}f[100010],l[100010];
int t,z,x,y,tot,n,m,maxn,a[10010];
int ru[100010],head[100010],head2[100010];
int dis[100010],low[100010],dfn[100010],ins[100010],sum[100010],sk[100010];
void add(int x,int y)
{
	 f[++t].to=y;
	 f[t].next=head[x];
	 head[x]=t;
}
void add2(int x,int y)
{
	 l[++t].to=y;
	 l[t].next=head2[x];
	 head2[x]=t;
}
void tarjan(int x)
{
	 dfn[x]=low[x]=++tot;  //标记时间戳
	 sk[++t]=x;  //入栈
	 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 (!ins[f[i].to]) low[x]=min(low[x],low[f[i].to]);
	 if (dfn[x]==low[x])  //连通块
	 {
	 	ins[x]=++z; 
	 	sum[z]+=a[x];  
	 	while (sk[t]!=x)
	 	{
	 		  sum[z]+=a[sk[t]];
	 		  ins[sk[t--]]=z;
		}
		t--;
	 }
}
void dp()
{
	 for (int i=1;i<=z;i++) 
	 {
	     if (!ru[i]) q.push(i);  //没有入度，直接入队
	     dis[i]=sum[i];
	 }
	 while (!q.empty())
	 {
	 	   int x=q.front();
	 	   q.pop();
	 	   for (int i=head2[x];i;i=l[i].next)
	 	   {
	 	   	   dis[l[i].to]=max(dis[l[i].to],dis[x]+sum[l[i].to]);    //更新最大点劝和
	 	   	   ru[l[i].to]--;
	 	   	   if (!ru[l[i].to]) q.push(l[i].to);
		   } 
	 }
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	    scanf("%d",&a[i]);
	for (int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);
	}
	t=0;
	for (int i=1;i<=n;i++)
	    if (!dfn[i]) tarjan(i);  //缩点
	t=0; 
	for (int i=1;i<=n;i++)  //构建新的邻接表
	    for (int j=head[i];j;j=f[j].next)
	        if (ins[i]!=ins[f[j].to])
	        {
	           add2(ins[i],ins[f[j].to]);  
	           ru[ins[f[j].to]]++; 
			}
	dp();
	for (int i=1;i<=z;i++)
	    maxn=max(maxn,dis[i]);
	printf("%d",maxn);
	return 0;
}