您的位置:首页 > 其它

hdu线段树专题训练

2012-09-30 10:52 375 查看



单点更新:

这是线段树中最基本的类型,只更新叶子节点,然后把信息用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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: