您的位置:首页 > 其它

HDU 1556

2015-08-18 11:24 246 查看
题解: 线段树或者是树状数组 线段树我用了两种方法试了试速度,时间差不多, 内存由于是先开的全局内存,也差不多。

线段树 两份代码一个最后做了维护,另外一个没有,时间上倒是差不多,维护为了最后方便输出。

树状数组有一个思路,就是 修改某个区间【a, b】 的时候, 把a的节点值加1, b+1的节点减1, 这样就成了区间求和,易于用树状数组。

线段树A代码, 最后做了个维护:

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int maxn=131072;

struct Node{
int l, r, num;
}tree[maxn*2];

int n, x, y;
int k;

void build(){
for(k=1;k<n;k<<=1);
int m=k<<1;
for(int i =k; i<m;++i){
tree[i].l=tree[i].r=i-k+1;
tree[i].num=0;
}

for(int i =k-1; i>0; --i){
tree[i].l=tree[i<<1].l;
tree[i].r=tree[i<<1|1].r;
tree[i].num=0;
}
}

void update(int L , int R, int o){

int l=tree[o].l;
int r=tree[o].r;
if(L==l && R==r)  { ++tree[o].num; return; }

int mid=(l+r)>>1;

if(R<=mid) update(L, R, o<<1);
else if(L>mid) update(L, R, o<<1|1);
else {
update(L, mid, o<<1);
update(mid+1, R, o<<1|1);
}
}

void maintain(int o, int val){
int l=tree[o].l;
int r=tree[o].r;
tree[o].num+=val;
if(l==r) return;

int mid=(l+r)>>1;

maintain(o<<1, tree[o].num);
maintain(o<<1|1, tree[o].num);
}

int main(){
while(scanf("%d", &n), n){

build();
for(int i =0; i<n; ++i){
scanf("%d%d", &x, &y);
update(x, y, 1);
}

maintain(1, 0);

for(int i=0; i<n;++i){
printf("%d%c", tree[i+k].num, i==n-1? 10: 32);
}
}
return 0;
}


线段树B代码, 最后没有维护,直接输出了:

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int maxn=131072;

struct Node{
int l, r, num;
}tree[maxn*4];

int n, x, y;
int k;

void build(int l, int r, int o){
tree[o].l=l;
tree[o].r=r;
tree[o].num=0;
if(l==r) return;
int mid=(l+r)>>1;
build(l, mid, o<<1);
build(mid+1, r, o<<1|1);
}

void update(int L , int R, int o){

int l=tree[o].l;
int r=tree[o].r;
if(L==l && R==r)  { ++tree[o].num; return; }

int mid=(l+r)>>1;

if(R<=mid) update(L, R, o<<1);
else if(L>mid) update(L, R, o<<1|1);
else {
update(L, mid, o<<1);
update(mid+1, R, o<<1|1);
}
}

void output(int o, int sum){
if(tree[o].l>n) return;
if(tree[o].l==tree[o].r) { printf("%d%c",sum+tree[o].num ,tree[o].r==n?10:32); return;}

output(o<<1, sum+tree[o].num);
output(o<<1|1, sum+tree[o].num);
}

int main(){
while(scanf("%d", &n), n){

build(1 ,n ,1);
for(int i =0; i<n; ++i){
scanf("%d%d", &x, &y);
update(x, y, 1);
}
output(1, 0);
}
return 0;
}


树状数组代码:

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

const int maxn= 132048;
int c[maxn];
int n ,x, y;

int lowbit(int x){ return x&(-x); }

void update(int x, int val){
while(x<=maxn){
c[x]+=val;
x+=lowbit(x);
}
}

int getsum(int x){
int sum=0;
while(x>0){
sum+=c[x];
x-=lowbit(x);
}
return sum;
}

int main(){
while(scanf("%d", &n), n){

memset(c, 0, sizeof(c));

for(int i = 0; i< n; ++i){
scanf("%d%d", &x, &y);
update(x, 1);
update(y+1, -1);
}
for(int i=1; i<=n;++i){
int nn=getsum(i);
printf("%d%c", nn, i==n? 10: 32);
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: