結果

問題 No.2822 Lights Up! (Tree Edition)
ユーザー FplusFplusFFplusFplusF
提出日時 2024-06-15 05:49:23
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,598 bytes
コンパイル時間 3,375 ms
コンパイル使用メモリ 265,556 KB
実行使用メモリ 30,324 KB
最終ジャッジ日時 2024-06-18 19:46:09
合計ジャッジ時間 8,092 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 107 WA * 35
権限があれば一括ダウンロードができます

ソースコード

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;
}
template<class T> struct fenwick_tree{
    vector<T> v;
    int n;
    fenwick_tree(int x){
        n=x+1;
        v.assign(n+1,0);
    }
    fenwick_tree(vector<T> &a){
        n=(int)a.size()+1;
        v.assign(n+1,0);
        if((int)a.size()==0) return;
        v[1]=a[0];
        for(int i=0;i<n-2;i++) v[i+2]=a[i+1]-a[i];
        for(int idx=1;idx<=n;idx++){
            if(n<idx+(idx&(-idx))) continue;
            v[idx+(idx&(-idx))]+=v[idx];
        }
    }
    void add_sub(int i,T x){
        i++;
        for(int idx=i;idx<=n;idx+=(idx&(-idx))){
            v[idx]+=x;
        }
    }
    void add(int l,int r,T x){  //[l,r)
        assert(0<=l&&l<=n);
        assert(0<=r&&r<=n);
        assert(l<=r);
        add_sub(l,x);
        add_sub(r,-x);
    }
    T get(int i){
        assert(0<=i&&i<n);
        i++;
        T ret=0;
        for(int idx=i;0<idx;idx-=(idx&(-idx))){
            ret+=v[idx];
        }
        return ret;
    }
    int lower_bound(T w){
        if(w<=0) return 0;
        int x=0;
        static int r=1;
        if(r==1){
            while((r<<1)<=n) r<<=1;
        }
        while((r<<1)<=n) r<<=1;
        for(int k=r;0<k;k>>=1){
            if(x+k<=n&&v[x+k]<w){
                w-=v[x+k];
                x+=k;
            }
        }
        x++;
        return x;
    }
};
vector<vector<ll>> g;
vector<ll> ord,in,out;
void dfs(ll x,ll p){
    in[x]=len(ord);
    ord.push_back(x);
    for(auto i:g[x]){
        if(i==p) continue;
        dfs(i,x);
    }
    out[x]=len(ord);
    ord.push_back(x);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    ll n;
    cin >> n;
    g.resize(n);
    in.resize(n);
    out.resize(n);
    vector<ll> p(n,-1);
    replr(i,1,n){
        cin >> p[i];
        p[i]--;
        g[i].push_back(p[i]);
        g[p[i]].push_back(i);
    }
    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]--;
    }

    dfs(0,-1);
    
    ll sz=len(ord);
    vector<set<ll>> lr(sz);
    rep(i,q){
        if(out[u[i]]>out[v[i]]) swap(u[i],v[i]);
        u[i]=out[u[i]];
        v[i]=out[v[i]];
        lr[u[i]].insert(v[i]);
    }
    
    fenwick_tree<ll> BIT(sz);
    replr(i,1,n){
        if(s[i]=='#') BIT.add(in[i],in[i]+1,1);
    }

    rep(i,sz){
        ll x=in[ord[i]],y=out[ord[i]];
        if(i==x) x=y;
        if(!((BIT.get(i)+BIT.get(x))&1)) continue;
        if(lr[i].empty()) continue;
        ll j=*begin(lr[i]);
        BIT.add(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();
    }
    

    rep(i,sz){
        ll x=in[ord[i]],y=out[ord[i]];
        if(i==x) x=y;
        if((BIT.get(i)+BIT.get(x))&1){
            cout << "No\n";
            return 0;
        }
    }
    cout << "Yes\n";
}
0