P3387 【模板】缩点

【模板】缩点

题目描述

给定一个 $n$ 个点 $m$ 条边有向图，每个点有一个权值，求一条路径，使路径经过的点权值之和最大。你只需要求出这个权值和。

允许多次经过一条边或者一个点，但是，重复经过的点，权值只计算一次。

输入格式

第一行两个正整数 $n,m$

第二行 $n$ 个整数，其中第 $i$ 个数 $a_i$ 表示点 $i$ 的点权。

第三至 $m+2$ 行，每行两个整数 $u,v$，表示一条 $u\to v$ 的有向边。

输出格式

共一行，最大的点权之和。

样例 #1

样例输入 #1

2 2
1 1
1 2
2 1

样例输出 #1

2

提示

对于 $100\%$ 的数据，$1\le n\le 10^4$，$1\le m\le 10^5$，$0\le a_i\le 10^3$。

题解：

缩点，具体步骤就是tarjan判环后，把同一环内的点全都标记为同一数字，然后再把所有边里两端不在同一数字的加入新图里就行了。

/*keep on going and never give up*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ll long long
#define endl "\n"
#define fast std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const double E = exp(1);
const double PI = acos(-1.0);
const int mod=1e9+7; 
const int maxn=2e5+10;
int cnt1,x,y,in[maxn],n,m,vis[maxn],dis[maxn],w[maxn];
int dfn[maxn],low[maxn],ans[maxn],ts,num[maxn],cnt;
vector <int> cb[maxn],rdr[maxn];
struct node{
    int to,nxt,x;
}e[maxn];int head[maxn],tot;
void add(int x,int y){ e[++tot]={y,head[x],x};head[x]=tot;}
void topsort(){
	queue<int>q;
	for(int i=1;i<=tot;i++){
        if(in[i]==0)
        q.push(i);
    }
    while(!q.empty()){
        int u=q.front();
        q.pop();
        ans[++cnt]=u;
        for(int i=0;i<cb[u].size();i++){
            int v=cb[u][i]; in[v]--;
            if(in[v]==0) q.push(v);
        }
    }
}
stack<int>s;
void tarjan(int x){
    dfn[x]=low[x]=++ts;
    s.push(x);
    vis[x]=1;
   	for(int i=head[x];i;i=e[i].nxt){
        if(!dfn[e[i].to]){
           	tarjan(e[i].to);
            low[x]=min(low[x],low[e[i].to]);
       	}
       	else if(vis[e[i].to])
        low[x]=min(low[x],dfn[e[i].to]);
    }
    if(low[x]==dfn[x]){
   		cnt1++;
   		while(1){
       		num[s.top()]=cnt1;	 
       		dis[cnt1]+=w[s.top()];
       		vis[s.top()]=0;
       		if(x==s.top()){
       			s.pop();
       			break;
       		}
			s.pop();
       	}// 缩点时具体统计什么信息 
    }
}
int f[maxn];
signed main(){
    fast
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>w[i];
    for(int i=1;i<=m;i++){
        cin>>x>>y;add(x,y);
    }
    for(int i=1;i<=n;i++) if(!dfn[i])tarjan(i);
    for(int i=1;i<=tot;i++){
        if(num[e[i].x]!=num[e[i].to]){
            x=num[e[i].x];y=num[e[i].to];
			in[y]++;cb[x].push_back(y);rdr[y].push_back(x);
        }
    }
    topsort();
    for(int i=1;i<=cnt;i++){
        int u=ans[i];
        f[u]=dis[u];
        for(int j=0;j<rdr[u].size();j++)
        f[u]=max(f[u],f[rdr[u][j]]+dis[u]);
    }
//    for(int i=1;i<=n;i++) cout<<dis[i]<<" ";	
	int sum=-1;
    for(int i=1;i<=cnt;i++) sum=max(f[i],sum);
    cout<<sum; 
}