NEUQOJ 1495 田鼠看热闹（Ⅰ）(枚举 + 模拟)

1495: 田鼠看热闹（Ⅰ）

时间限制: 1 Sec   内存限制: 128 MB
提交: 56   解决: 16
[ 提交][ 状态][ 讨论版]

题目描述

田鼠闲来无事到集市上偷吃干货，突然被人群吸引住了，他挤进人群中一看，原来他们在玩一个叫做“开心翻转”的游戏，游戏初始如下

有一串只含0和1的数字序列（例如：1  0  1  0  1  1  0 ），长度为n，玩家每次可以选择一个位置进行操作，每次操作可以将指定位置以及指定位置两边的数字翻转（如果两边有数字的话），每次翻转（0变成1,1变成0）需要消耗一个游戏币，如果可以经过K次操作全部翻转成1的话，大喊一声“为了部落！”，如果无论怎么翻转都无法得出答案的话大喊“荣誉属于联盟！”，有解得情况下，消耗最少游戏币的玩家获得胜利，无解的情况下，最快喊出答案的玩家获得胜利，小田鼠看着奖品直流口水，所以想请你帮帮他得出答案。

输入

多组数据输入，每一组输入第一行一个数字 n(1<=n<=10^6)。

接下来的一行输入长度为n的数字序列（只含0,1）。

输出

对于每组输入，如果有解，则输出最少消耗的游戏币，如果不可解，则输出“-1”具体看样例。

样例输入

3
0 0 0
2
1 0

样例输出

1
-1

提示

第一个样例只需要翻转第二个数字即可

第二个样例无论怎么翻转都无法得到全是1的情况

题目链接：http://ncc.neuq.edu.cn/oj/problem.php?id=1495

题目分析：两端是突破口，只影响两种状态，分成5种，第一个翻，第一个不翻，最后一个翻，最后一个不翻，第一个最后一个都翻，因为根据第一个状态递推后面，如果在i点且i-1的值为0那么i点必须翻，因为这是改变i-1的唯一机会，然后五种模拟完判断是否为胜利态，是则记录需要操作的最小次数，1和2要特判一下

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int const MAX = 1e6 + 5;
int const INF = 0x3fffffff;
int a[MAX], b[MAX];

bool judge(int n)
{
    for(int i = 1; i <= n; i++)
        if(b[i] == 0)
            return false;
    return true;
}

int main()
{
    int n;
    while(scanf("%d", &n) != EOF)
    {
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        if(n == 1)
            printf("%d\n", (a[1] == '1') ? 0 : 1);
        else if(n == 2)
        {
            if(a[1] == 1 && a[2] == 1)
                printf("0\n");
            else if(a[1] == 0 && a[2] == 0)
                printf("1\n");
            else 
                printf("-1\n");
        }
        else
        {
            memcpy(b + 1, a + 1, sizeof(a));
            b[1] = !b[1];
            b[2] = !b[2];
            int ans = INF, num = 1;
            for(int i = 2; i <= n; i++)
            {
                if(b[i - 1] == 0)
                {
                    num ++;
                    b[i - 1] = !b[i - 1];
                    b[i] = !b[i];
                    b[i + 1] = !b[i + 1];
                }
            }
            if(judge(n))
                ans = min(ans, num);
            memcpy(b + 1, a + 1, sizeof(a));
            num = 0;
            for(int i = 2; i <= n; i++)
            {
                if(b[i - 1] == 0)
                {
                    num ++;
                    b[i - 1] = !b[i - 1];
                    b[i] = !b[i];
                    b[i + 1] = !b[i + 1];
                }
            }
            if(judge(n))
                ans = min(ans, num);
            memcpy(b + 1, a + 1, sizeof(a));
            num = 1;
            b[n] = !b[n];
            b[n - 1] = !b[n - 1];
            for(int i = n - 1; i >= 1; i--)
            {
                if(b[i + 1] == 0)
                {
                    num ++;
                    b[i + 1] = !b[i + 1];
                    b[i] = !b[i];
                    b[i - 1] = !b[i - 1];
                }
            }
            if(judge(n))
                ans = min(ans, num);
            memcpy(b + 1, a + 1, sizeof(a));
            num = 0;
            for(int i = n - 1; i >= 1; i--)
            {
                if(b[i + 1] == 0)
                {
                    num ++;
                    b[i + 1] = !b[i + 1];
                    b[i] = !b[i];
                    b[i - 1] = !b[i - 1];
                }
            }
            if(judge(n))
                ans = min(ans, num);
            memcpy(b + 1, a + 1, sizeof(a));
            num = 2;
            b[n] = !b[n];
            b[n - 1] = !b[n - 1];
            b[1] = !b[1];
            b[2] = !b[2];
            for(int i = 2; i <= n - 1; i++)
            {
                if(b[i - 1] == 0)
                {
                    num ++;
                    b[i - 1] = !b[i - 1];
                    b[i] = !b[i];
                    b[i + 1] = !b[i + 1];
                }   
            }
            if(judge(n))
                ans = min(ans, num);
            if(ans != INF)
                printf("%d\n", ans);
            else
                printf("-1\n");
        }
    }
}