C. 连锁商店-CCPC女生赛-状压DP、Floyd

题目链接

题目大意：

有$n$个旅游景点，每个景点属于一个公司，在经过某一个景点的时候会得到该公司的红包，当得到该公司的红包之后就不能再获得该公司的红包，问从$1$号景点出发，到$[1,n]$景点可获得的红包最大值是多少。

题目分析：

如果一个公司只有一个旅游景点，那么该公司的红包可以直接获得，如果一个公司有两个以上的旅游景点，那么我们可以使用状压dp进行求解，$dp[i][st]$代表到$i$号景点，经过的公司状态为$st$时，可以获得的最大红包值。需要注意初始化时分类讨论起点的公司是否还有其它景点，若没有，初始化起点为0状态，反之初始化为$1<<rk[c[i]]$.

AC代码：

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	int n, m, tot = 0;
	cin >> n >> m;
	vector<int> c(n + 1), w(n + 1), cnt(n + 1), c2(n + 1), book(n + 1), rk(n + 1);
	vector<vector<int> > g(n + 1, vector<int>(n + 1, 0)), dp(n + 1, vector<int>(1 << (n / 2 + 1), 0)), ve(n + 1);
	for(int i = 1;i <= n;i++) {
		cin >> c[i];
		ve[c[i]].push_back(i);
	}
	for(int i = 1;i <= n;i++) {
		if(ve[i].empty()) continue;
		if(ve[i].size() == 1) book[ve[i][0]] = 1;
		else {
			rk[i] = tot;
			tot++;
		}
	}
	for(int i = 1;i <= n;i++) cin >> w[i];
	for(int i = 1;i <= m;i++) {
		int u, v;
		cin >> u >> v;
		g[u][v] = 1;
	}
	for(int k = 1; k <= n; k++) {
		for(int i = 1; i <= n; i++) {
			for(int j = 1; j <= n; j++) {
				if(i != j) {
					g[i][j] |= (g[i][k] & g[k][j]);
				}
			}
		}
	}
	if(book[1]) dp[1][0] = w[c[1]];
	else dp[1][1 << rk[c[1]]] = w[c[1]];
	for(int u = 1; u <= n; u++) {
		for(int st = 0; st < 1 << tot; st++) {
			for(int v = 1; v <= n; v++) {
				if(!g[u][v]) continue;
				if(book[v]) {
					dp[v][st] = max(dp[v][st], dp[u][st] + w[c[v]]);
				}
				else {
					if(!(st >> rk[c[v]] & 1)) {
						dp[v][st | (1 << rk[c[v]])] = max(dp[v][st | (1 << rk[c[v]])], dp[u][st] + w[c[v]]);
					}
				}
			}
		}
	}
	for(int i = 1;i <= n;i++) cout << *max_element(dp[i].begin(), dp[i].end()) << '\n';

	return 0;
}