結果

問題 No.2822 Lights Up! (Tree Edition)
ユーザー FplusFplusF
提出日時 2024-06-14 06:16:56
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 6,124 bytes
コンパイル時間 4,349 ms
コンパイル使用メモリ 284,876 KB
実行使用メモリ 71,300 KB
最終ジャッジ日時 2024-06-18 19:45:50
合計ジャッジ時間 11,591 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 105 WA * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using ull=unsigned long long;
using pll=pair<ll,ll>;
using tll=tuple<ll,ll,ll>;
using ld=long double;
const ll INF=(1ll<<60);
#define rep(i,n) for (ll i=0;i<(ll)(n);i++)
#define replr(i,l,r) for (ll i=(ll)(l);i<(ll)(r);i++)
#define all(v) v.begin(),v.end()
#define len(v) ((ll)v.size())
template<class T> inline bool chmin(T &a,T b){
    if(a>b){
        a=b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T &a,T b){
    if(a<b){
        a=b;
        return true;
    }
    return false;
}
#include<atcoder/lazysegtree>
ll op(ll a,ll b){
    return a^b;
}
ll e(){
    return 0;
}
ll mpg(ll f,ll x){
    return f^x;
}
ll comp(ll f,ll g){
    return f^g;
}
ll id(){
    return 0;
}
struct tree{
    int n,max_k=1;
    vector<vector<int>> g;
    vector<vector<int>> parent;
    vector<int> depth;
    tree(int g_size){
        n=g_size;
        g.resize(n);
    }
    void add_edge(int u,int v){
        g[u].push_back(v);
        g[v].push_back(u);
    }
    void LCA(int root=0){
        while((1<<max_k)<n) max_k++;
        parent.assign(max_k,vector<int>(n,-1));
        depth.assign(n,-1);
        dfs(root,-1,0);
        for(int k=0;k+1<max_k;k++){
            for(int i=0;i<n;i++){
                if(parent[k][i]==-1){
                    parent[k+1][i]=-1;
                }else{
                    parent[k+1][i]=parent[k][parent[k][i]];
                }
            }
        }
    }
    void dfs(int v,int par,int d){
        parent[0][v]=par;
        depth[v]=d;
        for(auto &i:g[v]){
            if(i!=par){
                dfs(i,v,d+1);
            }
        }
    }
    int doubling(int x,int diff){
        for(int k=0;k<max_k;k++){
            if(diff&(1<<k)){
                x=parent[k][x];
            }
        }
        return x;
    }
    int get_lca(int u,int v){
        if(depth[u]<depth[v]) swap(u,v);
        int diff=depth[u]-depth[v];
        for(int k=0;k<max_k;k++){
            if(diff&(1<<k)){
                u=parent[k][u];
            }
        }
        if(u==v) return u;
        for(int k=max_k-1;0<=k;k--){
            if(parent[k][u]!=parent[k][v]){
                u=parent[k][u];
                v=parent[k][v];
            }
        }
        return parent[0][u];
    }
    int get_dist(int u,int v){
        return depth[u]+depth[v]-2*depth[get_lca(u,v)];
    }
    int on_path(int x,int u,int v){
        return get_dist(u,x)+get_dist(x,v)==get_dist(u,v);
    }
};
struct HLD{
    int n;
    vector<vector<int>> g;
    vector<int> par,sz,in,out,head;
    int in_ord=0,out_ord=0;
    HLD(int x){
        n=x;
        g.resize(n);
        par.resize(n);
        sz.assign(n,0);
        in.resize(n);
        out.resize(n);
        head.resize(n);
    }
    void add_edge(int u,int v){
        g[u].push_back(v);
        g[v].push_back(u);
    }
    void dfs_sz(int x,int p){
        sz[x]++;
        if(0<g[x].size()&&g[x].front()==p) swap(g[x].front(),g[x].back());
        for(auto &i:g[x]){
            if(i==p) continue;
            dfs_sz(i,x);
            sz[x]+=sz[i];
            if(sz[g[x].front()]<sz[i]) swap(g[x].front(),i);
        }
    }
    void dfs_hld(int x,int p){
        par[x]=p;
        in[x]=in_ord;
        in_ord++;
        for(auto &i:g[x]){
            if(i==p) continue;
            if(i==g[x].front()) head[i]=head[x];
            else head[i]=i;
            dfs_hld(i,x);
        }
        out[x]=out_ord;
        out_ord++;
    }
    void solve(int root){
        dfs_sz(root,-1);
        dfs_hld(root,-1);
    }
    vector<pair<int,int>> query(int u,int v,bool edge){
        vector<pair<int,int>> ret;
        while(true){
            if(in[v]<in[u]) swap(u,v);
            if(head[u]==head[v]) break;
            ret.emplace_back(in[head[v]],in[v]+1);
            v=par[head[v]];
        }
        if(in[u]+edge!=in[v]+1) ret.emplace_back(in[u]+edge,in[v]+1);
        return ret;
    }
};

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    ll n;
    cin >> n;
    ll sz=n+n-1;
    vector<ll> p(n,-1);
    tree g(sz);
    HLD h(sz);
    replr(i,1,n){
        cin >> p[i];
        p[i]--;
        g.add_edge(i,n+i-1);
        g.add_edge(p[i],n+i-1);
        h.add_edge(i,n+i-1);
        h.add_edge(p[i],n+i-1);
    }
    string s(n,'x');
    replr(i,1,n) cin >> s[i];
    ll q;
    cin >> q;
    vector<ll> u(q),v(q);
    rep(i,q){
        cin >> u[i] >> v[i];
        u[i]--;
        v[i]--;
    }
    g.LCA();
    rep(i,q){
        ll x=g.get_lca(u[i],v[i]);
        ll nu=u[i],nv=v[i];
        if(u[i]==x){
            nu=g.doubling(v[i],g.depth[v[i]]-g.depth[u[i]]-1);
        }else{
            nu=g.parent[0][u[i]];
        }
        if(v[i]==x){
            nv=g.doubling(u[i],g.depth[u[i]]-g.depth[v[i]]-1);
        }else{
            nv=g.parent[0][v[i]];
        }
        u[i]=nu;
        v[i]=nv;
    }
    h.solve(0);
    vector<ll> out(sz),ord(sz);
    rep(i,sz) out[i]=h.out[i];
    rep(i,sz) ord[out[i]]=i;
    vector<set<ll>> lr(sz);
    rep(i,q){
        if(out[v[i]]<out[u[i]]) swap(u[i],v[i]);
        lr[u[i]].insert(v[i]);
    }
    atcoder::lazy_segtree<ll,op,e,ll,mpg,comp,id> seg(sz);

    auto update=[&](ll a,ll b,ll x){
        auto now=h.query(a,b,false);
        for(auto [l,r]:now){
            seg.apply(l,r,x);
        }
    };

    auto get=[&](ll a,ll b){
        auto now=h.query(a,b,false);
        ll ret=0;
        for(auto [l,r]:now){
            ret^=seg.prod(l,r);
        }
        return ret;
    };

    replr(i,1,n){
        update(n+i-1,n+i-1,s[i]=='#');
    }

    for(auto i:ord){
        if(get(i,i)==0) continue;
        if(lr[i].empty()) continue;
        ll j=*begin(lr[i]);
        update(i,j,1);
        lr[i].erase(j);
        if(len(lr[i])<len(lr[j])){
            for(auto k:lr[i]) lr[j].insert(k);
        }else{
            for(auto k:lr[j]) lr[i].insert(k);
            swap(lr[i],lr[j]);
        }
        lr[i].clear();
    }

    ll sum=0;
    replr(i,1,n){
        sum+=get(n+i-1,n+i-1);
    }
    if(sum==0) cout << "Yes\n";
    else cout << "No\n";
}
0