题目链接:Loppinha, the boy who likes sopinha
题意:给你一个01串,然后连续的x个1需要消耗x*(x+1)/2个能量,现在你可以将一些1变成0,问你最少变几个1使得小号的能量小于给定的能量。
思路:
- 首先想到的是贪心,每次都找最长的串然后劈成两半,显然这样不是最优的,比如11111
,总能量是3,如果贪心的话就是先变中间的,然后变成11011,然后还需要再变两个1,但是最好的方法其实只需要变成10101的样子,这样只需要两次操作。 - 然后就是记忆化搜索dp[i][j]表示当前处理完了前i个数字,还剩下j次换的机会,后面(n−i)个数字还需要消耗的能量。
参考博客:E - Loppinha, the boy who likes sopinha(记忆化搜索),总感觉自己写的太苍白,还是贴上大佬的博客吧。
然后是大佬的code
#include <bits/stdc++.h>
#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define lson l,mid,rt<<1
#define rson mid+1,r,(rt<<1)+1
#define CLR(x,y) memset((x),y,sizeof(x))
#define fuck(x) cerr << #x << "=" << x << endl
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int seed = 131;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int n, k;
int a[maxn];
int dp[505][505];//dp[i][j]表示处理了前i个数字,然后还可以换j次,i个数字后还所需要消耗的能量
int cal(int num) {
return num * (num + 1) / 2;
}
//递归边界是刚好处理完n个数字,并且要求k>=0
int dfs(int p, int k) {//前p个数字,换k次
if (k < 0) return INF;
if (p == n) return 0;
if (dp[p][k] != -1) return dp[p][k];
if (a[p] == 0) return dfs(p + 1, k);//为0就肯定不用换,继续递归
int i = 0;
dp[p][k] = INF;//先初始化为INF
for (i = p; i < n && a[i] == 1; i++) {//当有一个位置为0的时候跳出循环
dp[p][k] = min(dp[p][k], dfs(i + 1, k - 1) + cal(i - p));//考虑将第i个1换成0
}
dp[p][k] = min(dp[p][k], dfs(i + 1, k) + cal(i - p));//第i位置是0
return dp[p][k];
}
int main() {
scanf("%d%d", &n, &k);
for (int i = 0; i < n; i++) scanf("%1d", &a[i]);
CLR(dp, -1);
for (int i = 0; i <= n; i++) {
if (dfs(0, i) <= k) {
printf("%d\n", i);
return 0;
}
}
return 0;
}