acw - 选取数对—— dp

选取数对

---

题意：

给定长度为 n 的数列，要求找到 k 个长度为 m 的不相交区间。
求出所选区间元素总和的最大值。
$1\le m\times k\le n\le 5000，0\le ai\le 10^9$

---

分析：

看了一会没思路，那可能就是 dp 了。又看了下数据范围，没错了。

思路1：
定义状态：
f[i, j, 0]：从前 i 个位置中选出 j 个区间，当前位置不作为区间结尾，所得的总和最大值。
f[i, j, 1]：从前 i 个位置中选出 j 个区间，当前位置作为区间结尾，所得的总和最大值。

状态转移：

• 如果当前位置不作为区间结尾，那么当前位置的不选的状态为上一位置选或者不选的较优值：f[i, j, 0] = max(f[i-1, j, 0], f[i-1, j, 1]);
• 如果当前位置作为选择的区间最后一个位置，那么当前位置选的状态为第 i-m 个位置选或不选的较优值，加上当前区间的贡献：f[i, j, 1] = max(f[i-m, j-1, 0], f[i-m, j-1, 1]) + s[i] - s[i-m]; （s[i]为前缀数组）

那么最终的答案便是，最后一个位置选或不选的较优值：max(f[n, k, 0], f[n, k, 1]);

注意：还需要注意数组越界的情况，因为越界之后出现了莫名其妙的答案。。

Code：

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

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define mem(a,b) memset(a,b,sizeof a)
#define int long long
#define PII pair<int,int>
#define pb push_back
#define fi first
#define se second
#define endl '\n'
map<int,int> mp;

/**/

const int N = 5010, mod = 1e9+7;
int T, n, m;
int a[N], s[N];
int f[N][N][2];

signed main(){
	Ios;
	int k;
	cin>>n>>m>>k;
	
	for(int i=1;i<=n;i++) cin>>a[i], s[i] = s[i-1]+a[i];
	
	for(int i=1;i<=n;i++)
	{
		for(int j=1; j<=k && j<=i/m; j++)
		{
			f[i][j][0] = max(f[i-1][j][0], f[i-1][j][1]);
			if(i>=m) f[i][j][1] = max(f[i-m][j-1][0], f[i-m][j-1][1]) + s[i]-s[i-m];
		}
	}
	cout << max(f[n][k][0], f[n][k][1]);
	
	return 0;
}

思路没问题，但是交上去之后发现 MLE 了。
爆空间了，数组太大了。把最后一维压缩掉试试？

思路2：
发现，每次用 f[i, j] 的 [0], [1] 两种状态时，都是取两种状态的最大值。那么如果直接用 f[i, j] 记录两种状态的最大值可不可行呢？

状态定义：
f[i, j]：从前 i 个位置中，选出 j 个长度为 m 的区间，元素总和的最大值。

状态转移：

• 如果当前位置不作为区间结尾的话，那么当前位置的状态就是上一位置的状态：f[i-1, j]；
• 如果当前位置作为区间结尾的话，那么当前位置的状态就是第 i-m 个位置的状态 + 当前区间的贡献：f[i-m, j-1] + s[i] - s[i-m]；

当前位置的状态为这两种情况的较优值：f[i, j] = max(f[i-1, j], f[i-m, j-1]);
同时，需要注意越界的情况：

if(i>=m) f[i][j] = max(f[i-1][j], f[i-m][j-1] + s[i]-s[i-m]);
else f[i][j] = f[i-1][j];

Code：

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

#define Ios ios::sync_with_stdio(false),cin.tie(0)
#define mem(a,b) memset(a,b,sizeof a)
#define int long long
#define PII pair<int,int>
#define pb push_back
#define fi first
#define se second
#define endl '\n'
map<int,int> mp;

/**/

const int N = 5010, mod = 1e9+7;
int T, n, m;
int a[N], s[N];
int f[N][N];

signed main(){
	Ios;
	int k;
	cin>>n>>m>>k;
	
	for(int i=1;i<=n;i++) cin>>a[i], s[i] = s[i-1]+a[i];
	
	for(int i=1;i<=n;i++)
	{
		for(int j=1; j<=k && j<=i/m; j++)
		{
			if(i>=m) f[i][j] = max(f[i-1][j], f[i-m][j-1] + s[i]-s[i-m]);
			else f[i][j] = f[i-1][j];
		}
	}
	cout << f[n][k];
	
	return 0;
}

这样就可以了。