Tarjan+topsort 50pts WA#1#3#4#5#7求调
查看原帖
Tarjan+topsort 50pts WA#1#3#4#5#7求调
572676
Yexo楼主2023/7/29 17:57

rtrt

#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[10001];
int dfn[10001],low[10001];
vector<int> g[10001];
bool vis[10001];
int sign,top;
int st[10001];
int num;
int p[10001];
int belong[10001];
vector<int> rg[10001];
vector<int> rgb[10001];
int init[10001];
int ans;
void tarjan(int x){
    sign++;
    dfn[x]=low[x]=sign;
    st[++top]=x;
    vis[x]=true;
    for(int i=0;i<g[x].size();i++){
        int y=g[x][i];
        if(!dfn[y]){
            tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(vis[y]){
            low[x]=min(low[x],dfn[y]);
        }
    }
    if(dfn[x]==low[x]){
        num++;
        int t;
        do{
            t=st[top--];
            p[num]+=a[t];
            vis[t]=false;
            belong[t]=num;
        }while(t!=x);
    }
}
void build(){
    for(int i=1;i<=n;i++){
        for(int j=0;j<g[i].size();i++){
            int x=g[i][j];
            if(belong[i]!=belong[x]){
                init[belong[x]]++;
                rg[belong[i]].push_back(belong[x]);
                rgb[belong[x]].push_back(belong[i]);
            }
        }
    }
}
void topsort(){
    queue<int> q;
    for(int i=1;i<=num;i++){
        if(init[i]==0) q.push(i);
    }
    while(!q.empty()){
        int x=q.front();
        q.pop();
        int maxn=0;
        for(int i=0;i<rgb[x].size();i++){
            maxn=max(maxn,p[rgb[x][i]]);
        }
        p[x]+=maxn;
        ans=max(ans,p[x]);
        for(int i=0;i<rg[x].size();i++){
            int y=rg[x][i];
            init[y]--;
            if(!init[y]) q.push(y);
        }
    }
}
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++){
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
    }
    for(int i=1;i<=n;i++){
        if(!dfn[i]){
            tarjan(i);
        }
    }
    // for(int i=1;i<=num;i++) printf("%d ",p[i]);
    build();
    topsort();
    printf("%d",ans);
}
2023/7/29 17:57
加载中...