CSUST银川选拔赛 最大得分 (DP)

题目大意
你有一个长度为 n 的 a 序列，要求你将这个序列分成 k 个部分。

要求：

• 每一个部分都非空，

• 部分与部分不能重叠

• 每一个部分都是一串连续下标的数

每一个部分的分数等于这个部分的 gcd ，输出一个 sum 表示这个大小为 n 的序列，分成 k 部分所获得的最大分数。

思路

读完题的第一想法，n 到 1000，a_i 到 100，k 到 50 ，妥妥的就是想让我 n * 100 * k 的搞。
因为每一部分都是连续的，对于一个新的数，只需要考虑是把他合并到上一部分里还是单独开一个部分就好了。
定义 dp[i][j][k] 表示第 i 个数，在区间gcd等于 j 的第 k 个部分里。
状态转移方程为：

• dp[i][a[i]][k] = max(dp[i][a[i]][k], dp[i-1][j][k-1] + a[i]); // 把 a[i] 放到一个新的部分里
• dp[i][g][k] = max(dp[i][g][k], dp[i-1][j][k] - j + g); // 把 a[i] 放到第 k 个部分里， g = gcd(j, a[i])

初始化也很明显，把第一个数放到第一个部分里，即：dp[1][a[1]][1] = a[1];

最后的答案就是 max(dp[n][1…100][k])。

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e4+5;

int a[maxn];
int dp[maxn][105][55];

int main(int argc, char** argv) {
	
	memset(dp, -1, sizeof(dp));	
	
	int n, m;
	scanf("%d %d", &n, &m);
	for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
	
	dp[1][a[1]][1] = a[1];
	for(int i = 2; i <= n; i++){
		for(int j = 1; j <= 100; j++){
			int g = __gcd(a[i], j);
			for(int k = 1; k <= min(i, m); k++){ 	
				if(dp[i-1][j][k-1] != -1){ // 新段 
					dp[i][a[i]][k] = max(dp[i][a[i]][k], dp[i-1][j][k-1] + a[i]); 
				}
				if(dp[i-1][j][k] != -1){ // 合并 
					dp[i][g][k] = max(dp[i][g][k], dp[i-1][j][k] - j + g);
				}
			}
		}
	}
	
	int res = -1;
	for(int i = 1; i <= 100; i++) res = max(res, dp[n][i][m]);
	
	cout << res << endl;
	
	return 0;
}