洛谷 P2072 宗教问题 的题解

这又是一道 dp 题。咋眼一看，题目需要输出两个东西，那就两个 dp 呗。

我们建立两个 dp 数组 $tea{m}_i$ 和 $hazar{d}_i$ 分别表示前 $i$ 个教徒至少要 分几个集体 和 危险值总和。

转移方程为：

• $tea{m}_i=\min (tea{m}_i,tea{m}_{j-1}+1)$
• $hazar{d}_i=\min (hazar{d}_i,hazar{d}_{j-1}+num)$，其中 $num$ 表示在 $[j,i]$ 之间有几个有不同信仰的教徒。

代码如下（含注释）：

#include <bits/stdc++.h>
using namespace std;
int n, m, k, cnt = 0;
int a[1007], c[1007];
int team[1007], hazard[1007];
int main() {
	cin >> n >> m >> k;
	for(int i = 1; i <= n; i++)
		cin >> a[i];
	memset(team, 0x3f, sizeof(team));
	memset(hazard, 0x3f, sizeof(hazard));
	team[1] = a[1], hazard[1] = 1;
	team[0] = 0, hazard[0] = 0;
	for(int i = 1; i <= n; i++) { //遍历 n 个教徒 
		memset(c, 0, sizeof(c));
		cnt = 0;
		for(int j = i; j >= 1; j--) {
		//枚举前 i 个教徒的最后一个集体是否满足题目条件 
			if(c[a[j]] == 0) {
				c[a[j]] = 1;
				cnt++;
			}
			if(cnt > k) break;
			team[i] = min(team[i], team[j - 1] + 1);
			hazard[i] = min(hazard[i], hazard[j - 1] + cnt);
		}
	}
	cout << team[n] << endl << hazard[n];
	return 0;
}