oj.uz

Submission #1039336

# Submission time Handle Problem Language Result Execution time Memory
1039336 2024-07-30T18:28:41 Z vjudge1 Spring cleaning (CEOI20_cleaning) C++17
9 / 100
 168 ms 262144 KB 
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#pragma GCC target("popcnt")
using namespace std;
 
using ll = long long;
using ull = unsigned long long;
using lld = long double;
using vi = vector<int>;
using vll = vector<ll>;
using ii = pair<int,int>;
using pll = pair<ll, ll>;
using vii = vector<ii>;
using vpll = vector<pll>;
 
#define endl '\n'
#define all(x) x.begin(),x.end()
#define lsb(x) x&(-x)
#define gcd(a,b) __gcd(a,b)
#define sz(x) (int)x.size()
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define fls cout.flush()
 
#define fore(i,l,r) for(auto i=l;i<r;i++)
#define fo(i,n) fore(i,0,n)
#define forex(i,r,l) for(auto i=r; i>=l;i--)
#define ffo(i,n) forex(i,n-1,0)
 
bool cmin(int &a, int b){if(b<a){a=b;return 1;}return 0;}
bool cmax(int &a, int b){if(b>a){a=b;return 1;}return 0;}
void valid(ll in){cout<<((in)?"YES\n":"NO\n");}
ll lcm(ll a, ll b){return (a/gcd(a,b))*b;}
ll gauss(ll n){return (n*(n+1))/2;}
const int N = 2e5 + 7, LOG=20;
vll graph[N];
ll ni[N],szl[N],szn[N],tin[N],tout[N],heavy[N],pa[N],anc[N],cn[N];
ll timer,ans,r,n,q,k,h;
struct SegTree{
    SegTree *left,*right;
    ll l,r,c,lz;
    SegTree(){}
    SegTree(ll l,ll r):l(l),r(r),c(0),lz(0),left(nullptr),right(nullptr){
        if(l==r)return;
        ll m=(l+r)/2;
        left=new SegTree(l, m);
        right=new SegTree(m+1,r);
    }
    void push(){
        if(lz)c=r-l+1-c;
        if(l!=r){
            left->lz^=lz;
            right->lz^=lz;
        }
        lz=0;
    }
    void update(ll i,ll j){
        push();
        if(l>j||r<i)return;
        if(l>=i&&r<=j){
            lz^=1;
            push();
            return;
        }
        left->update(i,j);
        right->update(i,j);
        c=left->c+right->c;
    }
};
SegTree *root;
void dfs(ll u,ll p=0){
    szl[u]=(sz(graph[u])==1);
    h+=(sz(graph[u])==1);
    szn[u]=1;
    for(ll v:graph[u]){
        if(v==p)continue;
        ni[v]=ni[u]+1;
        pa[v]=u;
        dfs(v, u);
        szl[u]+=szl[v];
        szn[u]+=szn[v];
    }
    for(ll v:graph[u]){
        if(szn[heavy[u]]<szn[v]){
            heavy[u]=v;
        }
    }
}
void dfs1(ll u,ll p,ll anct){
    tin[u]=timer++;
    if(sz(graph[u])>1)dfs1(heavy[u], u, anct);
    anc[u]=anct;
    if(szl[u]&1)root->update(tin[u],tin[u]);
    for(ll v:graph[u]){
        if(v==p||v==heavy[u])continue;
        dfs1(v, u, v);
    }
}
void update(ll u){
    while(u != 0){
        root->update(tin[anc[u]], tin[u]);
        u=pa[anc[u]];
    }
}
void test_case(){
    cin>>n>>q;
    fo(i,n-1){
        ll a,b;
        cin>>a>>b;
        cn[a]++;
        cn[b]++;
        graph[a].pb(b);
        graph[b].pb(a);
    }
    fore(i,1,n+1)if(sz(graph[i])>1)r=i;
    root=new SegTree(0, n-1);
    dfs(r);
    dfs1(r, 0, r);
    while(q--){
        cin>>k;
        vll u(k);
        fo(i,k)cin>>u[i];
        ll ans=0;
        fo(i,k){
            cn[u[i]]++;
            if(cn[u[i]]==2)continue;
            h++;
            update(u[i]);
        }
        root->push();
        ans=2*n+k-2-root->c;
        if(h&1)cout<<-1<<endl;
        else cout<<ans<<endl;
        fo(i,k){
            cn[u[i]]--;
            if(cn[u[i]]==1)continue;
            h--;
            update(u[i]);
        }
    }
}
int main(){cin.tie(0)->sync_with_stdio(0);
    int t=1;
    // cin >> t;
    while(t--)test_case();
}

Compilation message

cleaning.cpp:4: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    4 | #pragma GCC optimization ("O3")
      | 
cleaning.cpp:5: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    5 | #pragma GCC optimization ("unroll-loops")
      | 
cleaning.cpp: In constructor 'SegTree::SegTree(ll, ll)':
cleaning.cpp:46:14: warning: 'SegTree::lz' will be initialized after [-Wreorder]
   46 |     ll l,r,c,lz;
      |              ^~
cleaning.cpp:45:14: warning:   'SegTree* SegTree::left' [-Wreorder]
   45 |     SegTree *left,*right;
      |              ^~~~
cleaning.cpp:48:5: warning:   when initialized here [-Wreorder]
   48 |     SegTree(ll l,ll r):l(l),r(r),c(0),lz(0),left(nullptr),right(nullptr){
      |     ^~~~~~~

Subtask #1
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5212 KB Output is correct
2 Runtime error 142 ms 262144 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 6748 KB Output is correct
2 Correct 37 ms 6744 KB Output is correct
3 Correct 47 ms 28620 KB Output is correct
4 Correct 66 ms 22724 KB Output is correct
5 Correct 79 ms 29684 KB Output is correct

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 40 ms 7768 KB Output is correct
2 Runtime error 138 ms 262144 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #4
0 / 16.0

# Verdict Execution time Memory Grader output
1 Runtime error 148 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #5
0 / 19.0

# Verdict Execution time Memory Grader output
1 Runtime error 151 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Runtime error 168 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #7
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5212 KB Output is correct
2 Runtime error 142 ms 262144 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -