hihocoder1114(模拟)

最新推荐文章于 2019-02-26 22:24:26 发布

Stayaccept

最新推荐文章于 2019-02-26 22:24:26 发布

阅读量218

点赞数

分类专栏：奇思妙想系列文章标签： acm hiho一下

本文链接：https://blog.csdn.net/stay_accept/article/details/73992041

版权

奇思妙想系列专栏收录该内容

47 篇文章 0 订阅

订阅专栏

链接：点击打开链接

题意：在一个大小为2*N的广场，其中第一行里的某一些格子里可能会有至多一个地雷，而第二行的格子里全都为数字，表示第一行中距离与这个格子不超过2的格子里总共有多少个地雷，即第二行的第i个格子里的数字表示第一行的第i-1个, 第i个, 第i+1个，三个格子（如果i=1或者N则不一定有三个）里的地雷的总数。

代码：

#include <set>
#include <map>
#include <queue>
#include <vector>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int siz=100005;
int n,a[siz],s1[siz],s2[siz];
vector<int> ans1,ans2;
int judge1(){
    int i;
    s1[1]=0,s1[2]=a[1]-s1[1];
    for(i=3;i<=n;i++)
    s1[i]=a[i-1]-s1[i-1]-s1[i-2];
    if(a[n]!=s1[n]+s1[n-1])
    return 1;
    for(i=1;i<=n;i++)
    if(s1[i]<0||s1[i]>1)
    return 1;
    return 0;
}
int judge2(){
    int i;
    s2[1]=1,s2[2]=a[1]-s2[1];
    for(i=3;i<=n;i++)
    s2[i]=a[i-1]-s2[i-1]-s2[i-2];
    if(a[n]!=s2[n]+s2[n-1])                     //末尾特殊处理
    return 1;
    for(i=1;i<=n;i++)
    if(s2[i]<0||s2[i]>1)
    return 1;
    return 0;
}
void print(){
    int i;
    printf("%d",ans1.size());
    for(i=0;i<ans1.size();i++)
    printf(" %d",ans1[i]);
    printf("\n");
    printf("%d",ans2.size());
    for(i=0;i<ans2.size();i++)
    printf(" %d",ans2[i]);
    printf("\n");
}
int main(){
    int t,i,j,op1,op2;
    scanf("%d",&t);
    while(t--){                                 //枚举第一个格是0还是1，就能推出全部的情况
        scanf("%d",&n);                         //如果两种都存在，则两种中相同位置为1或0的为
        for(i=1;i<=n;i++)                       //一定出现的
        scanf("%d",&a[i]);
        op1=judge1();
        op2=judge2();
        ans1.clear();
        ans2.clear();
        if(op1==1){
            for(i=1;i<=n;i++)
            if(s2[i]==1)
            ans1.push_back(i);
            for(i=1;i<=n;i++)
            if(s2[i]==0)
            ans2.push_back(i);
            print();
        }
        else if(op2==1){
            for(i=1;i<=n;i++)
            if(s1[i]==1)
            ans1.push_back(i);
            for(i=1;i<=n;i++)
            if(s1[i]==0)
            ans2.push_back(i);
            print();
        }
        else{
            for(i=1;i<=n;i++)
            if(s1[i]==1&&s2[i]==1)
            ans1.push_back(i);
            for(i=1;i<=n;i++)
            if(s1[i]==0&&s2[i]==0)
            ans2.push_back(i);
            print();
        }
    }
    return 0;
}