Tarjan WA+MLE 50 求助
查看原帖
Tarjan WA+MLE 50 求助
556362
Unnamed114514楼主2022/4/9 14:03
#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0;
	char ch=getchar();
	while(ch<'0'||ch>'9')
		ch=getchar();
	while(ch>='0'&&ch<='9'){
		res=(res<<1)+(res<<3)+(ch^'0');
		ch=getchar();
	}
	return res;
}
const int maxn=1e4+5;
int T,n,m,num,tot,dfn[maxn],low[maxn],vis[maxn],ans,siz[maxn],dp[maxn],a[maxn];
vector<int> G[maxn];
stack<int> s;
void Tarjan(int u){
	dfn[u]=low[u]=++num;
	s.push(u);
	for(int i=0,len=G[u].size();i<len;++i){
		int v=G[u][i];
		if(!dfn[v]){
			Tarjan(v);
			low[u]=min(low[u],low[v]);
		} else if(!vis[v])
			low[u]=min(low[u],dfn[v]);
	}
	if(low[u]==dfn[u]){
		vis[u]=++tot;
		siz[tot]+=a[u];
		while(s.top()!=u){
			vis[s.top()]=tot;
			siz[tot]+=a[s.top()];
			s.pop();
		}
		s.pop();
	}
}
int dfs(int u){
    if(~dp[u])
		return dp[u];
    int add=0;
    for(int i=1;i<=n;i++)
        if(vis[i]==u)
            for(int j=0,len=G[u].size();j<len;++j){
                int v=G[u][j];
                if(u!=vis[v])
               	 	add=max(add,dfs(vis[v]));
            }
    return dp[u]=add+siz[u];
}
int main(){
	memset(dp,-1,sizeof(dp));
	n=read(),m=read();
	for(int i=1;i<=n;++i)
		a[i]=read();
	for(int i=1;i<=m;++i){
		int u=read(),v=read();
		G[u].push_back(v);
	}
	for(int i=1;i<=n;++i)
		if(!dfn[i])
			Tarjan(i);
	for(int i=1;i<=tot;++i)
		ans=max(ans,dfs(i));
	printf("%d\n",ans);
	return 0;
}
2022/4/9 14:03
加载中...