首页 > 代码库 > BZOJ 3110 ZJOI 2013 K大数查询 树套树(权值线段树套区间线段树)
BZOJ 3110 ZJOI 2013 K大数查询 树套树(权值线段树套区间线段树)
题目大意:有一些位置,这些位置上可以放若干个数字。现在有两种操作。
1.在区间l到r上添加一个数字x
2.求出l到r上的第k大的数字是什么
思路:这种题一看就是树套树,关键是怎么套,怎么写。(话说我也不会来着。。)最容易想到的方法就是区间线段树套一个权值线段树,但是区间线段树上的标记就会变得异常复杂。所以我们就反过来套,用权值线段树套区间线段树。这样修改操作在外线段树上就变成了单点修改,外线段树就不用维护标记了。在里面的区间线段树上维护标记就容易多了。具体实现见代码。
CODE:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define MAX 50010 #define CNT (r - l + 1) using namespace std; int total,asks; struct ValSegTree{ ValSegTree *son[2]; int cnt,c; ValSegTree() { son[0] = son[1] = NULL; cnt = c = 0; } void PushDown(int k) { if(son[0] == NULL) son[0] = new ValSegTree(); if(son[1] == NULL) son[1] = new ValSegTree(); if(c) { son[0]->cnt += c * (k - (k >> 1)); son[0]->c += c; son[1]->cnt += c * (k >> 1); son[1]->c += c; c = 0; } } void Modify(int l,int r,int x,int y) { if(l == x && r == y) { cnt += CNT; c++; return ; } PushDown(CNT); int mid = (l + r) >> 1; if(y <= mid) son[0]->Modify(l,mid,x,y); else if(x > mid) son[1]->Modify(mid + 1,r,x,y); else { son[0]->Modify(l,mid,x,mid); son[1]->Modify(mid + 1,r,mid + 1,y); } cnt = son[0]->cnt + son[1]->cnt; } int Ask(int l,int r,int x,int y) { if(!cnt) return 0; if(l == x && r == y) return cnt; PushDown(CNT); int mid = (l + r) >> 1; if(y <= mid) return son[0]->Ask(l,mid,x,y); if(x > mid) return son[1]->Ask(mid + 1,r,x,y); int left = son[0]->Ask(l,mid,x,mid); int right = son[1]->Ask(mid + 1,r,mid + 1,y); return left + right; } }; struct IntSegTree{ IntSegTree *son[2]; ValSegTree *root; IntSegTree() { son[0] = son[1] = NULL; root = new ValSegTree(); } void Modify(int l,int r,int _l,int _r,int x) { root->Modify(1,total,_l,_r); if(l == r) return ; int mid = (l + r) >> 1; if(son[0] == NULL) son[0] = new IntSegTree(); if(son[1] == NULL) son[1] = new IntSegTree(); if(x <= mid) son[0]->Modify(l,mid,_l,_r,x); else son[1]->Modify(mid + 1,r,_l,_r,x); } int Ask(int l,int r,int _l,int _r,int k) { if(l == r) return l; int mid = (l + r) >> 1; if(son[0] == NULL) son[0] = new IntSegTree(); if(son[1] == NULL) son[1] = new IntSegTree(); int temp = son[1]->root->Ask(1,total,_l,_r); if(k > temp) return son[0]->Ask(l,mid,_l,_r,k - temp); else return son[1]->Ask(mid + 1,r,_l,_r,k); } }root; int main() { cin >> total >> asks; for(int flag,x,y,z,i = 1;i <= asks; ++i) { scanf("%d%d%d%d",&flag,&x,&y,&z); if(flag == 1) root.Modify(1,total,x,y,z); else printf("%d\n",root.Ask(1,total,x,y,z)); } return 0; }
BZOJ 3110 ZJOI 2013 K大数查询 树套树(权值线段树套区间线段树)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。