hiho 第三十周 小Hi小Ho的惊天大作战:扫雷·一

来源:互联网 发布:路由器封端口 编辑:程序博客网 时间:2024/04/27 18:43

 

 

如题:http://hihocoder.com/problemset/problem/1114

 

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

故事背景:密室、监视器与充满危机的广场

“我们还是循序渐进,先来考虑这样一个简单化问题:”小Hi思索片刻,道:“在一个大小为2*N的广场,其中第一行里的某一些格子里可能会有至多一个地雷,而第二行的格子里全都为数字,表示第一行中距离与这个格子不超过2的格子里总共有多少个地雷,即第二行的第i个格子里的数字表示第一行的第i-1个, 第i个, 第i+1个,三个格子(如果i=1或者N则不一定有三个)里的地雷的总数。”

“而我们要做的是——找出哪些地方一定是雷,哪些地方一定不是雷。”小Ho道:“不然,我可就要光荣牺牲了。”

提示:寻找关键点——那些一旦决定之后就能让局面豁然开朗的地方。

输入

每个测试点(输入文件)存在多组测试数据。

每个测试点的第一行为一个整数Task,表示测试数据的组数。

在一组测试数据中:

第1行为1个整数N,表示迷宫的宽度。

第2行为N个整数A_1 ... A_N,依次表示迷宫第二行的N个格子里标注的数字。

对于100%的数据,满足1<=N<=10^5, 0<=a_i<=3.<>

对于100%的数据,满足符合数据描述的地图一定存在。

输出

对于每组测试数据,输出2行,其中第一行先输出一定为地雷的格子的数量,然后按照从小到大的顺序输出所有一定为地雷的格子的位置,第二行先输出一定不为地雷的格子的数量,按照从小到大的顺序输出所有一定不为地雷的格子的位置。


样例输入
231 1 1101 2 1 2 2 3 2 2 2 2 
样例输出
1 22 1 37 1 3 5 6 7 9 10 3 2 4 8 

 

 

WA了好几次,最后竟然是wa在两种结果都可行的时候,对第二行输出的是一定不为地雷 我输出的是不一定为地雷,看别人代码猜发现。真是瞎了!

 

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAX 100005

int a[MAX];
int flg[MAX];
int flg1[MAX];
int n;

int fun(int f[],int hav)
{
 int sum=0;
 f[1]=0;
 if(hav==1)
 {
 f[1]=1;
 sum=1;
 }
 int i;
 for(i=1;i<n;i++)
 {
  if(a[i]-sum==1)
   f[i+1]=1;
  else if(a[i]-sum==0)
   f[i+1]=0;
  else
   return 0;
  sum=f[i]+f[i+1];
 }
 if(f[n-1]+f[n]==a[n])
 return 1;
 else
  return 0;
}
int main()
{
// freopen("C:\\1.txt","r",stdin);
 int T;
 cin>>T;
 while(T--)
 {
  cin>>n;
  int i;
  for(i=1;i<=n;i++)
   scanf("%d",&a[i]);
   int f1=fun(flg,0);
   int f2=fun(flg1,1);
   if(f2&&!f1)
   {
    int sum=0;
   for(i=1;i<=n;i++)
    if(flg1[i])
     sum++;
   printf("%d ",sum);
   for(i=1;i<=n;i++)
    if(flg1[i])
     printf("%d ",i);
   printf("\n");
   printf("%d ",n-sum);
   for(i=1;i<=n;i++)
    if(!flg1[i])
     printf("%d ",i);
   printf("\n");
   }
   else if(f1&&!f2)
   {
    int sum=0;
   for(i=1;i<=n;i++)
    if(flg[i])
     sum++;
   printf("%d ",sum);
   for(i=1;i<=n;i++)
    if(flg[i])
     printf("%d ",i);
   printf("\n");
   printf("%d ",n-sum);
   for(i=1;i<=n;i++)
    if(!flg[i])
     printf("%d ",i);
   printf("\n");
   }
   else
   {
    int sum=0;
    for(i=1;i<=n;i++)
    {
     if(flg[i]&&flg1[i])
      sum++;
    }
    printf("%d",sum);
    for(i=1;i<=n;i++)
     if(flg[i]&&flg1[i])
      printf(" %d",i);
    printf("\n");
    sum=0;
    for(i=1;i<=n;i++)
    {
     if((!flg[i])&&(!flg1[i]))
      sum++;
    }
    printf("%d",sum);
    for(i=1;i<=n;i++)
     if((!flg[i])&&(!flg1[i]))
      printf(" %d",i);
    printf("\n");
   }
 }
 return 0;
}

 

0 0
原创粉丝点击