Submission #1298473

#TimeUsernameProblemLanguageResultExecution timeMemory
1298473trandangquangTourism (JOI23_tourism)C++20
10 / 100
3469 ms14388 KiB
#include<bits/stdc++.h> using namespace std; #define foru(i,a,b) for(int i=(a); i<=(b); ++i) #define ford(i,a,b) for(int i=(a); i>=(b); --i) #define rep(i,a) for(int i=0; i<(a); ++i) #define sz(a) (int)(a).size() #define all(a) (a).begin(),(a).end() #define bit(s,i) (((s)>>(i))&1) #define ii pair<int,int> #define vi vector<int> #define vii vector<ii> #define fi first #define se second #define ll long long #define eb emplace_back #define pb push_back #define __builtin_popcount __builtin_popcountll #define _ << " " << template <class X, class Y> bool maxi(X &x, Y y){return x<y?x=y,true:false;} template <class X, class Y> bool mini(X &x, Y y){return x>y?x=y,true:false;} const int N=101010; const int B=2000; const int LG=17; int n,m,q,c[N],ans[N],h[N]; int tin[N],tout[N],tour[N],par[N],minH[LG+1][N],tim; vi adj[N]; struct Query{ int l,r,id; }; vector<Query> que[N]; void dfs(int u, int p=-1){ tin[u]=++tim; tour[tim]=u; for(int v:adj[u]) if(v!=p){ h[v]=h[u]+1; par[v]=u; dfs(v,u); } tout[u]=tim; } void buildLCA(){ foru(i,1,n){ minH[0][i]=tour[i]; } for(int i=1; 1<<i <= n; ++i){ for(int j=1; j+(1<<i)-1 <= n; ++j){ if(h[minH[i-1][j]] < h[minH[i-1][j+(1<<(i-1))]]){ minH[i][j]=minH[i-1][j]; } else{ minH[i][j]=minH[i-1][j+(1<<(i-1))]; } } } } int get(int l, int r){ int lg=__lg(r-l+1); if(h[minH[lg][l]] < h[minH[lg][r-(1<<lg)+1]]) return minH[lg][l]; return minH[lg][r-(1<<lg)+1]; } int lca(int u, int v){ if(u==v) return u; if(tin[u]>tin[v]) swap(u,v); return par[get(tin[u]+1,tin[v])]; } int dist(int u, int v){ return h[u]+h[v]-2*h[lca(u,v)]; } int nxt[N],prv[N],mostL,mostR,cost; int turn[N]; void del(int x){ int pos=tin[c[x]]; --turn[pos]; if(turn[pos]==0){ cost-=dist(tour[mostL], tour[mostR]); int l=prv[pos], r=nxt[pos]; if(l!=-1) cost-=dist(tour[l], tour[pos]), nxt[l]=r; if(r!=-1) cost-=dist(tour[pos],tour[r]), prv[r]=l; if(l!=-1 && r!=-1) cost+=dist(tour[l],tour[r]); if(nxt[l]==-1) mostR=l; if(prv[r]==-1) mostL=r; cost+=dist(tour[mostL],tour[mostR]); } } vector<pair<int&, int>> vt; void delRoll(int x){ int pos=tin[c[x]]; vt.eb(turn[pos],turn[pos]); --turn[pos]; if(turn[pos]==0){ vt.eb(cost,cost); cost-=dist(tour[mostL], tour[mostR]); int l=prv[pos], r=nxt[pos]; if(l!=-1) cost-=dist(tour[l], tour[pos]), vt.eb(nxt[l],nxt[l]), nxt[l]=r; if(r!=-1) cost-=dist(tour[pos],tour[r]), vt.eb(prv[r],prv[r]), prv[r]=l; if(l!=-1 && r!=-1) cost+=dist(tour[l],tour[r]); if(nxt[l]==-1) vt.eb(mostR,mostR), mostR=l; if(prv[r]==-1) vt.eb(mostL,mostL), mostL=r; cost+=dist(tour[mostL],tour[mostR]); } } void roll(){ while(vt.size()){ vt.back().fi=vt.back().se; vt.pop_back(); } } void process(){ foru(i,0,n/B){ sort(all(que[i]),[](Query x, Query y){return x.r>y.r;}); cost=0; foru(j,1,n) turn[j]=0; foru(j,i*B,m-1) turn[tin[c[j]]]++; int cur=-1; foru(j,1,n) if(turn[j]>0){ if(cur==-1) mostL=j; prv[j]=cur; cur=j; } cur=-1; ford(j,n,1) if(turn[j]>0){ if(cur==-1) mostR=j; nxt[j]=cur; cur=j; if(nxt[j]!=-1){ cost+=dist(tour[j],tour[nxt[j]]); } } cost+=dist(tour[mostL],tour[mostR]); int r=m-1; for(auto j:que[i]){ while(r>j.r) del(r--); int l=i*B; while(l<j.l) delRoll(l++); ans[j.id]=cost; roll(); } } } void solve(){ cin>>n>>m>>q; rep(i,n-1){ int u,v; cin>>u>>v; adj[u].eb(v); adj[v].eb(u); } rep(i,m) cin>>c[i]; rep(i,q){ int l,r; cin>>l>>r; --l, --r; que[l/B].pb({l,r,i}); } dfs(1); buildLCA(); process(); rep(i,q) cout<<ans[i]/2+1<<'\n'; } int32_t main(){ #define task "test" if(fopen(task".inp", "r")){ freopen(task".inp", "r", stdin); freopen(task".out", "w", stdout); } cin.tie(0)->sync_with_stdio(0); int tc=1; //cin>>tc; foru(i,1,tc){ solve(); } }

Compilation message (stderr)

tourism.cpp: In function 'int32_t main()':
tourism.cpp:188:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  188 |         freopen(task".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
tourism.cpp:189:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  189 |         freopen(task".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
#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...