hiho #1114 : 小Hi小Ho的惊天大作战：扫雷·一

#1114 : 小Hi小Ho的惊天大作战：扫雷·一

时间限制: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行，其中第一行先输出一定为地雷的格子的数量，然后按照从小到大的顺序输出所有一定为地雷的格子的位置，第二行先输出一定不为地雷的格子的数量，按照从小到大的顺序输出所有一定不为地雷的格子的位置。

样例输入

2
3
1 1 1
10
1 2 1 2 2 3 2 2 2 2 

样例输出

1 2
2 1 3
7 1 3 5 6 7 9 10
3 2 4 8 

思路：

用dfs深搜，找出ans数组，对于存在多种答案的情况，讲不都是0或不都是1的，至为2，输出的时候，输出所有的0，1位置。

AC代码：

 1 #include <iostream>
 2 #include <stdio.h>
 3 #include <string.h>
 4 #define MAX 100005
 5 using namespace std;
 6
 7 int n, a[MAX],up[MAX],ans[MAX];
 8
 9 void dfs(int pos)
10 {
11     if (pos == n+1)
12     {
13         for (int i = 1; i <= n; i++)
14         {
15             if(ans[i]==-1)    ans[i] = up[i];
16             else if (ans[i] != up[i]) ans[i] = 2;      //存在几种可能的情况下，如果不都是0，或不都是1，至为2.
17         }
18         return;
19     }
20
21     for (int i = 0; i < 2; i++)
22     {
23         up[pos] = i;
24         if (up[pos] + up[pos - 1]>a[pos])
25             continue;
26
27         if (pos > 1 && up[pos] + up[pos - 1] + up[pos - 2] != a[pos - 1])
28             continue;
29
30         if (pos == n && up[pos] + up[pos - 1] != a[pos])
31             continue;
32
33         dfs(pos + 1);
34     }
35 }
36
37 int main()
38 {
39     int task;
40     cin >> task;
41     while (task--)
42     {
43         int x=0, o=0;
44
45         cin >> n;
46         for (int i = 1; i <= n; i++)
47         {
48             cin >> a[i];
49         }
50         memset(ans, -1, sizeof(ans));
51         dfs(1);
52
53         for (int i = 1; i <= n; i++)
54         {
55             if (ans[i] == 0)
56                 o++;
57             else if (ans[i] == 1)
58                 x++;
59         }
60
61         cout << x;
62         for (int i = 1; i <= n; i++){
63             if (ans[i] == 1)
64                 cout << " " << i;
65         }
66         cout << endl;
67         cout << o;
68         for (int i = 1; i <= n; i++){
69             if (ans[i] == 0)
70                 cout << " " << i;
71         }
72         cout << endl;
73     }
74     system("pause");
75 }