原题链接
题意
输入一个数 n n n ,代表有 n + 1 n + 1 n+1 个点,然后 1 − ( n − 1 ) 1 - (n - 1) 1−(n−1),每个点都有一条边连向自己后面那条边。
然后对于点 1 − n 1 - n 1−n,到点 n + 1 n + 1 n+1, 有 n n n 个数
- 如果 a i = = 0 a_i == 0 ai==0 ,代表有一条边从 i i i 连向 n + 1 n + 1 n+1
- 如果 a i = = 1 a_i == 1 ai==1 ,代表有一条边从 n + 1 n + 1 n+1 连向 i i i
思路
- 第一种情况如果 a 1 = = 1 a_1 == 1 a1==1,输出 n + 1 n + 1 n+1 、 1 − n 1 - n 1−n
- 第一种情况如果 a n = = 0 a_n == 0 an==0,输出 1 − n 1 - n 1−n 、 n + 1 n + 1 n+1
- 第三种情况只要有 连续的 两个数 是 01,那么也肯定有解
可以总结出:一定有解 !
代码
#include<bits/stdc++.h>
using namespace std;
int main()
{
int t; cin >> t;
while (t -- )
{
int n; cin >> n;
int a[n + 10];
for (int i = 1; i <= n; i ++ )
{
cin >> a[i];
}
if (a[1] == 1)
{
cout << n + 1 << " ";
for (int i = 1; i <= n; i ++ )
{
cout << i << " ";
}
cout << endl;
continue;
}
if (a[n] == 0)
{
for (int i = 1; i <= n; i ++ )
{
cout << i << " ";
}
cout << n + 1 << " ";
cout << endl;
continue;
}
bool f = 0;
for (int i = 1; i <= n; i ++ )
{
cout << i << " ";
if (a[i] == 0 && a[i + 1] == 1 && f == 0)
{
cout << n + 1 << " ";
f = 1;
}
}
cout << endl;
}
return 0;
}
总结
画画图就可以推出规律啦!