这是线段树中最基本的类型,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来。
hdu 1166 敌兵布阵
代码如下
#include<iostream> #include<stdlib.h> #include<cstdio> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 const int maxn=55555; int sum[maxn<<2]; void PushUP(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void build(int l,int r,int rt ) { if(l==r) { scanf("%d",&sum[rt]); return; } int m=(l+r)>>1; build(lson); build(rson); PushUP(rt); } void update(int p,int add,int l,int r,int rt) { if(l==r) { sum[rt]=sum[rt]+add; return; } int m=(l+r)>>1; if(p<=m) update(p,add,lson); else update(p,add,rson); PushUP(rt); } int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return sum[rt]; int m=(l+r)>>1; int ret=0; if(L<=m) ret=ret+query(L,R,lson); if(R>m) ret=ret+query(L,R,rson); return ret; } int main(void) { int T,n,t=1; int i,j; scanf("%d",&T); for(t=1;t<=T;t++) { printf("Case %d:\n",t); scanf("%d",&n); build(1,n,1); char s[10]; while(scanf("%s",s)) { if(s[0]=='E') break; int a,b; scanf("%d%d",&a,&b); if(s[0]=='Q') printf("%d\n",query(a,b,1,n,1)); else if(s[0]=='S') update(a,-b,1,n,1); else update(a,b,1,n,1); } } system("Pause"); return 0; }
hdu 1754 I Hate It
代码如下:
#include<iostream> #include<cstdio> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 const int maxn=222222; int MAX[maxn<<2]; void PushUP(int rt) { MAX[rt]=max(MAX[rt<<1],MAX[rt<<1|1]); } void build(int l,int r,int rt) { if(l==r) { scanf("%d",&MAX[rt]); return; } int m=(l+r)>>1; build(lson); build(rson); PushUP(rt); } void update(int p,int add,int l,int r,int rt) { if(l==r) { MAX[rt]=add; return; } int m=(l+r)>>1; if(p<=m) update(p,add,lson); else update(p,add,rson); PushUP(rt); } int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return MAX[rt]; int m=(l+r)>>1; int ret=0; if(L<=m) ret=max(ret,query(L,R,lson)); if(R>m) ret=max(ret,query(L,R,rson)); return ret; } int main(void) { int N,M; while(scanf("%d%d",&N,&M)!=EOF) { build(1,N,1); char ch[2]; int a,b; for(int i=1;i<=M;i++) { scanf("%s",ch); scanf("%d%d",&a,&b); if(ch[0]=='Q') printf("%d\n",query(a,b,1,N,1)); if(ch[0]=='U') update(a,b,1,N,1); } } return 0; }
hdu 1394 Minimum Inversion Number
题意:求Inversion后的最小逆序数
思路:用O(nlogn)复杂度求出最初逆序数后,就可以用O(1)的复杂度分别递推出其他解
线段树功能:update:单点增减 query:区间求和
代码如下:
#include<iostream> #include<cstdio> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 const int maxn=5500; int sum[maxn<<2]; void PushUP(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void build(int l,int r,int rt) { if(l==r) { sum[rt]=0; return; } int m=(l+r)>>1; build(lson); build(rson); PushUP(rt); } void update(int p,int l,int r,int rt) { if(l==r) { sum[rt]++; return ; } int m=(l+r)>>1; if(p<=m) update(p,lson); else update(p,rson); PushUP(rt); } int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return sum[rt]; int m=(l+r)>>1; int ret=0; if(L<=m) ret=ret+query(L,R,lson); if(R>=m) ret=ret+query(L,R,rson); return ret; } int main(void) { int i,n; while(scanf("%d",&n)!=EOF) { build(0,n-1,1); int a[maxn],sum=0; for(i=0;i<n;i++) { scanf("%d",&a[i]); sum=sum+query(a[i],n-1,0,n-1,1); update(a[i],0,n-1,1); } int ret=sum; for(i=0;i<n;i++) { sum+=n-a[i]-a[i]-1; ret=min(ret,sum); } printf("%d\n",ret); } return 0; }
hdu 2795 BillBoard
代码如下:
#include <cstdio> #include <algorithm> using namespace std; #define lson l , m , rt << 1 #define rson m + 1 , r , rt << 1 | 1 const int maxn = 222222; int h , w , n; int MAX[maxn<<2]; void PushUP(int rt) { MAX[rt] = max(MAX[rt<<1] , MAX[rt<<1|1]); } void build(int l,int r,int rt) { MAX[rt] = w; if (l == r) return ; int m = (l + r) >> 1; build(lson); build(rson); } int query(int x,int l,int r,int rt) { if (l == r) { MAX[rt] -= x; return l; } int m = (l + r) >> 1; int ret = (MAX[rt<<1] >= x) ? query(x , lson) : query(x , rson); PushUP(rt); return ret; } int main() { while (~scanf("%d%d%d",&h,&w,&n)) { if (h > n) h = n; build(1 , h , 1); while (n --) { int x; scanf("%d",&x); if (MAX[1] < x) puts("-1"); else printf("%d\n",query(x , 1 , h , 1)); } } return 0; }
poj 2828 Buy Tickets
题目大意:
这道题是从最后一个数开始建立线段树,区间存储的范围是该区间有多少个空位剩余,对于最后一个插入的数,它肯定和其插入的位置一致,例如对于测试数据的第一组,69插入2这个位置,是肯定的,它将前面占有2位置的数向后面推,同理对于倒数第二个数,占有1这个位置,如果前面有数占有1这个位置,此时前面的必定是向后面推。假设P为插入的位置,因次每次查看左区间有没有大于等于p的空位,如果有则搜索,否者搜索右孩子。
代码如下:
#include<iostream> #include<cstdio> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 const int maxn=200000; int sum[maxn<<2]; int n,cur,ans[maxn]; void PushUP(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void build(int l,int r,int rt) { if(l==r) { sum[rt]=1; return ; } int m=(l+r)>>1; build(lson); build(rson); PushUP(rt); } void update(int L,int R,int l,int r,int rt) { if(l==r) { sum[rt]--; ans[l]=cur; return ; } int m=(l+r)>>1; if(L<=sum[rt<<1]) update(L,R,lson); else { L-=sum[rt<<1]; update(L,R,rson); } PushUP(rt); } struct node { int p,v; }queue[maxn]; int main(void) { int i,j; while(scanf("%d",&n)!=EOF) { build(1,n,1); for(i=1;i<=n;i++) scanf("%d%d",&queue[i].p,&queue[i].v); for(i=n;i>0;i--) { cur=queue[i].v; update(queue[i].p+1,n,1,n,1); } for(i=1;i<=n;i++) { if(i-1) printf(" "); printf("%d",ans[i]); } printf("\n"); } return 0; }