zoukankan      html  css  js  c++  java
  • 树状数组

    树状数组

    资料借鉴:

    https://www.luogu.com.cn/problemnew/solution/P3374

    适用范围

    单次查询时间复杂度:O(logN)

    区间和、区间异或和、区间乘积和静态RMQ

    支持单点、区间修改

    形式

    红点是树状数组,白点是原信息数组

    对于树状数组中的每一个红点i,应有:

    算上其本身的讯息,总共存储了2^k的白点信息,并且白点信息是连续的

    其中k表示,k取最大时,能使得x能被2^k整除

    观察上面红点,它有一个神奇的特点:

    对于每个i而言,一旦i被修改,i+2^k必会跟着一起进行相应变换

    使用

    1.k的计算(函数lowbit)

    x&-x计算2k,其中k表示,k取最大时,能使得x能被2k整除(这里要用到补码的知识进行证明,略)

    2.维护和初始化

    对于白点的原信息,我们可以看作是对位置i,进行了增加white[i]的维护

    所以我们该如何维护嘞?我们先看对一个点的维护

    利用上面红点带来的性质,我们不难发现,当i被修改,i+2k也被修改;同理i+2k被修改,那么...

    所以,我们很好地可以得到维护代码

    void renew(int x,int k)//x指当前修改的位置,k是维护值
    {
    	for(;x<=n;x+=(x&-x))tree[x]+=k;//这里维护为区段求和
    }
    

    3.查询

    对于树状数组的每一个点而言,它总共存储了2^k的原始点信息,并且他们是连续的

    为了查询点i的前缀和,而i又只能保存2^k的信息,

    所以我们将i减去2k,进一步去找点(i-2k)的前缀和,并不断重复上述过程

    查询代码:

    int query(int x)//查询位置为x的前缀和
    {
    	int t=0;//累加器
    	for(;x;x-=(x&-x))t+=tree[x];
    	return t;
    }
    

    RMQ

    用树状数组写RMQ实在没有优势

    存储空间上与线段树相当,建立的复杂度也相当,但是查询开销要到O(logn*logn),慢太多

    如果是追求效率查询又不如ST表

    有点鸡肋,贴一个大佬博客详细介绍树状数组求RMQ的方法

    大佬博客:https://blog.csdn.net/ljsspace/article/details/6674273

    原数组&&差分数组

    对于树状数组而言,查询和修改只能至多有一个是对区间进行的

    对于洛谷的两个模版而言,很好地诠释了这一点

    1,对单点修改,对区段查询,要将树状数组建立在原数组

    2,对区段修改,对单点查询,树状数组建立在差分数组

    如果操作都是对区间进行的,那么只能把线段树掏出来了


    洛谷树状数组模版1 P3374 对单点修改 对区段查询

    树状数组建立在原数组

    #include<iostream>
    #include<cstring>
    #include<cstdio>
    #define INF 0x3f3f3f3f
    #define maxn 500005
    #define minn -105
    #define ll long long int
    #define ull unsigned long long int
    #define uint unsigned int
    inline int read()
    {
        int ans=0;
        char last=' ',ch=getchar();
        while(ch<'0'|ch>'9')last=ch,ch=getchar();
        while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
        if(last=='-')ans=-ans;
    	return ans;
    }
    int tree[maxn];
    int n,m;
    void renew(int x,int k)
    {
        for(;x<=n;x+=(x&-x))tree[x]+=k;
    }
    int cal(int x)
    {
        int t=0;
        for(;x;x-=(x&-x))t+=tree[x];
        return t;
    }
    int main()
    {
    	n=read(),m=read();
    	memset(tree,0,sizeof(tree));
    	for(int i=1;i<=n;i++)
    	{
    	    int cur=read();
    	    renew(i,cur);
    	}
    	for(int i=1;i<=m;i++)
    	{
    	    int a=read(),b=read(),c=read();
    	    if(a==1)renew(b,c);
    	    else std::cout<<cal(c)-cal(b-1)<<'
    ';
    	}
    	return 0;
    }
    

    洛谷树状数组模版2 P3368 对区段修改 对单点查询

    树状数组建立在差分数组

    #include<iostream>
    #include<cstring>
    #include<cstdio>
    #include<queue>
    #define INF 0x3f3f3f3f
    #define maxn 500005
    #define minn -105
    #define ll long long int
    #define ull unsigned long long int
    #define uint unsigned int
    inline int read()
    {
        int ans=0;
        char last=' ',ch=getchar();
        while(ch<'0'|ch>'9')last=ch,ch=getchar();
        while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
        if(last=='-')ans=-ans;
        return ans;
    }
    int tree[maxn];
    int save[maxn];
    int n,m;
    void renew(int x,int k)
    {
        for(;x<=n;x+=(x&-x))tree[x]+=k;
    }
    void seg(int l,int r,int k)
    {
        renew(l,k);
        renew(r+1,-k);
    }
    int cal(int x)
    {
        int t=0;
        for(;x;x-=(x&-x))t+=tree[x];
        return t;
    }
    int main()
    {
    	n=read(),m=read();
    	memset(tree,0,sizeof(tree));
    	memset(save,0,sizeof(save));
    	int pre=0,temp=0;
    	for(int i=1;i<=n;i++)//读入,并且计算差分
    	{
    	    int cur=read();
    	    pre=cur;
    	    cur-=temp;
    	    renew(i,cur);
    	    temp=pre;
    	}
    	for(int i=1;i<=m;i++)
    	{
    	    int a,b,c;
    	    a=read();
    	    if(a==1)
    	    {
    	        a=read(),b=read(),c=read();
    	        seg(a,b,c);
    	    }
    	    else
    	    {
    	        a=read();
    	        std::cout<<cal(a)<<'
    ';
    	    }
    	}
    	return 0;
    }
    

    康托排序模版 P5367

    最开始我还以为二分能过,用了vector和lower_bound,忽然想起vector删除复杂度为n...

    其实也可以掏线段树,只不过我懒得写pushdown,lazy—tag这些

    so,下面是使用

    #include<iostream>
    #include<cstring>
    #include<cstdio>
    #define maxn 1000005
    #define ll long long int
    #define mod 998244353
    using namespace std;
    inline int read()
    {
        int ans=0;
        char last=' ',ch=getchar();
        while(ch<'0'|ch>'9')last=ch,ch=getchar();
        while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
        if(last=='-')ans=-ans;
        return ans;
    }
    ll n;
    ll a[maxn],A[maxn];
    ll ans(1);
    void add(ll x,ll k)
    {
        for(;x<=n;x+=(x&-x))a[x]+=k;
    }
    ll query(ll x)
    {
        ll t=0;
        for(;x;x-=(x&-x))t+=a[x];
        return t;
    }
    int main()
    {
        n=read();
        A[0]=1;
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
        {
            A[i]=A[i-1]*i%mod;
            add(i,1);
        }
        for(int i=1;i<=n;i++)
        {
            ll x=read();
            ans=(ans+(query(x)-1)*A[n-i])%mod;
            add(x,-1);
        }
        std::cout<<ans;
        return 0;
    }
  • 相关阅读:
    linux下进程权限分析
    基于Token的身份验证——JWT
    hadoop2.7.3+spark2.1.0+scala2.12.1环境搭建(3)http://www.cnblogs.com/liugh/p/6624491.html
    hadoop2.7.3+spark2.1.0+scala2.12.1环境搭建(2)安装hadoop
    hadoop2.7.3+spark2.1.0+scala2.12.1环境搭建(1)安装jdk
    配置ssh无密钥登陆
    Linux实战案例(7)安装jdk
    Scala入门(1)Linux下Scala(2.12.1)安装
    Linux实战案例(6)yum查找、卸载、和安装软件
    Linux实战案例(5)关闭Centos的防火墙
  • 原文地址:https://www.cnblogs.com/et3-tsy/p/12814240.html
Copyright © 2011-2022 走看看