CF 2002 D1. DFS Checker (Easy Version) （*1900）思維

Showball發表於2024-09-02

CF 2002 D1. DFS Checker (Easy Version) （*1900）思維

題目連結

題意：

給你一棵 \(n\) 個節點組成的完全二叉樹，並給出一個排列 \(p\) 。接下來進行 \(q\) 次詢問。

每次詢問給你 \(x\) 和 \(y\) ，你需要交換 \(p_x\) 和 \(p_y\)。並且回答交換之後的排列 \(p\) 是否是這棵

完全二叉樹的DFS序。交換是持久的。

思路：

我們可以將每個節點在DFS序中的位置看做點權。那麼我們發現對於每棵子樹中的節點，它的點權一定是連續的。

由於是排列，不會重複，因此我們只需要比較最小值和最大值即可。這樣每次判斷每個節點即可。我們可以對每個節點維護其兒子的集合。

對於修改：我們直接暴力維護好set即可。

時間複雜度 \(O(nlog^2n)\)

程式碼：

#include<bits/stdc++.h>

using namespace std;

#define ff first
#define ss second
#define pb push_back
#define all(u) u.begin(), u.end()
#define endl '\n'
#define debug(x) cout<<#x<<":"<<x<<endl;

typedef pair<int, int> PII;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10, M = 105;
const int mod = 1e9 + 7;
const int cases = 1;

void Showball(){
   int n,q;
   cin>>n>>q;
   vector<int> e[n+1],fa(n+1),p(n+1),a(n+1);
   set<int> son[n+1];
   for(int i=1;i<=n;i++) e[i].clear(),son[i].clear();
   for(int i=2;i<=n;i++){
     cin>>fa[i];
     e[fa[i]].pb(i);
   }
   for(int i=1;i<=n;i++){
     cin>>p[i];
     a[p[i]]=i;
   }

   vector<int> sz(n+1);
   int cnt=0;

   auto check=[&](int u){
     return a[u]==*son[u].begin()&&a[u]+sz[u]-1==*son[u].rbegin();
   };

   function<void(int,int)> dfs=[&](int u,int father){
      sz[u]=1;son[u].insert(a[u]);
      for(auto v:e[u]){
        if(v==father) continue;
        dfs(v,u);
        sz[u]+=sz[v];
        for(auto it:son[v]) son[u].insert(it); 
      }
      cnt+=check(u);
   };

   dfs(1,0);
   while(q--){
    int x,y;
    cin>>x>>y;
    swap(p[x],p[y]);
    x=p[x],y=p[y];
    vector<int> cand;
    for(int i=x;i;i=fa[i]) cand.pb(i);
    for(int i=y;i;i=fa[i]) cand.pb(i);
    sort(all(cand));
    cand.erase(unique(all(cand)),cand.end());
    for(auto it:cand) cnt-=check(it);
    for(int i=x;i;i=fa[i]) son[i].erase(a[x]);
    for(int i=y;i;i=fa[i]) son[i].erase(a[y]);
    swap(a[x],a[y]);
    for(int i=x;i;i=fa[i]) son[i].insert(a[x]);
    for(int i=y;i;i=fa[i]) son[i].insert(a[y]);
    for(auto it:cand) cnt+=check(it);
    cout<<(cnt==n?"Yes\n":"No\n");
   }

}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T=1;
    if(cases) cin>>T;
    while(T--)
    Showball();
    return 0;
}

相關文章