【悬关】Tarjan+拓扑+DP 87pts WA on #3 #11 求调
查看原帖
【悬关】Tarjan+拓扑+DP 87pts WA on #3 #11 求调
804607
rainygame楼主2023/8/25 20:59

思路:

先 Tarjan 缩点,然后再用拓扑上 DP 求出最长路。最后找出所酒馆所在的 SCC 并更新答案。

代码:

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define MAXN 500001

int n, m, s, p, u, v, cnt, ssum, ans;
int a[MAXN], f[MAXN], us[MAXN], vs[MAXN], in[MAXN];
int dfn[MAXN], low[MAXN], scc[MAXN], sum[MAXN];
vector<int> vec;
vector<int> e[MAXN], g[MAXN];
bitset<MAXN> vis;
stack<int> st;
queue<int> que;

void tarjan(int x){
	dfn[x] = low[x] = ++cnt;
	vis.set(x);
	st.push(x);
	for (auto i: e[x]){
		if (!dfn[i]){
			tarjan(i);
			low[x] = min(low[x], low[i]);
		}else if (vis.test(i)) low[x] = min(low[x], dfn[i]);
	}
	
	if (dfn[x] == low[x]){
		scc[x] = ++ssum;
		sum[ssum] = a[x];
		while (st.top() != x){
			scc[st.top()] = ssum;
			sum[ssum] += a[st.top()];
			vis.reset(st.top());
			st.pop();
		}
		st.pop();
		vis.reset(x);
	}
}

void topsort(){
	que.push(scc[s]);
	f[scc[s]] = sum[scc[s]];
	while (!que.empty()){
		u = que.front();
		que.pop();
		
		for (auto i: g[u]){
			f[i] = max(f[i], f[u]+sum[i]);
			if (!(--in[i])) que.push(i);
		}
	}
}

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    
    cin >> n >> m;
    for (int i(1); i<=m; ++i){
    	cin >> us[i] >> vs[i];
    	e[us[i]].push_back(vs[i]);
	}
	for (int i(1); i<=n; ++i) cin >> a[i];
	cin >> s >> p;
	while (p--){
		cin >> u;
		vec.push_back(u);
	}
	tarjan(s);
	
	for (int i(1); i<=m; ++i){
		u = scc[us[i]];
		v = scc[vs[i]];
		if (u != v){
			g[u].push_back(v);
			++in[v];
		}
	}
	
	topsort();
	for (int i: vec) ans = max(ans, f[scc[i]]);
	cout << ans;

    return 0;
}

2023/8/25 20:59
加载中...