Atcoder Beginner Contest 55 D Menagerie (枚举+验证)

题意:

有两种动物,羊会说事实,狼会故意说谎,一群动物站成一圈(只包含狼和羊),他们没人分别说自己旁边的两种动物是否相同.题目里是给出一串字符串,字符串里'X'代表站在这个位置的动物说它两边的动物种类不同,'O'代表站在这个位置的动物说它两边的动物种类相同,求符合这个字符串的动物排列方式,如果没有输出"-1".

解题思路:

数据范围是1e5,没有办法去枚举所有答案.

但是根据题目给出的字符串,我们只需要枚举前两个的情况,假设我们已经知道前两个是什么动物,那么我们就能够根据题目提供的他们认为两边的动物是否相同的信息来推出第三个动物是什么动物,以此类推,我们就在已知前两个动物种类的情况下,求出所有位置动物情况,但是怎么验证我们所求出的答案是否正确呢,我们需要check一下.

因为所有动物站成一圈,所以他们之间的 关系是可以循环的,也就是说,我们可以由最后两个动物推出第一个动物的情况,进而由最后一个动物和第一个动物推出第二个动物的情况,

假如我们在这里推出来的第一.二个动物的情况和我们所枚举的第一二个动物的情况相同,那么说明我们所求出的这个排列是可以循环的是正确的,如果不相符,就是错误的.

代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
char s[maxn];
int  is[maxn];
bool solve(int x, int y, int len)
{
      int i, j;
      is[0]=x;
      is[1]=y;
      for(i=2; i<=len+1; i++)
      {
	 if(is[i-1])
	 {
	    if(s[i-1]=='o')is[i]=is[i-2];
	    else is[i]=!is[i-2];
	 }
	 else  //前面的人说了假话
	 {
	    if(s[i-1]=='o')is[i]=!is[i-2];
	    else is[i]=is[i-2];
	 }
      }
	  /* for(i=0; i<len+2; i++)
	   {
	      printf(is[i]?"S":"W");
	   }
	   printf("\n");
	   */
      if(is[len]== is[0] && is[len+1]==is[1])
      {
	   for(i=0; i<len; i++)
	   {
	      printf(is[i]?"S":"W");
	   }
	   printf("\n");
	   return 1;
      }
      return 0;

}

int main()
{
   int n;
   scanf("%d", &n);
   int i;
   scanf("%s", s);
   int len=strlen(s);
   s[len]=s[0];
   s[len+1]=s[1];
   s[len+2]='\0';
   int j;
   for(i=0; i<2; i++)
   {
     for(j=0; j<2; j++)
     {
	if(solve(i, j, len))return 0;
     }
   }
   printf("-1\n");
}