标签:
“我们还是循序渐进,先来考虑这样一个简单化问题:”小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 }
hiho #1114 : 小Hi小Ho的惊天大作战:扫雷·一
标签:
原文地址:http://www.cnblogs.com/SeekHit/p/5650371.html