洛谷：P1453 城市环路（状态机dp）

城市环路

题意：

• 在一个环上建商店，要求没有任意两个商店相邻。
• 线性状态机的拓展。

思路：

• 定义 $f[i][0]、f[i][1]$，含义为：从尾部选到 i 位置，同时 i 位置 选和不选 的 最大利益
• 任取环上一条边，切断。假设两点为 a、b；
• 从边上两点分别 dfs 下去求 f 数组；
• 根据定义最终答案为 $min(f[a][0],f[b][0])$，因为两店相邻，必定有一个是没开的。

$Code:$

#include<bits/stdc++.h>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define endl "\n"
#define xx first
#define yy second
//[博客地址]：https://blog.csdn.net/weixin_51797626?t=1
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;

const int N = 100010, M = 200010, MM = N;
int INF = 0x3f3f3f3f, mod = 998244353;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D;
int h[N], ne[M], e[M], idx;
double d[N], dp[N][2], kk;
int l, r, p[N];

int find(int x) {
	return p[x] == x ? p[x] : p[x] = find(p[x]);
}

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs(int x, int fa) {
	dp[x][1] = d[x] * kk, dp[x][0] = 0;
	for (int i = h[x]; ~i; i = ne[i]) {
		int j = e[i];
		if (j == fa)continue;
		dfs(j, x);
		dp[x][0] += max(dp[j][1], dp[j][0]);
		dp[x][1] += dp[j][0];
	}
}

int main() {

	cin >> n;
	mem(h, -1);
	forr(i, 1, n)sca("%lf", &d[i]), p[i] = i;

	forr(i, 1, n) {
		int a, b;
		sca("%d%d", &a, &b);
		a++, b++;
		if (find(a) == find(b)) { l = a, r = b; continue; }
		//因为边顺序随意给，所以用并查集找断边

		add(a, b), add(b, a), p[find(a)] = find(b);
	}
	cin >> kk;

	double ans;
	dfs(l, -1), ans = dp[l][0];
	dfs(r, -1), ans = max(ans, dp[r][0]);
	pri("%.1f", ans);

	return 0;
}