zoukankan      html  css  js  c++  java
  • BZOJ4825 [Hnoi2017]单旋 【线段树】

    题目链接

    BZOJ4825

    题解

    手模一下操作,会发现一些很优美的性质:
    每次旋到根,只有其子树深度不变,剩余点深度(+1)
    每次旋到根,【最小值为例】右儿子接到其父亲的左儿子,其余点形态不改变,然后将该点接到根之上,原根变为其右儿子
    每次插入,都是插入到其前驱后继深度较大的那一个点之下

    所以我们很容易模拟出树的形态,同时用线段树维护离散化后各权值的深度

    #include<iostream>
    #include<cstdio>
    #include<cmath>
    #include<set>
    #include<cstring>
    #include<algorithm>
    #define LL long long int
    #define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
    #define REP(i,n) for (int i = 1; i <= (n); i++)
    #define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
    using namespace std;
    const int maxn = 100005,maxm = 100005,INF = 1000000000;
    inline int read(){
    	int out = 0,flag = 1; char c = getchar();
    	while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
    	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    	return out * flag;
    }
    set<int> S;
    struct Que{
    	int opt,v;
    }Q[maxn];
    int fa[maxn],ls[maxn],rs[maxn];
    int b[maxn],tot,m;
    int getn(int x){return lower_bound(b + 1,b + 1 + tot,x) - b;}
    int val[maxn << 2],tag[maxn << 2];
    void pd(int u){
    	if (tag[u]){
    		val[u << 1] += tag[u]; tag[u << 1] += tag[u];
    		val[u << 1 | 1] += tag[u]; tag[u << 1 | 1] += tag[u];
    		tag[u] = 0;
    	}
    }
    void change(int u,int l,int r,int pos,int v){
    	if (l == r) {val[u] = v; return;}
    	pd(u);
    	int mid = l + r >> 1;
    	if (mid >= pos) change(u << 1,l,mid,pos,v);
    	else change(u << 1 | 1,mid + 1,r,pos,v);
    }
    void modify(int u,int l,int r,int L,int R,int v){
    	if (l >= L && r <= R) {val[u] += v; tag[u] += v; return;}
    	pd(u);
    	int mid = l + r >> 1;
    	if (mid >= L) modify(u << 1,l,mid,L,R,v);
    	if (mid < R) modify(u << 1 | 1,mid + 1,r,L,R,v);
    }
    int query(int u,int l,int r,int pos){
    	if (l == r) return val[u];
    	pd(u);
    	int mid = l + r >> 1;
    	if (mid >= pos) return query(u << 1,l,mid,pos);
    	return query(u << 1 | 1,mid + 1,r,pos);
    }
    void print(int u,int l,int r){
    	if (l == r) printf("%d ",val[u]);
    	else {
    		pd(u);
    		int mid = l + r >> 1;
    		print(u << 1,l,mid);
    		print(u << 1 | 1,mid + 1,r);
    	}
    }
    int main(){
    	m = read(); int n = 0;
    	REP(i,m){
    		Q[i].opt = read();
    		if (Q[i].opt == 1) b[++n] = Q[i].v = read();
    	}
    	sort(b + 1,b + 1 + n); tot = 1;
    	for (int i = 2; i <= n; i++) if (b[i] != b[tot]) b[++tot] = b[i];
    	S.insert(0); S.insert(tot + 10);
    	int pre,nxt,d1,d2,u,v,rt;
    	for (int i = 1; i <= m; i++){
    		if (Q[i].opt == 1){
    			u = getn(Q[i].v);
    			S.insert(u);
    			pre = *--S.find(u);
    			nxt = *++S.find(u);
    			if (pre == 0 && nxt == tot + 10)
    				change(1,1,tot,u,1),rt = u;
    			else if (pre == 0){
    				d1 = query(1,1,tot,nxt);
    				change(1,1,tot,u,d1 + 1);
    				fa[u] = nxt; ls[nxt] = u;
    			}
    			else if (nxt == tot + 10){
    				d2 = query(1,1,tot,pre);
    				change(1,1,tot,u,d2 + 1);
    				fa[u] = pre; rs[pre] = u;
    			}
    			else {
    				d1 = query(1,1,tot,nxt);
    				d2 = query(1,1,tot,pre);
    				if (d1 > d2){
    					change(1,1,tot,u,d1 + 1);
    					fa[u] = nxt; ls[nxt] = u;
    				}
    				else {
    					change(1,1,tot,u,d2 + 1);
    					fa[u] = pre; rs[pre] = u;
    				}
    			}
    			ls[u] = rs[u] = 0;
    			printf("%d
    ",query(1,1,tot,u));
    		}
    		else if (Q[i].opt == 2){
    			u = *++S.find(0);
    			printf("%d
    ",query(1,1,tot,u));
    			if (!fa[u]) continue;
    			modify(1,1,tot,fa[u],tot,1);
    			change(1,1,tot,u,1);
    			v = fa[u];
    			ls[v] = rs[u]; if (rs[u]) fa[rs[u]] = v;
    			rs[u] = rt; fa[rt] = u; fa[u] = 0;
    			rt = u;
    		}
    		else if (Q[i].opt == 3){
    			u = *--S.find(tot + 10);
    			printf("%d
    ",query(1,1,tot,u));
    			if (!fa[u]) continue;
    			modify(1,1,tot,1,fa[u],1);
    			change(1,1,tot,u,1);
    			v = fa[u];
    			rs[v] = ls[u]; if (ls[u]) fa[ls[u]] = v;
    			ls[u] = rt; fa[rt] = u; fa[u] = 0;
    			rt = u;
    		}
    		else if (Q[i].opt == 4){
    			u = *++S.find(0);
    			printf("%d
    ",query(1,1,tot,u));
    			if (fa[u]) modify(1,1,tot,u,fa[u] - 1,-1);
    			else modify(1,1,tot,1,tot,-1);
    			if (v = fa[u]){
    				ls[v] = rs[u]; if (rs[u]) fa[rs[u]] = v;
    				rs[u] = 0;
    			}
    			else if (rs[u]) fa[rs[u]] = 0,rt = rs[u],rs[u] = 0;
    			S.erase(u);
    		}
    		else if (Q[i].opt == 5){
    			u = *--S.find(tot + 10);
    			printf("%d
    ",query(1,1,tot,u));
    			if (fa[u]) modify(1,1,tot,fa[u] + 1,u,-1);
    			else modify(1,1,tot,1,tot,-1);
    			if (v = fa[u]){
    				rs[v] = ls[u]; if (ls[u]) fa[ls[u]] = v;
    				ls[u] = 0;
    			}
    			else if (ls[u]) fa[ls[u]] = 0,rt = ls[u],ls[u] = 0;
    			S.erase(u);
    		}
    		/*printf("rt = %d
    ",b[rt]);
    		for (int i = 1; i <= tot; i++){
    			printf("node%d  fa = %d  ,ls = %d  rs = %d
    ",b[i],b[fa[i]],b[ls[i]],b[rs[i]]);
    		}
    		print(1,1,tot); puts("");*/
    	}
    	return 0;
    }
    
    
  • 相关阅读:
    python图像仿射变换实战-提取手写数字图片样本
    详解增强算术赋值:“-=”操作是怎么实现的?
    Pycharm激活码,2020最新Pycharm永久激活码!
    Python爬虫视频教程百度云网盘免费分享
    尚学堂高淇Python400集全套视频教程百度云网盘分享
    Python入门教程哪个好?看看这套怎么样!
    python框架Django实战商城项目之工程搭建
    Python入门书籍电子版PDF百度云网盘免费下载
    廖雪峰老师Python教程及配套视频教程
    manjaro(arch)里的vbox 安装centos7后,centos无法联网的解决办法
  • 原文地址:https://www.cnblogs.com/Mychael/p/9017742.html
Copyright © 2011-2022 走看看