Kosaraju 60pts 求助qwq
查看原帖
Kosaraju 60pts 求助qwq
291541
EndlIa楼主2021/9/24 13:03
#include <iostream>
#include <algorithm>
#include <stack>
using std::cin;
using std::cout;
using std::max;
using std::ios;
using std::stack;
const int MAXN=1e4+10,MAXM=1e5+10,INF=2147483647;
struct edge
{
	int y,next,x;
}e[3*MAXM];
int ltp,lk1[MAXN],lk2[MAXN],lk3[MAXN];
inline void ist1(int x,int y,int z) 
{
	e[++ltp]=(edge){y,lk1[x],z};
	lk1[x]=ltp;
}
inline void ist2(int x,int y) 
{
	e[++ltp]=(edge){y,lk2[x]};
	lk2[x]=ltp;
}
inline void ist3(int x,int y) 
{
	e[++ltp]=(edge){y,lk3[x]};
	lk3[x]=ltp;
}
int n,m,k;
int a[MAXN],c[MAXN],t[MAXN]; //c是强连通编号 t是每个分量的权值和 
stack<int> s;
bool vis[MAXN];
inline void dfs1(int x)
{
	vis[x]=true;
	for(int i=lk1[x];i;i=e[i].next)
	 if(!vis[e[i].y])
	  dfs1(e[i].y);
	s.push(x);
}
inline void dfs2(int x)
{
	c[x]=k;
	t[k]+=a[x];
	for(int i=lk2[x];i;i=e[i].next)
	 if(!c[e[i].y])
	  dfs2(e[i].y);
}
void kosa()
{
	k=0;
	for(int i=1;i<=n;i++)
     if(!vis[i])
	  dfs1(i);
    while(!s.empty()) 
    {
    	if(!c[s.top()])
	    {
           k++;
           dfs2(s.top());
        }
        s.pop();
	}
    for(int i=1;i<=2*m;i+=2)
    {
    	if(c[e[i].x]!=c[e[i].y])
    	 ist3(c[e[i].x],c[e[i].y]);
	}
}
int ans=0,f[MAXN];
void dp(int x)
{
	if(f[x])
	 return;
	for(int i=lk3[x];i;i=e[i].next)
	{
		if(!f[e[i].y])
		 dp(e[i].y);
		f[x]=max(f[x],t[e[i].y]);
	}
	f[x]+=t[x];
	ans=max(ans,f[x]);
}
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	 cin>>a[i];
	int a,b;
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b;
		ist1(a,b,a);
		ist2(b,a);
	}
	kosa();
	for(int i=1;i<=k;i++)
	 if(!f[i])
	  dp(i);
	cout<<ans;
return 0;
}
2021/9/24 13:03
加载中...