Submission #1110919

#TimeUsernameProblemLanguageResultExecution timeMemory
1110919koukirocksThe Potion of Great Power (CEOI20_potion)C++17
38 / 100
597 ms262144 KiB
#include <bits/stdc++.h> #define all(x) (x).begin(),(x).end() #define F first #define S second namespace { using namespace std; } typedef pair<int,int> pii; struct SEG { struct Node { vector<int> lz; int lft,rt; Node():lft(0),rt(0) {} }; int n; vector<Node> tree; SEG(int _n):n(_n) { tree.push_back(Node()); } int add() { tree.push_back(Node()); return tree.size()-1; } void update(int L,int R,int val,int l,int r,int id) { if (L<=l and r<=R) { tree[id].lz.push_back(val); return; } int mid=l+r>>1; if (L<=mid) { if (!tree[id].lft) tree[id].lft = add(); update(L,R,val,l,mid,tree[id].lft); } if (mid<R) { if (!tree[id].rt) tree[id].rt = add(); update(L,R,val,mid+1,r,tree[id].rt); } } void query(int t,vector<int> &h,int l,int r,int id) { for (int i:tree[id].lz) h.push_back(i); int mid=l+r>>1; if (t<=mid) { if (tree[id].lft) query(t,h,l,mid,tree[id].lft); } else { if (tree[id].rt) query(t,h,mid+1,r,tree[id].rt); } } }; namespace { vector<SEG> seg; int n,d,u; vector<int> h; } void init(int N, int D, int H[]) { n=N;d=D; h.resize(n); for (int i=0;i<n;i++) h[i]=H[i]; } void curseChanges(int U, int A[], int B[]) { u=U; seg.resize(n,SEG(u)); map<pii,int> tm; for (int i=0;i<u;i++) { if (A[i]>B[i]) swap(A[i],B[i]); if (tm[pair(A[i],B[i])]==0) tm[pair(A[i],B[i])]=i+1; else { seg[A[i]].update(tm[pair(A[i],B[i])],i,h[B[i]],1,u,0); seg[B[i]].update(tm[pair(A[i],B[i])],i,h[A[i]],1,u,0); tm[pair(A[i],B[i])]=0; } } for (auto [p,t]:tm) { if (t==0) continue; seg[p.F].update(t,u,h[p.S],1,u,0); seg[p.S].update(t,u,h[p.F],1,u,0); } } int question(int x, int y, int v) { vector<int> a,b; seg[x].query(v,a,1,u,0); seg[y].query(v,b,1,u,0); sort(all(a)); sort(all(b)); // for (int i:a) cout<<i<<" "; // cout<<"a\n"; // for (int i:b) cout<<i<<" "; // cout<<"b\n"<<flush; int ans=1e9; for (int i:a) { auto ptr = lower_bound(all(b),i); if (ptr!=b.end()) ans=min(ans,abs(*ptr-i)); if (ptr!=b.begin()) ans=min(ans,abs(*prev(ptr)-i)); } return ans; } /* 6 5 11 4 2 42 1000 54 68 234 0 1 2 0 3 4 3 5 3 5 1 3 5 3 0 5 3 0 1 3 3 5 0 3 4 26 3 0 8 0 0 5 5 1000000000 3 0 11 14 */

Compilation message (stderr)

potion.cpp: In member function 'void SEG::update(int, int, int, int, int, int)':
potion.cpp:30:18: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   30 |         int mid=l+r>>1;
      |                 ~^~
potion.cpp: In member function 'void SEG::query(int, std::vector<int>&, int, int, int)':
potion.cpp:42:18: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   42 |         int mid=l+r>>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...