tarjan+topoWA#40pts求助
查看原帖
tarjan+topoWA#40pts求助
658995
Spring_qwq楼主2023/9/29 11:19

rt

#include<bits/stdc++.h>
#define ri register int
#define ll long long
#pragma G++ optimize(3)
using namespace std;
int t=0,dfn[100010],low[100010],n,m,cnt=0,lt[100010],h[100010],h2[100010],val[100010],root[100010],c2=0,in[100010],dis[100010];
bool vis[100010],bj[100010];
struct node{
	int from,to,next;
}e[100010],e2[100010];
stack<int>z;
void add(int u,int v)
{
	e[++cnt].next=h[u];
	e[cnt].from=u;
	e[cnt].to=v;
	h[u]=cnt;
}
void dfs(int x)
{
	dfn[x]=++t;
	low[x]=t;
	z.push(x);
	vis[x]=1;
	for(int i=h[x];i;i=e[i].next)
	{
		int v=e[i].to;
		if(!dfn[v])
		{
			dfs(v);
			low[x]=min(low[x],low[v]);
		}
		else if(vis[v])
		low[x]=min(low[x],low[v]);
	}
	if(low[x]==dfn[x])
	{
		while(1)
		{
			int y=z.top();
			root[y]=x;
			
			vis[y]=0;
			if(y==x)break;
			z.pop();
			val[x]+=val[y];
		}
	}
}
void topo()
{
	queue<int>z;
	for(int i=1;i<=n;i++)
	{
		if(root[i]==i&&!in[i])
		{
			dis[i]=val[i];
			z.push(i);
		}
	}
	while(!z.empty())
	{
		int k=z.front();
		z.pop();
		for(int i=h2[k];i;i=e2[i].next)
		{
			int v=e2[i].to;
			dis[v]=max(dis[v],dis[k]+val[v]);
			in[v]--;
			if(in[v]==0)z.push(v);
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++)ans=max(ans,dis[i]);
	printf("%d",ans);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&val[i]);
	for(int i=1;i<=m;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);
	}
	for(int i=1;i<=n;i++)if(!dfn[i])dfs(i);
	for(int i=1;i<=m;i++)
	{
		int x=root[e[i].from],y=root[e[i].to];
		if(x!=y)
		{
			e[++c2].next=h[x];
			e[c2].from=x;
			e[c2].to=y;
			h[x]=c2;
			in[y]++;
		}
	}
	topo();
	return 0;
}
2023/9/29 11:19
加载中...