Submission #812890

#TimeUsernameProblemLanguageResultExecution timeMemory
812890blackyukiDigital Circuit (IOI22_circuit)C++17
100 / 100
1049 ms23588 KiB
#include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<ll,ll> P; typedef vector<ll> vi; typedef vector<vi> vvi; typedef vector<P> vp; typedef vector<vp> vvp; typedef vector<bool> vb; #define rep(i,n) for(ll i=0;i<(ll)(n);i++) #define REP(i,k,n) for(ll i=(ll)(k);i<(ll)(n);i++) #define all(a) a.begin(),a.end() #define fi first #define se second #define pb emplace_back #define lb(v,k) (lower_bound(all(v),k)-v.begin()) template<class T> bool chmin(T&a,T b){if(a>b){a=b;return true;}return false;} template<class T> bool chmax(T&a,T b){if(a<b){a=b;return true;}return false;} template<class T> void out(T a){cout<<a<<'\n';} template<class T> void outv(T v){rep(i,v.size()){if(i)cout<<' ';cout<<v[i];}cout<<'\n';} const ll inf=1001001001001001001; const ll mod=1000002022; #include "circuit.h" struct segtree{ ll N=1; vi seg,lazy,sum; ll f(ll a,ll b){ return (a+b)%mod; } void init(vi a,vi b){ while(N<a.size())N<<=1; seg=vi(N*2);lazy=vi(N*2);sum=vi(N*2); rep(i,a.size()){ sum[i+N]=a[i]; seg[i+N]=a[i]*b[i]; } for(ll i=N-1;i>0;i--){ sum[i]=f(sum[i*2],sum[i*2+1]); seg[i]=f(seg[i*2],seg[i*2+1]); } } void eval(ll k,ll l,ll r){ if(!lazy[k])return; seg[k]=(sum[k]-seg[k])%mod; if(r-l>1)rep(t,2)lazy[k*2+t]^=1; lazy[k]=0; } ll get(){ eval(1,0,N); return seg[1]; } void upd(ll a,ll b,ll k=1,ll l=0,ll r=-1){ if(r==-1)r=N; if(a<=l&&r<=b){ lazy[k]^=1;eval(k,l,r);return; } if(r<=a||b<=l){ eval(k,l,r);return; } eval(k,l,r); upd(a,b,k*2,l,(l+r)/2); upd(a,b,k*2+1,(l+r)/2,r); seg[k]=f(seg[k*2],seg[k*2+1]); } }; segtree seg; ll n,N_; void init(int N, int M, std::vector<int> P, std::vector<int> A){ N_=N; vvi ch(N+M); REP(i,1,N+M)ch[P[i]].pb(i); vi prod(N+M,1); for(ll i=N-1;i>=0;i--){ prod[i]=ch[i].size(); for(ll x:ch[i])prod[i]=prod[i]*prod[x]%mod; } vi dp(N+M,1); rep(i,N){ vi tl(ch[i].size()+1,1),tr(ch[i].size()+1,1); rep(j,ch[i].size())tl[j+1]=tl[j]*prod[ch[i][j]]%mod; for(ll j=ch[i].size()-1;j>=0;j--)tr[j]=tr[j+1]*prod[ch[i][j]]%mod; rep(j,ch[i].size())dp[ch[i][j]]=dp[i]*tl[j]%mod*tr[j+1]%mod; } n=M; vi a(n),b(n); rep(i,n)a[i]=dp[N+i]; rep(i,n)b[i]=A[i]; seg.init(a,b); } int count_ways(int L, int R) { ll l=L-N_,r=R+1-N_; seg.upd(l,r); ll ans=seg.get(); if(ans<0)ans+=mod; return ans; }

Compilation message (stderr)

circuit.cpp: In member function 'void segtree::init(vi, vi)':
circuit.cpp:32:12: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |     while(N<a.size())N<<=1;
      |           ~^~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...