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

原题地址

回溯+搜索

枚举每个位置上能否放地雷，当第i个位置枚举完成后，第i-1个位置的情况就确定了，此时，检查第i-1个位置是否满足要求，即左右间隔为1的范围内地雷数是否等于申明数字，如果满足条件，那么继续搜索下去，如果不满足条件，抛弃这个搜索分支。

搜索完成后，将所有可行解按位置“与”一下 ，找到那些一定为地雷或一定为空的格子。

最后输出即可。

代码：

 #include <iostream>
 #include <cstring>

 using namespace std;

 void merge(int *t, int *r, int n) {
   for (int i = ; i < n; i++) {
     if (t[i] == - || t[i] == r[i])
       t[i] = r[i];
     else
       t[i] = -;
   }
 }

 void find(int *a, int *r, int *t, int p, int n) {
   if (p >= n) {
     if ((n ==  && a[p - ] == r[p - ])
         || ((n > ) && a[p - ] == r[p - ] + r[p - ]))
       merge(t, r, n);
   }
   else if (p == ) {
     for (int i = ; i < ; i++) {
       r[p] = i;
       find(a, r, t, p + , n);
     }
   }
   else if (p == ) {
     for (int i = ; i < ; i++) {
       r[p] = i;
       if (a[p - ] == r[p - ] + r[p])
         find(a, r, t, p + , n);
     }
   }
   else {
     for (int i = ; i < ; i++) {
       r[p] = i;
       if (a[p - ] == r[p] + r[p - ] + r[p - ])
         find(a, r, t, p + , n);
     }
   }
 }

 int main() {
   int n;

   cin >> n;
   while (n--) {
     int N;
     cin >> N;
     int *a = new int[N];
     int *r = new int[N];
     int *t = new int[N];

     for (int i = ; i < N; i++)
       cin >> a[i];
     memset(r, , N * sizeof(int));
     memset(t, -, N * sizeof(int));

     find(a, r, t, , N);

     int mine = ;
     int not_mine = ;

     for (int i = ; i < N; i++) {
       mine += (t[i] ==  ?  : );
       not_mine += (t[i] ==  ?  : );
     }

     cout << mine;
     for (int i = ; i < N; i++)
       if (t[i] == )
         cout << " " << i + ;
     cout << endl;

     cout << not_mine;
     for (int i = ; i < N; i++)
       if (t[i] == )
         cout << " " << i + ;
     cout << endl;

     delete t;
     delete r;
     delete a;
   }

   return ;
 }