码迷,mamicode.com
首页 > 编程语言 > 详细

算法笔记--树状数组

时间:2017-10-29 14:35:15      阅读:293      评论:0      收藏:0      [点我收藏+]

标签:while   blank   logs   clu   ++   http   lowbit   event   std   

区间和模板:

const int N=1e5+5;
int c[N];
int n;
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret=0;
    while(x)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
void update(int x,int d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}

1.单点更新,区间求和

HDU - 1166

技术分享
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e5+5;
int c[N];
int n;
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret=0;
    while(x)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
void update(int x,int d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T,a,l,r;
    int cnt=0;
    string s;
    cin>>T;
    while(T--)
    {
        cin>>n;
        mem(c,0); 
        for(int i=1;i<=n;i++)cin>>a,update(i,a);
        cout<<"Case "<<++cnt<<":"<<endl;
        while(cin>>s)
        {
            if(s=="End")break;
            if(s=="Query")
            {
                cin>>l>>r;
                cout<<sum(r)-sum(l-1)<<endl;
            }
            else if(s=="Add")
            {
                cin>>l>>r;
                update(l,r);
            }
            else if(s=="Sub")
            {
                cin>>l>>r;
                update(l,-r);
            }
        } 
    }
    return 0;
}
View Code

2.区间更新,单点求值

HDU - 1556

技术分享
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e5+5;
int c[N];
int n;
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret=0;
    while(x)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
void update(int x,int d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int l,r;
    while(cin>>n)
    {
        mem(c,0);
        for(int i=1;i<=n;i++)
        {
            cin>>l>>r;
            update(l,1);
            update(r+1,-1);
        } 
        for(int i=1;i<=n;i++)
        {
            cout<<sum(i);
            if(i!=n)cout<< ;
            else cout<<endl;
        } 
    }
    return 0;
}
View Code

参考博客:http://blog.csdn.net/yexiaohhjk/article/details/51077545

算法笔记--树状数组

标签:while   blank   logs   clu   ++   http   lowbit   event   std   

原文地址:http://www.cnblogs.com/widsom/p/7749930.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!