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