題目鏈接

題意

給一個長度為$M$的序列，問它是否能作為給定的一棵樹的dfs序的一部分

思路

比賽的時候和隊友在寫假算法，本來以為會$TLE$或者$MLE$，但是一直$WA$。
回來問了學長才過的。
按照給定序列的順序從下到上打上同一個標記，遇到打過標記的點就$return$，然后按照標記從小到大$dfs$得到一個$dfs$序，最后查找序列是否出現在$dfs$序列里面。
因為$dfs$結束之后才加入隊列，所以序列前面的節點最先訪問，這是就給他打上一個小的標記，這樣只要序列正確，一定會出現在$dfs$序中

#include <bits/stdc++.h> #define LL long long #define P pair<int, int> #define lowbit(x) (x & -x) #define mem(a, b) memset(a, b, sizeof(a)) #define mid ((l + r) >> 1) #define lc rt<<1 #define rc rt<<1|1 using namespace std; const int maxn = 1e6 + 1;vector<int> g[maxn], num; int fa[maxn], sign[maxn], now; void dfs(int x) {for (int it : g[x]) {dfs(it);}num.push_back(x); }bool cmp(int x, int y) {return sign[x] < sign[y]; }int main () {ios::sync_with_stdio(0);cin.tie(0), cout.tie(0);int T;scanf("%d", &T);while (T--) { int n, m; scanf("%d%d", &n, &m);vector<int> list(m);now = 0;for (int i = 1; i <= n; ++i) sign[i] = 0, g[i].clear();for (int i = 2, x; i <= n; ++i) {scanf("%d", &x);g[x].push_back(i);fa[i] = x;}for (int i = 0; i < m; ++i) {scanf("%d", &list[i]);now = i+1;int tmp = list[i];while (sign[tmp] == 0) {sign[tmp] = now;tmp = fa[tmp];if (tmp == 0) break;}}for (int i = 1; i <= n; ++i) {sort(g[i].begin(), g[i].end(), cmp);}num.clear();dfs(1);int flag = 1;int j = 0;for (int i = 0; i < n; ++i) {if (num[i] == list[j]) j++;if (j == m) break;}if (j != m) flag = 0;puts(flag ? "NOT BAD":"BAD GUGU");}return 0; }

總結

以上是生活随笔為你收集整理的zzuli 2525: 咕咕的搜索序列的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。