[vijos 1066和1052] 简单树状数组练习题 标签: vijos 2017-05-25 09:33 46人阅读 评论(0)

打鼹鼠
背景

SuperBrother在机房里闲着没事干(再对比一下他的NOIP,真是讽刺啊……),于是便无聊地开始玩“打鼹鼠”……
描述

在这个“打鼹鼠”的游戏中,鼹鼠会不时地从洞中钻出来,不过不会从洞口钻进去(鼹鼠真胆大……)。洞口都在一个大小为n(n<=1024)的正方形中。这个正方形在一个平面直角坐标系中,左下角为(0,0),右上角为(n-1,n-1)。洞口所在的位置都是整点,就是横纵坐标都为整数的点。而SuperBrother也不时地会想知道某一个范围的鼹鼠总数。这就是你的任务。
格式

输入格式

每个输入文件有多行。
第一行,一个数n,表示鼹鼠的范围。
以后每一行开头都有一个数m,表示不同的操作:
m=1,那么后面跟着3个数x,y,k(0<=x,y

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n;
int c[1025][1025];
int lowbit(int x){
    return x&-x;
}
void update(int x,int y,int k){//二维
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=n;j+=lowbit(j))
            c[i][j]+=k;
}
int  getsum(int x,int y){
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            sum+=c[i][j];
    return sum;
} 
int main(){
    scanf("%d",&n);
    int m;
    while(scanf("%d",&m)){
        if(m==3) return 0;  
        if(m==1){
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            update(x+1,y+1,k);//从0开始 改为从1开始就行
        }   
        if(m==2){
            int x1,x2,y1,y2;
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
    int ans=getsum(x2+1,y2+1)-getsum(x2+1,y1)getsum(x1,y2+1)+getsum(x1,y1);
    //就是求矩形面积 自己画个图就ok 
            cout<<ans<<endl;
        }
    }   
    return 0;
}

弱弱的战壕
描述
永恒和mx正在玩一个即时战略游戏,名字嘛~~恕本人记性不好,忘了-_-b。
mx在他的基地附近建立了n个战壕,每个战壕都是一个独立的作战单位,射程可以达到无限(“mx不赢定了?!?”永恒ftING…@_@)。
但是,战壕有一个弱点,就是只能攻击它的左下方,说白了就是横纵坐标都不大于它的点(mx:“我的战壕为什么这么菜”ToT)。这样,永恒就可以从别的地方进攻摧毁战壕,从而消灭mx的部队。
战壕都有一个保护范围,同它的攻击范围一样,它可以保护处在它左下方的战壕。所有处于它保护范围的战壕都叫做它的保护对象。这样,永恒就必须找到mx的战壕中保护对象最多的点,从而优先消灭它。
现在,由于永恒没有时间来计算,所以拜托你来完成这个任务:
给出这n个战壕的坐标xi、yi,要你求出保护对象个数为0,1,2……n-1的战壕的个数。
格式

输入格式

第一行,一个正整数n(1<=n<=15000)
接下来n行,每行两个数xi,yi,代表第i个点的坐标
(1<=xi,yi<=32000)
注意:可能包含多重战壕的情况(即有数个点在同一坐标)
输出格式

输出n行,分别代表保护对象为0,1,2……n-1的战壕的个数。
样例1
样例输入1
5
1 1
5 1
7 1
3 3
5 5
样例输出1
1
2
1
1
0
限制
各点2s(算是宽限吧^_^)
来源
URAL1028战役版

//开二维会炸内存 开一维加排序就行了
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int c[32005];
int ans[15005];
struct node{
    int x,y; 
}p[15005];
bool cmp(node xx,node yy){
    if(xx.y!=yy.y) return xx.y<yy.y;
    else return xx.x<yy.x;
}
int lowbit(int x){
    return x&-x;
}
void update(int x){
    for(;x<=32000;x+=lowbit(x))
        c[x]++;
}
int getsum(int x){
    int sum=0;
    for(;x;x-=lowbit(x))
        sum+=c[x];
    return sum;
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&p[i].x,&p[i].y);//按y从小到大进行排序 y相等时x从小到大排 
    sort(p+1,p+1+n,cmp);
    for(int i=1;i<=n;i++){
        int temp=getsum(p[i].x);
//因为已经进行了排序 所以y一定是大于以前的战壕的 这个时候找x小于当前点的就可以了 
        ans[temp]++;
        update(p[i].x);//存入树状数组 
    }
    for(int i=0;i<=n-1;i++)
        printf("%d
",ans[i]);

    return 0;
}

线段树做法

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n,ans[15000];
struct node{int r,l,v;}tr[100001];
struct data{int x,y;}f[15000];
inline bool cmp(data a,data b){return (a.x==b.x)?(a.y<b.y):(a.x<b.x);}
void buildtree(int size,int lf,int rt){
    tr[size].v=0;
    tr[size].l=lf;
    tr[size].r=rt;
    if(lf==rt)return; 
    int mid=(lf+rt)>>1;
    buildtree(size<<1,lf,mid);
    buildtree(size<<1|1,mid+1,rt);
}
void insert(int st,int ed,int size)
{
    int mid=(tr[size].l+tr[size].r)>>1;
    if(tr[size].l==st&&tr[size].r==ed)
    {
        tr[size].v++;
        return;
    }
    else if(tr[size].r==tr[size].l)return;
    else if(ed<=mid)insert(st,ed,size<<1);
    else if(st>mid)insert(st,ed,size<<1|1);
    else
    {
        insert(st,mid,size<<1);
        insert(mid+1,ed,size<<1|1);
    }
}
int query(int x,int size){
    if(tr[size].l==tr[size].r)return tr[size].v;
    int mid=(tr[size].l+tr[size].r)>>1;
    if(x<=mid)return tr[size].v+(query(x,size<<1));
    else return tr[size].v+(query(x,size<<1|1));
}
int main()
{
    scanf("%d",&n);
    buildtree(1,0,32001);
    for(int i=0;i<n;i++)
        scanf("%d%d",&f[i].x,&f[i].y);
    sort(f,f+n,cmp);
    for(int i=0;i<n;i++)
    {
        ans[query(f[i].y,1)]++;//找的时候往前找
        insert(f[i].y,32001,1);
/*注意:区间是f[i]到32001的原因是 后面的战壕可以管道当前的战壕 所以加后面的 不能写成0到f[i] 这样显然是错的啊*/
    }
    for(int i=0;i<n;i++)
        printf("%d
",ans[i]);
    return 0;
}
原文地址:https://www.cnblogs.com/xljxlj/p/7183644.html