POJ 3160 Father Christmas flymouse(강연통+DP)

2580 단어

POJ 3160 Father Christmas flymouse


제목 링크
제목: 유방향도를 정하고 점마다 권한이 있다(마이너스일 수 있다). 지금은 경로를 요구한다. 점마다 권한을 얻거나 얻을 수 있다. 한 점에서 최대 한 번, 최대 값을 얻는 길을 묻는다.
사고방식: 그림에 고리가 있기 때문에 먼저 축소한 다음에DAG의 가장 긴 길이다. dp를 만들면 된다.
코드:
#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
#include <algorithm>
#include <stack>
using namespace std;

const int N = 30005;

int n, m, val[N];
vector<int> g[N], scc[N];

#define MP(a,b) make_pair(a,b)
typedef pair<int, int> pii;
map<pii, int> vis;
stack<int> S;

int pre[N], dfn[N], dfs_clock, sccn, sccno[N], scc_val[N];

void dfs_scc(int u) {
	pre[u] = dfn[u] = ++dfs_clock;
	S.push(u);
	for (int i = 0; i < g[u].size(); i++) {
		int v = g[u][i];
		if (!pre[v]) {
			dfs_scc(v);
			dfn[u] = min(dfn[u], dfn[v]);
		} else if (!sccno[v]) dfn[u] = min(dfn[u], pre[v]);
	}
	if (pre[u] == dfn[u]) {
		sccn++;
		int sum = 0;
		while (1) {
			int x = S.top(); S.pop();
			if (val[x] > 0)
				sum += val[x];
			sccno[x] = sccn;
			if (x == u) break;
		}
		scc_val[sccn] = sum;
	}
}

void find_scc() {
	dfs_clock = sccn = 0;
	memset(pre, 0, sizeof(pre));
	memset(sccno, 0, sizeof(sccno));
	for (int i = 0; i < n; i++)
		if (!pre[i]) dfs_scc(i);
}

int dp[N];

int dfs(int u) {
	if (dp[u] != -1) return dp[u];
	dp[u] = 0;
	for (int i = 0; i < scc[u].size(); i++) {
		int v = scc[u][i];
		dp[u] = max(dp[u], dfs(v));
	}
	dp[u] += scc_val[u];
	return dp[u];
}

int main() {
	while (~scanf("%d%d", &n, &m)) {
		vis.clear();
		for (int i = 0; i < n; i++) {
			scanf("%d", &val[i]);
			g[i].clear();
		}
		int u, v;
		while (m--) {
			scanf("%d%d", &u, &v);
			g[u].push_back(v);
		}
		find_scc();
		for (int i = 1; i <= sccn; i++) scc[i].clear();
		for (int u = 0; u < n; u++) {
			for (int i = 0; i < g[u].size(); i++) {
				int v = g[u][i];
				pii tmp = MP(u, v);
				if (sccno[u] == sccno[v] || vis.count(tmp)) continue;
				vis[tmp] = 1;
				scc[sccno[u]].push_back(sccno[v]);
			}
		}
		int ans = 0;
		memset(dp, -1, sizeof(dp));
		for (int i = 1; i <= sccn; i++)
			ans = max(ans, dfs(i));
		printf("%d
", ans); } return 0; }

좋은 웹페이지 즐겨찾기