CF 327 A Flipping Game

题意: 给定长度为n的整数序列 a[i], 包含值0和1, 现在可以进行一次如下操作: 翻转区间, [i, j]之间的所有数, 即1变成0, 0变成1. 求解这样的区间, 使得最后序列包含的1是最多的.

思路: 很容易想到动态规划. dp[i, j] 表示翻转区间 [i, j]后, 数字1的增量. 假设cnt是原有数列1的总数, 则翻转[i, j]区间后, 序列中1的总数为 cnt + dp[i][j].

初始化时, 若a[i] == 1, 则dp[i][i] = -1, 否则 dp[i][i] = 1.

递归函数如下;

若a[i] == 1, 则dp[i][j] = dp[i][j-1] + 1;

否则, dp[i][j] = dp[i][j-1] - 1;

代码如下:

#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 105
using namespace std;
int n, maxval, maxi, maxj, cnt;
int a[MAXN];
int dp[MAXN][MAXN]; // incremental

void init()
{
    memset(dp, 0, sizeof(dp));
    cnt = 0;
    for(int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
        cnt += a[i];
        if(a[i])
            dp[i][i] = -1;
        else
            dp[i][i] = 1;
    }
}

int main()
{
    while (~scanf("%d", &n))
    {
        init();

        if(n == 1)
        {
            printf("%d\n", 1-a[0]);
            continue;
        }

        for(int i = 0; i < (n-1); i++)
        {
//            printf("%d %d : %d\n", i, i, dp[i][i]);
            for(int j = i+1; j < n; j++)
            {
                if(a[j] == 0)
                    dp[i][j] = dp[i][j-1] + 1;
                else
                    dp[i][j] = dp[i][j-1] - 1;
//                printf("%d %d : %d\n", i, j, dp[i][j]);
            }
        }
        // find max score
        maxi = maxj = 0;
        maxval = dp[0][1];
        for(int i = 0; i < n; i++)
        {
            for(int j = i; j < n; j++)
            {
//                printf("%d %d : %d\n", i, j, dp[i][j]);
                if(dp[i][j] > maxval)
                {
                    maxi = i;
                    maxj = j;
                    maxval = dp[i][j];
                }

            }
        }
//        printf("%d %d %d\n", maxi, maxj, (cnt + maxval));
        printf("%d\n", (cnt + maxval));
    }
    return 0;
}