【动态规划】【模型】【contest1】E

思想升华：

我们写转移方程时，我们要明确：
1.f的含义
（一般是到i处的满足要求最佳情况，但此题的双边问题则是只满足一边最佳）
2.f的基础转移状态+其他可能的转移状态

题目

contest1
CodeForces - 1353E

题解

要求：（用灯的开关模拟）

求使得n个灯，亮着的（1）必须相隔为i和i+k的关系 的 最小动开关的次数

思路：

多画几下就会发现，原序列被分成了k个子序列（如子序列1:1,1+k,1+2k…1+ck）,要求只能有一个子序列有连续的1,其它的序列都为0
代价的前缀和以方便求区间代价，这也很牛
利用前缀和求反向前缀和：pre[n]-pre[i+1];//比较妙

难点：

转移方程：

难点一：
要保证亮着的地方是连续的
-> 要么只能从上一个亮着的地方过来/要么前面全部黑掉
难点二：
中间连续就好，右边也可以没有
->（对f的含义产生干扰：f只定义成f处亮着，那右边就没有变暗的地方）
->动态转移方程的f[i]都是在i处可且从1到i的情况（右边管不了）
->可以用两个方向相反的数组合起来解决这一问题
难点三：
左右分别统计最后又聚集在一起，会有那种算重复或者算少了的情况，需要注意（本次代码中的值得一看）

代码

我的代码

#include<iostream>
#include<string>
using namespace std;
const int N = 1e6+10;
int pre[N], f[N], g[N];
int main()
{
    //freopen("C:\\Users\\bearb\\Desktop\\in.txt","r", stdin);
    int t;
    cin>>t;
    while(t--)
    {
        //cin
        int n, k;
        string s;
        cin>>n>>k>>s;
        //init
        pre[0] = 0;
        for(int i=1; i<= n; i++)
        {
            pre[i] = pre[i-1]+s[i-1]-'0';
        }
        //particular
        if(pre[n]==0)
        {
            cout<<0<<endl;
            continue;
        }
        //operate f
        for(int i = 1;i <= k; i++)
        {
            f[i] = pre[i-1];
            if(s[i-1]=='0') f[i]++;
        }
        for(int i = k+1; i <= n; i++)
        {
            f[i] = min(pre[i-1], f[i-k]+pre[i-1]-pre[i-k]);
            if(s[i-1]=='0') f[i]++;
        }
        //operate g
        for(int i = n;i >= n-k+1; i--)
        {
            g[i] = pre[n]-pre[i];//比较妙
            if(s[i-1]=='0') g[i]++;
        }
        for(int i = n-k; i >= 1; i--)
        {
            g[i] = min(pre[n]-pre[i], g[i+k]+pre[i+k-1]-pre[i]);
            if(s[i-1]=='0') g[i]++;
        }
        //calculate answer
        int ans = N;
        for(int i = 1; i <= n; i++)
        {
            int tmp = f[i]+g[i];
            if(s[i-1]=='0') tmp--;
            ans = min(ans, tmp);
        }
        cout << ans<<endl;
    }
    return 0;
}

#include <iostream>
using namespace std;
const int N = 1e6 + 10;
int t, n, k;
int pre[N], f[N], g[N];
char s[N];
int main()
{
    freopen("C:\\Users\\bearb\\Desktop\\in.txt", "r", stdin);
    int i, ans, x;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &k);
        scanf("%s", s);
        for (i = 0; i < n; i++)// 计算"将i+1前的0~i全部赋值成0的代价"
            pre[i+1] = pre[i] + s[i] - '0';
        if (pre[n] == 0) {// 若全为0，输出0(后面的算法再怎么都会将至少一个地方赋值成1,无法排除这种情况)
            printf("0\n");
            continue;
        }
        // 递推计算，比较若使i为打开状态
        // i点以前的灯全灭需要的开关次数pre[i - 1]
        // 和按照k间隔打开的开关次数
        for (i = 1; i <= n; i++) {//f[i]:i打开满足题意的最小代价
            f[i] = pre[i-1];//"将前面全部赋值为0"是一种大家都有的可能最小代价，将其作为初值
            if (i - k > 0)// "非前面有k个元素" 还可以从 "i-k处灯也亮着" 的情况递推
                f[i] = min(f[i], f[i-k] + pre[i-1] - pre[i-k-1]);
            if (s[i - 1] == '0')
                f[i]++;
        }
        // 递推计算，比较i点以后的灯全灭需要的开关次数
        // 和按照k间隔打开的开关次数
        for (i = n; i >= 1; i--) {
            g[i] = pre[n] - pre[i];
            if (i + k <= n)
                g[i] = min(g[i], g[i+k] + pre[i+k-1] - pre[i]);
            if (s[i - 1] == '0')
                g[i]++;
        }
        //  计算最终结果
        ans = n + 1;
        for (i = 1; i <= n; i++) {
            x = f[i] + g[i];
            //cout << f[i]<<":"<<g[i]<<endl;
            // 去除重复计算
            if (s[i-1] == '0')
                x--;
            if (x < ans)
                ans = x;
        }
        printf("%d\n", ans);
    }
    return 0;
}