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 = 0; i < n; i++) {
    if (t[i] == -1 || t[i] == r[i])
      t[i] = r[i];
    else
      t[i] = -2;
  }
}

void find(int *a, int *r, int *t, int p, int n) {
  if (p >= n) {
    if ((n == 1 && a[p - 1] == r[p - 1])
        || ((n > 1) && a[p - 1] == r[p - 1] + r[p - 2]))
      merge(t, r, n);
  }
  else if (p == 0) {
    for (int i = 0; i < 2; i++) {
      r[p] = i;
      find(a, r, t, p + 1, n);
    }
  }
  else if (p == 1) {
    for (int i = 0; i < 2; i++) {
      r[p] = i;
      if (a[p - 1] == r[p - 1] + r[p])
        find(a, r, t, p + 1, n);
    }
  }
  else {
    for (int i = 0; i < 2; i++) {
      r[p] = i;
      if (a[p - 1] == r[p] + r[p - 1] + r[p - 2])
        find(a, r, t, p + 1, 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 = 0; i < N; i++)
      cin >> a[i];
    memset(r, 0, N * sizeof(int));
    memset(t, -1, N * sizeof(int));

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

    int mine = 0;
    int not_mine = 0;

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

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

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

    delete t;
    delete r;
    delete a;
  }

  return 0;
}