hihoCoder 1114 小Hi小Ho的惊天大作战：扫雷·一 最详细的解题报告

题目来源：小Hi小Ho的惊天大作战：扫雷·一

解题思路：因为只要确定了第一个是否有地雷就可以推算出后面是否有地雷（要么为0，要么为1，如果不是这两个值就说明这个方案行不通），如果两种可能中有一种成功，只需要计算包含有多少个1和多少个0，如果两种可能都成功了，都为1的才是有雷，都为0的才是没有地雷。

具体算法（java版，可以直接AC）

 import java.util.Scanner;

 public class Main {

     public static boolean flag1 = true;//当第一个为1（有雷）时，依次推算后的结果
     public static boolean flag2 = true;//当第一个为0（没雷）时，依次推算后的结果

     public static void solve(int[] maze, int[][] mine, int N) {
         mine[0][1] = 1;//第一个有雷
         mine[1][1] = 0;//第一个没雷

         for (int i = 2; i <= N; i++) {
             if (flag1) {
                 mine[0][i] = maze[i - 1] - mine[0][i - 1] - mine[0][i - 2];
                                 //要么有雷，要么没雷
                 if (mine[0][i] == 1 || mine[0][i] == 0) {
                     flag1 = true;
                 } else {
                     flag1 = false;//推算失败
                     break;
                 }
             }
         }

         for (int i = 2; i <= N; i++) {
             if (flag2) {
                 mine[1][i] = maze[i - 1] - mine[1][i - 1] - mine[1][i - 2];
                 if (mine[1][i] == 1 || mine[1][i] == 0) {
                     flag2 = true;
                 } else {
                     flag2 = false;
                     break;
                 }
             }
         }
         if (flag1) {//验证最后一个是否正确
             if (maze[N] != mine[0][N - 1] + mine[0][N]) {
                 flag1 = false;
             }
         }
         if (flag2) {
             if (maze[N] != mine[1][N - 1] + mine[1][N]) {
                 flag2 = false;
             }
         }
     }

     public static void main(String[] args) {
         Scanner scanner = new Scanner(System.in);
         int task = scanner.nextInt();
         while (task > 0) {
             task--;
             int N = scanner.nextInt();
             int[] maze = new int[N + 1];
             int[][] mine = new int[2][N + 1];
             for (int i = 1; i <= N; i++) {
                 maze[i] = scanner.nextInt();
             }
             flag1 = flag2 = true;
             solve(maze, mine, N);
             int hasMine = 0, noMine = 0;//统计有雷和没雷的数量
             int[] hasMineAns = new int[N];
             int[] noMineAns = new int[N];
             if (flag1 && flag2) {//两种可能都成功
                 for (int i = 1; i <= N; i++) {
                     if (mine[0][i] == 1 && mine[1][i] == 1) {//同时为1（有雷）
                         hasMineAns[hasMine++] = i;
                     } else if (mine[0][i] == 0 && mine[1][i] == 0) {//同时为0（没雷）
                         noMineAns[noMine++] = i;
                     }
                 }
             } else if (flag1 && !flag2) {//其中一种可能是成功的，另外一种失败
                 for(int i=1;i<=N;i++){
                     if(mine[0][i]==1){
                         hasMineAns[hasMine++] = i;
                     }else{
                         noMineAns[noMine++] = i;
                     }
                 }
             } else if (!flag1 && flag2) {
                 for(int i=1;i<=N;i++){
                     if(mine[1][i]==1){
                         hasMineAns[hasMine++] = i;
                     }else{
                         noMineAns[noMine++] = i;
                     }
                 }
             }
             System.out.print(String.format("%d", hasMine));
             for(int i=0;i<hasMine;i++){
                 System.out.print(String.format(" %d", hasMineAns[i]));
             }
             System.out.print(String.format("\n%d", noMine));
             for(int i=0;i<noMine;i++){
                 System.out.print(String.format(" %d", noMineAns[i]));
             }
             System.out.println();
         }
         scanner.close();
     }
 }