题目背景
缩点+DP
题目描述
给定一个n个点m条边有向图,每个点有一个权值,求一条路径,使路径经过的点权值之和最大。你只需要求出这个权值和。
允许多次经过一条边或者一个点,但是,重复经过的点,权值只计算一次。
输入输出格式
输入格式:
第一行,n,m
第二行,n个整数,依次代表点权
第三至m+2行,每行两个整数u,v,表示u->v有一条有向边
输出格式:
共一行,最大的点权之和。
输入输出样例
输入样例#1:
2 2
1 1
1 2
2 1
输出样例#1:
2
说明
n<=10^4,m<=10^5,|点权|<=1000 算法:Tarjan缩点+DAGdp
代码
#include<bits/stdc++.h>
using namespace std;
#define maxn 1000005
int n,num_e,dfs_T,ans,m,u,vv,S=0,x,y,scc_num;
int b[maxn];
struct edge{
int to,nex,fa;
}e[maxn];
int low[maxn],head[maxn],bj[maxn],val[maxn];
int v[maxn],dp[maxn];
int sceno[maxn],pre[maxn];
stack<int> st;
void add_edge(int x,int y){
e[++num_e].to=y;
e[num_e].fa=x;
e[num_e].nex=head[x];
head[x]=num_e;
}
void scc(int x){
low[x]=pre[x]=++dfs_T;
st.push(x);
for(int o=head[x];o;o=e[o].nex){
if(!pre[e[o].to]) scc(e[o].to),low[x]=min(low[x],low[e[o].to]);
else if(!sceno[e[o].to]) low[x]=min(low[x],pre[e[o].to]);
}
if(low[x]==pre[x]){
scc_num++;
int y;
do{
y=st.top();
st.pop();
sceno[y]=scc_num;
v[scc_num]+=val[y];
}while(x^y);
}
}
void dfs(int x){
b[x]=1;
for(int i=head[x];i;i=e[i].nex){
int y=e[i].to;
if(!b[y] && x!=y) dfs(y),dp[x]=max(dp[x],dp[y]+v[y]);
}
b[x]=0;
}
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++){
scanf("%d%d",&x,&y);
add_edge(x,y);
}
for(int i=1;i<=n;i++){
if(!pre[i]) scc(i),add_edge(S,i);
}
memset(head,0,sizeof(head));
for(int i=1;i<=num_e;i++){
e[i].fa=sceno[e[i].fa];
e[i].to=sceno[e[i].to];
e[i].nex=head[e[i].fa];
head[e[i].fa]=i;
}
dfs(S);
printf("%d",dp[S]);
return 0;
}