BZOJ 4538: [Hnoi2016]網路
阿新 • • 發佈:2018-12-21
這個題目就是特別裸啊,很明顯就是先樹鏈剖分,然後線上段樹每個節點上維護兩個堆,來維護插入和刪除,查詢的時候就暴力查就好了。
似乎很簡單啊,我竟然在luogu上1A了,結果交到bzoj上MLE,看過討論後把找重兒子改成大於等於就A了。。
不過我感覺這樣是假的啊,一條鏈線上段樹上有\(\log^2n\)段,每個暴力插的話就是\(O(n\log^3n)\)啊,但是還跑的挺快的,這卡的也太不滿了吧。。
#include<cstdio> #include<cstring> #include<algorithm> #include<cmath> #include<queue> #define qmin(x,y) (x=min(x,y)) #define qmax(x,y) (y=max(x,y)) using namespace std; typedef long long ll; const int Maxn=410000; int to[Maxn],nxt[Maxn],first[Maxn],tot=1; int tl[Maxn],tr[Maxn],u[Maxn],v[Maxn],w[Maxn]; int dep[Maxn],fa[Maxn],ran[Maxn],top[Maxn],son[Maxn]; int n,m,x,y,opt,cnt; priority_queue<int> t[Maxn][2]; struct node{ int l,r; }s[Maxn]; inline void add(int u,int v) { to[tot]=v; nxt[tot]=first[u]; first[u]=tot++; to[tot]=u; nxt[tot]=first[v]; first[v]=tot++; } int dfs1(int root) { int siz=1,sizm=0; for(int i=first[root];i;i=nxt[i]) if(!dep[to[i]]) { dep[to[i]]=dep[root]+1; fa[to[i]]=root; int temp=dfs1(to[i]); siz+=temp; if(temp>=sizm) { sizm=temp; son[root]=to[i]; } } return siz; } void dfs2(int root,int topp) { top[root]=topp;ran[root]=++cnt; if(son[root]) dfs2(son[root],topp); for(int i=first[root];i;i=nxt[i]) if(!ran[to[i]]) dfs2(to[i],to[i]); } void build(int root,int l,int r) { tl[root]=l,tr[root]=r; int mid=l+r>>1; if(l==r) return; build(root<<1,l,mid); build((root<<1)|1,mid+1,r); } int gettop(int root) { while(!t[root][1].empty()&&t[root][0].top()==t[root][1].top()) t[root][1].pop(),t[root][0].pop(); return t[root][0].empty()?-1:t[root][0].top(); } int query(int root,int x) { int l=tl[root],r=tr[root]; int mid=l+r>>1; int ans=gettop(root); if(l==r) return ans; if(x<=mid) return max(ans,query(root<<1,x)); else return max(ans,query((root<<1)|1,x)); } void add(int root,int l,int r,int opt,int x) { int lc=tl[root],rc=tr[root]; int mid=lc+rc>>1; if(l<=lc&&r>=rc) { t[root][opt].push(x); return ; } if(l<=mid) add(root<<1,l,r,opt,x); if(r>mid) add((root<<1)|1,l,r,opt,x); } int cmp(node a,node b) { return a.l<b.l; } void change(int x,int y,int w,int opt) { int topp=0; while(top[x]!=top[y]) { if(dep[top[x]]<dep[top[y]]) swap(x,y); s[++topp]=(node){ran[top[x]],ran[x]}; x=fa[top[x]]; } if(dep[x]>dep[y]) swap(x,y); s[++topp]=(node){ran[x],ran[y]}; sort(s+1,s+topp+1,cmp); int zhy=0; for(int i=1;i<=topp;zhy=s[i++].r) if(zhy+1<s[i].l) add(1,zhy+1,s[i].l-1,opt,w); if(zhy<n) add(1,zhy+1,n,opt,w); } int main() { // freopen("test.in","r",stdin); scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { scanf("%d%d",&x,&y); add(x,y); }dep[1]=1; dfs1(1),dfs2(1,1); build(1,1,n); for(int i=1;i<=m;i++) { scanf("%d",&opt); if(opt==2) { scanf("%d",&x); printf("%d\n",query(1,ran[x])); } else if(opt==1) { scanf("%d",&x); change(u[x],v[x],w[x],opt); } else { scanf("%d%d%d",&u[i],&v[i],&w[i]); change(u[i],v[i],w[i],opt); } } return 0; }