zoukankan      html  css  js  c++  java
  • HDU-敌兵布阵(线段树 || 树状数组)

    ## 敌兵布阵

    Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
    Total Submission(s): 162203 Accepted Submission(s): 67064

    Problem Description
    C国的死对头A国这段时间正在进行军事演习,所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段,所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动,可能增加或减少若干人手,但这些都逃不过C国的监视。
    中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动,而Derek每次询问的段都不一样,所以Tidy不得不每次都一个一个营地的去数,很快就精疲力尽了,Derek对Tidy的计算速度越来越不满:"你个死肥仔,算得这么慢,我炒你鱿鱼!”Tidy想:“你自己来算算看,这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下,Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说:“死肥仔,叫你平时做多点acm题和看多点算法书,现在尝到苦果了吧!”Tidy说:"我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼,这么算他真的会崩溃的,聪明的读者,你能写个程序帮他完成这项工作吗?不过如果你的程序效率不够高的话,Tidy还是会受到Derek的责骂的.

    Input
    第一行一个整数T,表示有T组数据。
    每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
    接下来每行有一条命令,命令有4种形式:
    (1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)
    (2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);
    (3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;
    (4)End 表示结束,这条命令在每组数据最后出现;
    每组数据最多有40000条命令

    Output
    对第i组数据,首先输出“Case i:”和回车,
    对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

    Sample Input
    1
    10
    1 2 3 4 5 6 7 8 9 10
    Query 1 3
    Add 3 6
    Query 2 7
    Sub 10 2
    Add 6 3
    Query 3 10
    End

    Sample Output
    Case 1:
    6
    33
    59

    Author
    Windbreaker

    Recommend
    Eddy

    思路:

    最简单的单点修改,区间查询

    树状数组写法

    #include<bits/stdc++.h>
    using namespace std;
    const int N = 1e5+6;
    int n,m;
    int tr[N],a[N];
    int lowbit(int x){
    	return x & (-x);
    }
    void add(int x,int y){
    	while(x<=n){
    		tr[x]+=y;
    		x+=lowbit(x);
    	}
    //	for(int i=0;i<=x;i+=lowbit(i)) tr[i]+=y;
    }
    int sum(int x){
    	int res=0;
    	while(x>0){
    		res+=tr[x];
    		x-=lowbit(x);
    	}
    //	for(int i=x;i>0;i-=lowbit(i)) res+=tr[i];
    	return res;
    }
    int main(){
    	int t;
    	scanf("%d",&t);
    	for(int i=1;i<=t;i++){
    		cout<<"Case "<<i<<":"<<endl;
    		memset(tr,0,sizeof tr);
    		memset(a,0,sizeof a);
    	//	int n;
    		scanf("%d",&n);
    		for(int j=1;j<=n;j++){
    			scanf("%d",&a[j]);
    			add(j,a[j]);
    		} 
    		char s[10];
    		int x,y;
    		while(~scanf("%s",s)&&strcmp(s,"End")){
    			scanf("%d%d",&x,&y);
    			if(s[0]=='Q') cout<<sum(y)-sum(x-1)<<endl;
    			else if(s[0]=='A') add(x,y);
    			else if(s[0]=='S') add(x,-y);
    			//else break;
    		}
    	}
    	return 0;
    } 
    

    线段树写法

    #pragma GCC optimize(2)
    #include<iostream>
    #include<cstdio>
    #include<string>
    #include<ctime>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    #include <vector>
    #include<stdio.h>
    using namespace std;
    typedef long long ll;
    #define I_int ll
    inline int read()
    {
        int x=0,f=1;char ch=getchar();
        while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
        while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
        return x*f;
    }
    char F[200];
    inline void out(I_int x) {
        if (x == 0) return (void) (putchar('0'));
        I_int tmp = x > 0 ? x : -x;
        if (x < 0) putchar('-');
        int cnt = 0;
        while (tmp > 0) {
            F[cnt++] = tmp % 10 + '0';
            tmp /= 10;
        }
        while (cnt > 0) putchar(F[--cnt]);
        //cout<<" ";
    }
    const int maxn=51000;
    struct node{
        int l,r;
        int sum;///维护总人数
    }tr[maxn*4];
    int n,a[maxn];
    void pushup(int u){
        tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
    }
    void build(int u,int l,int r){
        if(l==r){
            tr[u].l=l;tr[u].r=r;tr[u].sum=a[r];
        }
        else{
            tr[u].l=l;tr[u].r=r;
            int mid=tr[u].l+tr[u].r >>1;
            build(u<<1,l,mid);
            build(u<<1|1,mid+1,r);
            pushup(u);
        }
    }
    void add(int u,int x,int y){
        if(tr[u].l==tr[u].r) tr[u].sum+=y;
        else{
            int mid= tr[u].l+tr[u].r >> 1;
            if(x<=mid) add(u<<1,x,y);
            else add(u<<1|1,x,y);
            pushup(u);
        }
    }
    int query(int u,int l,int r){
        if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
        int mid=tr[u].l + tr[u].r >> 1;
        int sum=0;
        if(l<=mid) sum=query(u<<1,l,r);
        if(r>mid) sum+=query(u<<1|1,l,r);
        return sum;
    }
    int main(){
        int t;t=read();
        for(int num=1;num<=t;num++){
           // memset(tr,0,sizeof tr);
           // memset(a,0,sizeof a);
            n=read();
            for(int i=1;i<=n;i++)
                a[i]=read();
            build(1,1,n);
            string op;
            int x,y;
            printf("Case %d:
    ",num);
            while(1){
                cin>>op;
                if(op=="End") break;
                else if(op=="Add"){
                    x=read();y=read();
                    add(1,x,y);
                }
                else if(op=="Sub"){
                    x=read();y=read();
                    add(1,x,-y);
                }
                else{
                    x=read();y=read();
                    printf("%d
    ",query(1,x,y));
                }
            }
        }
        return 0;
    }
    
  • 相关阅读:
    SpringBoot第四集:整合JDBC和JPA(2020最新最易懂)
    SpringBoot第六集:整合监听器/过滤器和拦截器(2020最新最易懂)
    SpringBoot第八集:静态资源与首页定制(2020最新最易懂)
    TM1621断码液晶驱动IC的原理、驱动代码
    代码编码格式批量转换工具
    LAN8720 调试笔记
    C# 读取电脑CPU、主板、硬盘序列号等信息
    C# 读取串口设备列表
    C# 窗口全屏、置顶、获取焦点
    STM32 获取系统时钟频率
  • 原文地址:https://www.cnblogs.com/OvOq/p/14853179.html
Copyright © 2011-2022 走看看