处理区间问题之树状数组与莫队算法

最新推荐文章于 2024-06-01 09:45:47 发布
最新推荐文章于 2024-06-01 09:45:47 发布
阅读量275 收藏
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/the301stdoub/article/details/47748777
版权

感觉我直到前两天才算是真正学会用树状数组啊囧。。

有一类离线的区间询问问题，可以有以下解法（我暂时会这么多）：离线+树状数组||线段树||莫队算法

线段树先不说了，好久没写过了，而且本文的标题是树状数组与莫队囧。。

用树状数组的关键在于，每次插入一个点或者删除一个点要维护什么东西，一般是插入右端点删除左端点，维护的时候也要注意想清楚对哪部分答案有影响，删除之后要及时处理。

而莫队的不同就在于，离线的时候是分块处理询问的，然后每次只需要维护当前区间内的信息，大多数情况下需要维护的东西都比树状数组等少很多，因此想起来和写起来都非常舒服，这种情况下增加一个端点和删除一个端点要做的事情也不会很多。不过要注意的一点就是对于10w级别的数据，这里的增加和删除操作必须是O(1)才不会超时。

还是随便贴一下代码吧：

hdu4358

莫队

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<iomanip>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<list>

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;

#define rep(i,k,n) for(int i=(k);i<=(n);i++)
#define red(i,k,n) for(int i=(k);i>=(n);i--)
#define sqr(x) ((x)*(x))
#define clr(x,y) memset((x),(y),sizeof(x))
#define mod 1000000007
#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)>(b)?(b):(a))
const int MAXN = 100010;

int aa[MAXN],a[MAXN],l[MAXN],r[MAXN],p[MAXN];
int n,k,m,dfn,cnt;

struct Q
{
    int l,r,id,pos;
    bool operator < (const Q &a){return pos==a.pos?r<a.r:pos<a.pos;}
}q[MAXN];

int ans[MAXN];

vector<int> e[MAXN];

bool vis[MAXN];

void dfs(int u)
{
    l[u]=++dfn;
    a[dfn]=aa[u];
    vis[u]=1;
    rep(i,0,e[u].size()-1)
    {
        if(!vis[e[u][i]])
        {
            dfs(e[u][i]);
        }
    }
    r[u]=dfn;
}

bool cmp(int x,int y){return aa[x]<aa[y];}

int d[MAXN];

void dis()
{
    rep(i,1,n)d[i]=i;
    sort(d+1,d+1+n,cmp);
    int pre=aa[d[1]];
    aa[d[1]]=1;
    int acnt=1;
    rep(i,2,n)
    {
        if(aa[d[i]]==pre)aa[d[i]]=acnt;
        else {pre=aa[d[i]];aa[d[i]]=++acnt;}
    }
}

void add(int x)
{
    if(p[a[x]]==k-1)cnt++;
    if(p[a[x]]==k)cnt--;
    p[a[x]]++;
}

void del(int x)
{
    if(p[a[x]]==k+1)cnt++;
    if(p[a[x]]==k)cnt--;
    p[a[x]]--;
}

int main()
{
//#define LOCAL
#ifdef LOCAL
    freopen("e:\\read.txt","r",stdin);
    //freopen("e:\\write.txt","w",stdout);
#endif
    int T;
    cin>>T;
    rep(ii,1,T)
    {
        scanf("%d%d",&n,&k);
        int gn=sqrt(n*1.0);
        rep(i,1,n){scanf("%d",&aa[i]);e[i].clear();}
        dis();
        rep(i,1,n-1)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            e[u].push_back(v);
            e[v].push_back(u);
        }
        dfn=0;
        clr(vis,0);
        dfs(1);
        scanf("%d",&m);
        rep(i,1,m)
        {
            int qq;
            scanf("%d",&qq);
            q[i].l=l[qq];
            q[i].r=r[qq];
            q[i].id=i;
            q[i].pos=q[i].l/gn;
        }
        sort(q+1,q+1+m);
        clr(p,0);
        int ll=1,rr=0;
        cnt=0;
        rep(i,1,m)
        {
            while(ll<q[i].l)del(ll++);
            while(ll>q[i].l)add(--ll);
            while(rr>q[i].r)del(rr--);
            while(rr<q[i].r)add(++rr);
            ans[q[i].id]=cnt;
        }
        printf("Case #%d:\n",ii);
        rep(i,1,m)printf("%d\n",ans[i]);
        if(ii<T)puts("");
    }


    return 0;
}

树状数组

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<iomanip>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<list>

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;

#define rep(i,k,n) for(int i=(k);i<=(n);i++)
#define red(i,k,n) for(int i=(k);i>=(n);i--)
#define sqr(x) ((x)*(x))
#define clr(x,y) memset((x),(y),sizeof(x))
#define mod 1000000007
#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)>(b)?(b):(a))
const int MAXN = 100010;

int aa[MAXN],a[MAXN],l[MAXN],r[MAXN],nxt[MAXN];
int n,k,m,dfn,acnt;

struct Q
{
    int l,r,id;
    bool operator < (const Q &a){return r<a.r;}
}q[MAXN];

int ans[MAXN];

struct Bit
{
    int bit[MAXN];
    void cl(){clr(bit,0);}
    void add(int k,int v)
    {
        for(int i=k;i>0;i-=i&-i)bit[i]+=v;
    }
    int sum(int k)
    {
        int ret=0;
        for(int i=k;i<=n;i+=i&-i)ret+=bit[i];
        return ret;
    }
}B;

map<int,int> mp,L,R;
vector<int> e[MAXN];

bool vis[MAXN];

void dfs(int u)
{
    l[u]=++dfn;
    a[dfn]=aa[u];
    vis[u]=1;
    rep(i,0,e[u].size()-1)
    {
        if(!vis[e[u][i]])
        {
            dfs(e[u][i]);
        }
    }
    r[u]=dfn;
}

int main()
{
//#define LOCAL
#ifdef LOCAL
    freopen("e:\\read.txt","r",stdin);
    //freopen("e:\\write.txt","w",stdout);
#endif
    int T;
    cin>>T;
    rep(ii,1,T)
    {
        B.cl();
        scanf("%d%d",&n,&k);
        rep(i,1,n){scanf("%d",&aa[i]);e[i].clear();}
        mp.clear();
        rep(i,1,n-1)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            e[u].push_back(v);
            e[v].push_back(u);
        }
        dfn=0;
        clr(vis,0);
        dfs(1);
        scanf("%d",&m);
        rep(i,1,m)
        {
            int qq;
            scanf("%d",&qq);
            q[i].l=l[qq];
            q[i].r=r[qq];
            q[i].id=i;
        }
        sort(q+1,q+1+m);
        int t=1;
        L.clear();
        R.clear();
        clr(nxt,0);
        rep(i,1,n)
        {
            mp[a[i]]++;
            if(L[a[i]]==0)
            {
                L[a[i]]=i;
            }
            else
            {
                nxt[R[a[i]]]=i;
            }

            if(mp[a[i]]==k)
            {
                B.add(L[a[i]],1);
            }
            else if(mp[a[i]]>k)
            {
                B.add(L[a[i]],-2);
                L[a[i]]=nxt[L[a[i]]];
                B.add(L[a[i]],1);
            }
            R[a[i]]=i;
            while(t<=m&&i==q[t].r){ans[q[t].id]=B.sum(q[t].l);t++;}
            if(t>m)break;
        }
        printf("Case #%d:\n",ii);
        rep(i,1,m)printf("%d\n",ans[i]);
        if(ii<T)puts("");
    }


    return 0;
}

hdu4638

用莫队的话很多题都变成水题了。。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<iomanip>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<list>

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;

#define rep(i,k,n) for(int i=(k);i<=(n);i++)
#define red(i,k,n) for(int i=(k);i>=(n);i--)
#define sqr(x) ((x)*(x))
#define clr(x,y) memset((x),(y),sizeof(x))
#define mod 1000000007
#define MAX(a,b) ((a)>(b)?(a):(b))
#define MIN(a,b) ((a)>(b)?(b):(a))
const int MAXN = 100010;

int n,m,a[MAXN],ku,cnt;
bool in[MAXN];

struct Q
{
    int l,r,id,pos;
    bool operator < (const Q &a){return pos==a.pos?r<a.r:pos<a.pos;}
    void read(int d){scanf("%d%d",&l,&r);id=d;pos=l/ku;}
}q[MAXN];

int ans[MAXN];

void add(int x)
{
    if(in[x-1]&&in[x+1])cnt--;
    else if(!in[x-1]&&!in[x+1])cnt++;
    in[x]=true;
}

void del(int x)
{
    if(in[x-1]&&in[x+1])cnt++;
    else if(!in[x-1]&&!in[x+1])cnt--;
    in[x]=false;
}

int main()
{
//#define LOCAL
#ifdef LOCAL
    freopen("e:\\read.txt","r",stdin);
    //freopen("e:\\write.txt","w",stdout);
#endif
    int T;
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        rep(i,1,n)scanf("%d",&a[i]);
        ku=sqrt(n*1.0);
        rep(i,1,m)q[i].read(i);
        sort(q+1,q+1+m);
        int l=1,r=0;
        cnt=0;
        clr(in,false);
        rep(i,1,m)
        {
            while(r<q[i].r){r++;add(a[r]);}
            while(r>q[i].r){del(a[r]);r--;}
            while(l<q[i].l){del(a[l]);l++;}
            while(l>q[i].l){l--;add(a[l]);}
            ans[q[i].id]=cnt;
        }
        rep(i,1,m)printf("%d\n",ans[i]);
    }


    return 0;
}


窝嘞割草
关注
树状数组（Binary Indexed Tree，BIT）高效地处理动态的区间求和问题
06-08
树状数组 树状数组（Binary Indexed Tree，BIT）是一种数据结构，用于高效地处理动态的区间求和问题。它可以在 O(log n) 的时间复杂度内进行单点更新和区间查询操作。 树状数组的核心思想是利用二进制表示中的低位来表示区间的信息。每个节点存储一些区间的信息，而叶子节点则对应于原始数组中的单个元素。 树状数组的构建如下： 1. 初始化一个长度为 n 的数组 BIT，全部初始化为 0。 2. 对于每个数组元素 arr[i]，更新 BIT[i] += arr[i]。 3. 对于所有的 i，计算 BIT[i] += BIT[i - (i & -i)]，直到 i & -i 等于 0。 这样构建完成后，BIT 中的每个元素表示的是原始数组中一系列元素的区间和。 树状数组的应用包括但不限于： 1. 单点更新、区间查询：可以通过更新单点来修改原始数组，并且可以高效地查询任意区间的和。 2. 求逆序对：通过修改排序算法，可以在 O(n log n) 的时间复杂度内求出一个数组中的逆序对数。 3. 离散化：将一个区间的值映射到一个连续的区间上，使得区间和的计算更为方便。
莫队算法
The Shawshank Redemption
12-23 268
题目大意：给你n个数，有q个询问，每次询问[L,R]区间内有多少个不同的数。 思考1：区间问题，线段树？但是对于线段树来说如何用区间合并更新答案呢？我们可以在每个节点处申请一个用于保存的数组，然后记录下出现的值以及其出现位置的左右端点，但显而易见，把问题更复杂化了。 思考2：冗余计算在哪里？首先看看如果我们最暴力的扫描一遍能得到什么。我们可以统计出有多少个不同的数以及他们各自出现了多少次。现在...
BZOJ 3289 莫队算法+树状数组
Desico的博客
03-27 301
大意是询问将每个询问区间里的数恢复成有序最少需要的移动次数（只能相邻的两个数交换）学习莫队算法的练习题。俗话说得好啊 莫队就是分块之后暴力乱搞233假设已知[L,R]区间需要交换的次数，计算[L-1,R]需要交换的次数即为在[L-1,R]区间内有几个数比a[L-1]小，只需要加上这些数的个数即可。同理计算[L,R+1]找的是比a[R+1]大的数的个数。注意区间扩张和缩小时区间长度的小细节。利用树状...
HDU4638 Group （线段树，莫队算法）
moon_NO2015的博客
08-18 669
 题意：给出一个数组，每次查询l,r，看区间内能分成几个连续的数列也即时求求一个区间中连续ID区间的个数。 需要离线处理所有查询。我们从左往右扫描，依次考虑右端点为i的询问，设dp[l]为区间[l，i]的连续区间个数，po[i]为i出现的位置，若还未出现，则为 0，设我们当前考虑的右端点为a[i]，首先我们假设a[i]不能和区间[1,i-1]中的任何一个数分到一组，则我们要将dp[1]到d
莫队算法（区间处理）
weixin_34174132的博客
12-28 121
莫队算法（区间处理） 0x00 概论 莫队算法主要是用于离线解决通常不带修改只有查询的一类区间问题。 以前遇到区间问题的时候一般都是用线段树解决，当然能用线段树解决的问题也在多数。线段树的主要思路就是通过一个左半拉序列和一个右半拉序列来维护它们的父亲（也就是两条序列接合在一起的完整序列），通过层层递归下去从而完成对整体序列的维护。但在有些时候可能会遇到如下的问题： 给...
求区间不同数的个数 树状数组||莫队算法
weixin_30666943的博客
07-24 208
题目：https://www.nowcoder.com/acm/contest/139/J题意：给出n个数，求 [1,L],[R,n]这两个区间不同数的个数其实你只要把区间扩大一倍，就是求 [R,L+n]这个区间了 求区间内不同数的个数解决方法有很多 像用离线树状数组、离线莫队、线段树、主席树等等 不过听说主席树会TLE，所以这里主要说一下树状数组和莫队算法1.树状数组（BIT）其实可以用树...
树状数组之区间最值
DC_Fan的博客
02-09 1340
树状数组之区间最值 原理 数学原理： 建立树状数组 利用上面的性质，在树状数组的尾部插入数据，来建立一个树状数组 void push(int pos){ int i,lb = lowbit(pos); c[pos] = a[pos]; for(i=1;i<lb;i <<=1){ c[pos] = max(c[pos],c[pos-i]); } } 树的维护 void update(int pos,int v){ i
算法--树状数组与线段树
程天爽的博客
03-03 1004
树状数组和线段树概念、模板代码及相应练习题
树状数组（区间修改，区间查询）
hacker_51的博客
12-04 797
本题链接：用户登录 树状数组的区间修改，区间查询，操作起来有点繁琐了些，但还是可以理解的。这里区间修改，我们还是需要需要通过差分来进行区间修改的。不同的是，在于区间查询这一块。我们先模拟一遍区间修改后，差分数组 dif，原数组 a得： 差分数组的前缀和，就是获得当前原数组的数值 而我们需要求的是区间和，所以当我们设区间和数组为 sum可知 所以，当我们区间查询区间求和的时候可以得出，我们区间求和，相应的差
树状数组维护区间最值问题
fuhaitao9999的博客
11-05 366
使用树状数组需要理解其具体原理。，记住其维护的区间是。
算法NOIP树状数组校门外的树.pdf
05-02
树状数组是一种用于处理数组中连续区间更新和查询前缀和的高效数据结构。在数据量大、更新频繁的情况下，传统的暴力方法会因为时间复杂度过高而不适用。树状数组的核心思想是通过特定的二进制索引操作，将一个一维...
树状数组_洛谷_树状数组_
10-02
树状数组，也被称为线段树（Segment Tree），是一种数据结构，主要用于高效地处理动态区间查询和修改问题。在编程竞赛和算法设计中，树状数组是解决许多问题的利器，尤其是在面对需要频繁更新和查询区间信息的问题时...
c++/c语言之用树状数组来进行（区间修改和询问）
LovelyCat_mm的博客
06-01 1003
我们以上图为例，tree[1]=C1=A1，tree[2]=C2=C1+A2，tree[3]=C3=A3，tree[4]=C4=C2+C3+A4，tree[5]=C5=A5，tree[6]=C5+A6，tree[7]=A7，tree[8]=C4+C6+C7+A8。其实这个名字看起来还挺高级的，但相信经过我的介绍，大家可以都听得明白，嗯呐，接下来，我们先详细而系统性地介绍一下树状数组的定义，概念，以及最为重要的编码和应用。然后去掉最后的一个1，得到000，就会惊奇的发现，没有了，那就直接return。
树状数组求区间最值
UniverseofHK的博客
06-29 5094
闲谈一下 树状数组最基本的功能是加速前缀和的更新。 求一个数组的前缀和本来是O(n)的复杂度，用树状数组则为O(nlogn)。 但树状数组优点在于单点更新时复杂度为O(logn)，而正常的为O(n)，这也就使得树状数组能够进行大规模的更新。 虽然查询速度(O(logn))稍有些慢(相对于O(n)而言)，但依旧可以用于大规模的查询。 总之，遇到有不断更新的数组前缀和时可以考虑树状数组。 树状数组求区...
5116-微信小程序电影院订票选座系统设计及实现+ssm（源码+数据库+lun文）.zip
09-15
本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。
JavaScript 中的 `Array.prototype.filter` 方法全解析
09-15
在 JavaScript 编程中，处理数组是一项基本而重要的任务。数组的过滤操作是其中的一个常见需求，Array.prototype.filter 方法为此提供了强大的支持。本文将深入探讨 Array.prototype.filter 方法的工作原理、使用场景、代码示例以及与其他数组方法的比较。 Array.prototype.filter 是 JavaScript 中处理数组的强大工具，它允许开发者以声明式的方式轻松筛选出符合特定条件的元素。通过结合使用 Array.prototype.filter 和其他数组方法，可以解决各种复杂的数据筛选问题。 通过本文的详细介绍和示例代码，你应该能够掌握 Array.prototype.filter 的工作原理，并能够在实际开发中灵活运用它来处理数组数据。此外，了解其与 Array.prototype.map 和 Array.prototype.reduce 的结合使用，可以帮助你更好地编写高效且易于维护的代码。
5108-微信小程序的书橱+ssm（源码+数据库+lun文）.zip
09-15
本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。
5046-微信小程序校园二手交易平台的小程序+ssm（源码+数据库+lun文）.zip
09-15
本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。本系统主要针对计算机相关专业的正在做毕业设计的学生和需要项目实战练习的学习者，可作为毕业设计、课程设计、期末大作业。
SMT小型视觉贴片机控制系统源码源代码图纸 DI
最新发布
09-15
SMT小型视觉贴片机控制系统源码源代码图纸 DI
树状数组详解与应用：解决区间求和与修改问题
"本文详细介绍了树状数组的原理，并用树状数组解决‘校门外的树’这个问题。" 树状数组是一种高效的数据结构，主要...在实际应用中，树状数组常常出现在解决动态区间问题的竞赛题目中，是编程竞赛选手必备的技能之一。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值