题目大意:
给你n个数,在n个数中从前往后选取几个数,使得选取数的和尽可能的大,但是相邻两次选取的数的下标至少间隔m,(第一次最早选取的数是a[i + m],如果选取了a[i],那么下一次的选取最早是a[i + m])。求最大值。
问题分析:
思路1:
用dp[i]表示当前的得分的最大值,那么我们知道当i <= m的时候,dp[i] = 0;
当i >= m + 1时,这时我们对于这个数可以选也可以不选,如果我们没有选择这个数,dp[i] = dp[i - 1],但是如果我们选择了这个数,因为如果选取了一次,我们至少m次无法选取,dp[i] = dp[i - m] + a[i];
因此我么能得到状态转移方程:
1.dp[i] = 0;(i <= m)
2.dp[i] = max(dp[i - m] + a[i],dp[i - 1]);
代码1:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 300000 + 5;
int a[maxn], dp[maxn];
int n, m;
int main()
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++)
scanf("%d", a + i);
memset(dp, 0, sizeof(dp));
for(int i = 1; i <= n; i++)
{
if(i <= m)
dp[i] = 0;
else
dp[i] = max(dp[i - 1], dp[i - m] + a[i]);
}
printf("%d\n", dp[n]);
return 0;
}
思路2:
我们对于这个问题可以反着考虑,也就是dp[i]表示在i的时候做出选择,到最后能够得到多少分。
同样的我们可以得到如下的状态转移方程:
1.dp[i] = dp[i + 1];(i < m)
2.dp[i] = max(dp[i + 1],dp[i + m] + a[i]);(i >= m)
代码2:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 300000 + 5;
int dp[maxn], a[maxn];
int n, m;
int main()
{
memset(dp, 0, sizeof(dp));
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++)
scanf("%d", a + i);
priority_queue<int, vector<int>, less<int> > q;
for(int i = n; i >= 1; i--)
{
if(i > m)
dp[i] = max(dp[i + 1], dp[i + m] + a[i]);
else
dp[i] = dp[i + 1];
}
printf("%d\n", dp[1]);
return 0;
}