【Tarjan】【模板】缩点

题目链接

题目背景

缩点+DP

题目描述

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

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

输入

第一行，n,m

第二行，n个整数，依次代表点权

第三至m+2行，每行两个整数u,v，表示u->v有一条有向边

输出

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

输入样例

2 2
1 1
1 2
2 1

输出样例

2

说明

n<=10^4,m<=10^5,点权<=1000

算法：Tarjan缩点+DAGdp

题解

说多了都是泪啊。。。自学敲了三个多小时，最后还是在大佬的帮助下发现了bug。

代码

#include<bits/stdc++.h>
using namespace std;
const int M = 100100;

struct edge {
	int to, next, from;
}e[M * 2], ed[M * 2];

int n, m;
int k, tot, top, s;

int p[M], head[M], sd[M];
int dfn[M], low[M];
int stac[M], vis[M];
int h[M], in[M], dist[M];

void insert1(int u, int v) {
	e[++k].next = head[u];
	e[k].from = u;
	e[k].to = v;
	head[u] = k;
}

void tarjan(int x) {
	dfn[x] = low[x] = ++tot;
	stac[++top] = x;
	vis[x] = 1;
	
	for (int i = head[x]; i; i = e[i].next) {
		int v = e[i].to;
		if (!dfn[v]) {   //has never been
			tarjan(v);
			low[x] = min(low[x], low[v]);
		}else if(vis[v]){
			low[x] = min(low[x], dfn[v]);
		}
	}
	
	if (dfn[x] == low[x]) {    //时间戳和次序号相等 
		int y;
		while (y = stac[top--]) {
			sd[y] = x;
			vis[y] = 0;
			if (x == y) break;
			p[x] += p[y];
		}
	}
}

void topp() {
	queue <int> q;
	int tmp = 0;
	for (int i = 1; i <= n; i++) {
		if (sd[i] == i && !in[i]) {
			q.push(i);
			dist[i] = p[i];
		}
	}
	while (!q.empty()) {
		int kk = q.front();
		q.pop();
		for (int i = h[kk]; i; i = ed[i].next) {
			int v = ed[i].to;
			dist[v] = max(dist[v], dist[kk] + p[v]);
			in[v]--;
			if (!in[v]) {
				q.push(v);
			}
		}
	}
	int ans = 0;
	for (int i = 1; i <= n; i++) {
		ans = max(ans, dist[i]);
	}
	cout << ans;
}

int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> p[i];
	}
	for (int i = 1; i <= m; i++) {
		int u, v;
		cin >> u >> v;
		insert1(u, v);
	}
	for (int i = 1; i <= n; i++) {
		if (!dfn[i]) tarjan(i);
	}
	for (int i = 1; i <= m; i++) {
		int x = sd[e[i].from], y = sd[e[i].to];
		if (x != y) {
			ed[++s].next = h[x];
			ed[s].to = y;
			ed[s].from = x;
			h[x] = s;
			in[y]++;
		}
	}
	topp();
	return 0;
}