이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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;
vi graph[N];
int ni[N],szl[N],szn[N],tin[N],tout[N],heavy[N],pa[N],anc[N],cn[N];
int timer,ans,r,n,q,k,h;
struct SegTree{
SegTree *left,*right;
int l,r,c,lz;
SegTree(){}
SegTree(int l,int r):l(l),r(r),c(0),lz(0),left(nullptr),right(nullptr){
if(l==r)return;
int 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(int i,int 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(int u,int p=0){
szl[u]=(sz(graph[u])==1);
h+=(sz(graph[u])==1);
szn[u]=1;
for(int 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(int v:graph[u]){
if(szn[heavy[u]]<szn[v]){
heavy[u]=v;
}
}
}
void dfs1(int u,int p,int 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(int v:graph[u]){
if(v==p||v==heavy[u])continue;
dfs1(v, u, v);
}
}
void update(int 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){
int 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;
vi u(k);
fo(i,k)cin>>u[i];
int 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();
}
컴파일 시 표준 에러 (stderr) 메시지
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(int, int)':
cleaning.cpp:46:15: warning: 'SegTree::lz' will be initialized after [-Wreorder]
46 | int 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(int l,int r):l(l),r(r),c(0),lz(0),left(nullptr),right(nullptr){
| ^~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |