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;
    }

}