hihocoder1114 小Hi小Ho的惊天大作战：扫雷·一

本文主要是介绍hihocoder1114 小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

import java.io.BufferedInputStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;public class Main {public static void main(String[] args) {new Task().solve();}}class Task {int[] table, answer;int n;void solve() {Scanner cin = new Scanner(new BufferedInputStream(System.in));PrintWriter cout = new PrintWriter(System.out);int t = cin.nextInt();while (t-- > 0) {n = cin.nextInt();table = new int[n + 2];answer = new int[n + 2];for (int i = 1; i <= n; i++)table[i] = cin.nextInt();int[][] state = new int[2][n + 1];Arrays.fill(state[0], 0);Arrays.fill(state[1], 0);boolean[] yes = new boolean[2];yes[0] = yes[1] = false;for (int i = 0; i <= 1; i++) {if (test(i)) {yes[i] = true;System.arraycopy(answer, 1, state[i], 1, n);}}List<Integer>[] res = new List[2];res[0] = new ArrayList<Integer>();res[1] = new ArrayList<Integer>();if (yes[0] && yes[1]) {for (int i = 1; i <= n; i++) {if (state[0][i] == 1 && state[1][i] == 1)res[1].add(i);if (state[0][i] == 0 && state[1][i] == 0)res[0].add(i);}} else {int k = -1;if (yes[0])k = 0;if (yes[1])k = 1;for (int i = 1; i <= n; i++) {if (state[k][i] > 0)res[1].add(i);if (state[k][i] == 0)res[0].add(i);}}for (int i = 1; i >= 0; i--) {cout.print(res[i].size());for (int j : res[i]) {cout.print(" " + j);}cout.println();}}cout.flush();}boolean test(int start) {answer[1] = start;answer[0] = answer[n + 1] = 0;for (int i = 2; i <= n; i++) {answer[i] = table[i - 1] - answer[i - 2] - answer[i - 1];}for (int i = 1; i <= n; i++) {if (answer[i] < 0 || answer[i] > 1)return false;if ((answer[i - 1] + answer[i] + answer[i + 1]) != table[i])return false;}return true;}}