Codeforces Round #738 (Div. 2)_C. Mocha and Hiking

C. Mocha and Hiking

题目传送门：

题目传送门

题面：

题目大意：

给你一张有向图，共$n+1$个点，其中前$n$个点的第$i$个点到第$i+1$个点都连着一条有向边。
数组a表示$1n$这$n$个村庄与第$n+1$个村庄之间路的关系。

1. 若为0，表示可以从第$i$个村庄走到第$n+1$个村庄。
2. 若为1则反过来，表示可以从第$n+1$个村庄走到第$i$个村庄
   这些都是有向边。
   问你是否存在经过所有点一次且仅一次的路径。

思路：

序列1～n是原本就存在的，可以直接从头尾添加$n+1$这个点有以下两种情况：

1. a[n]==0，直接$1..n+1$。
2. a[1]==1，直接先$n+1$，然后再跑$1...n$。

不然就找一下是否存在$a_i==0\&a_i{+}_1==1$，然后在这中间插一下$n+1$也行。

代码：

#include<bits/stdc++.h>

using namespace std;
const int maxn = 1e4 + 10;
int a[maxn];

int main() {
    int T;
    cin >> T;
    while (T--) {
        int n;
        cin >> n;
        int p = -1;
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            if (i == 1) continue;
            if (a[i] == 1 && a[i - 1] == 0) {
                p = i - 1;
            }
        }
        if (a[n] == 0 || a[1] == 1) {
            if (a[n] == 0) {
                for (int i = 1; i <= n + 1; ++i) {
                    printf("%d ", i);
                }
                printf("\n");
            } else if (a[1] == 1) {
                printf("%d ", n + 1);
                for (int i = 1; i <= n; ++i) {
                    printf("%d ", i);
                }
                printf("\n");
            }
        } else {
            if (p == -1) {
                printf("-1\n");
            } else {
                for (int i = 1; i <= n; i++) {
                    printf("%d ", i);
                    if (i == p)
                        printf("%d ", n + 1);
                }
                printf("\n");
            }
        }
    }
}