zoukankan      html  css  js  c++  java
  • 莫队算法

    莫队算法

    基本莫队算法介绍

    莫队算法是一个分块算法,是由某国家集训队大佬提出的一个算法,我们就用一道 SDOI2009 的题来引入吧。

    HH的项链

    大概概括一下题意就是我们给定一个长度为 \(n\) 的自然数数列 \(a_1,a_2,a_3......a_{n-1},a_n\)。有 \(q\) 次询问,每次给出 \(l,r\),我们需要回答在 \([l,r]\) 内有多少不同的数。其中,\(n\le 5\cdot 10^4,m\le 2\cdot 10^5,a_i\le 10^6\)(这是原数据的范围,洛谷上有所加强)。

    这就是题目的样例,那么我们知道,最暴力的方法就是对于每一次询问清空累加数组,每次都重新统计,时间复杂度为 \(O(q\cdot n)\)

    那么我们再加一个小优化,我们知道,对于第 \(2,3\) 次询问来说,他们有重复部分 \([3,5]\)。那么这个部分我们就不用重复处理,只需要加上 \([2,2],[6,6]\) 两个区间即可。

    但是这样时间复杂度依然还是 \(O(q\cdot n)\)。我们如果造了这样一组数据,使得相邻两次的询问区间都没有交集,这样的时间复杂度反而还不如之前的暴力(指常数)。因此,我们一定需要一种较为稳定,且时间复杂度小的一种算法,也就是莫队。

    众所周知,莫队是一个离线的算法。我们先对询问排序,这样可以让算法的时间复杂度大大退化,达到 \(O(q\cdot \sqrt{n}+ n\cdot \sqrt{n})\)

    莫队算法引用了分块的思想,我们定义:一个连续的数列称为一个块。 一个块的长度就是这个块所包含的数的个数。所以一个长度为 \(n\) 的数列可以分成至少 \(\lfloor\frac{n}{\lfloor \sqrt{n}\rfloor}\rfloor\) 个长度为 \(\lfloor \sqrt{n}\rfloor\) 的块,还有一个长度小于 \(\sqrt{n}\) 的块(这个块的长度可以为0)。注意我们接下来的讨论全部忽视这个小块,其涉及到的常数问题不讨论,且认为 \(\lfloor\frac{n}{\lfloor \sqrt{n}\rfloor}\rfloor=\sqrt{n}\) 这会大大减少我的Latex公式编辑量。这样我们的时间复杂度和后续一些问题会比较好理解一些。毕竟这点误差还是在我们的接受范围内的。除非你要做 POJ 上的题,老爷机会让你崩溃的

    咳咳,我们回到正题。我们说一下排序的方式,然后再说一下为什么这样排序。我们先把数列分成 \(\sqrt{n}\) 个 长度为 \(\sqrt{n}\) 的块,如图。再说一遍:这里认为 \(\sqrt{n}=2\),但是这里有三个块,我们后面仍然认为只有 \(\sqrt{n}\)

    分好块了,我们现在说一下排序的细则:我们以询问区间的左端点 \(l\) 所在的块的编号为第一关键字,以右端点 \(r\) 的编号为第二关键字进行排序(注意有些人学的莫队是以右端点 \(r\) 所在的块的编号为第二关键字的,这个在部分情况下会有略微差别,一般认为是卡常,因为时间复杂度差别不是很大,本文中无说明情况下,以 \(r\) 的编号为第二关键字)。对样例进行完了排序后就是这样的一幅图:

    可能这个样例看着不是很明显优化在哪里,但是如果我们来推算一下时间复杂度,就知道这个算法的优点了。

    我们首先考虑左端点的移动情况,由于我们排过序,所有在一个块内的移动最多 \(\sqrt{n}\),因为一个块的长度为 \(\sqrt{n}\)。如果是跨块的移动,可能为 \(n\),但这是unsustainable不可持续的,一旦移动到了下一个块中,就不会后移,所以跨块操作的平均情况应该是 \(\sqrt{n}\)。所以左端点的移动复杂度就为 \(O(q\sqrt{n})\)
    我们再来看右端点,对于右端点来说,她是按编号排的序。所以对于左端点在于一个块中的询问来说,右端点的极端情况是从 \(1\) 一直移动到 \(n\)。我们总共有 \(\sqrt{n}\) 个块,所以与端点的比较好估算,是 \(O(n\sqrt{n})\)
    所以莫队算法的总时间复杂度就是 \(O(n\sqrt{n}+q\sqrt{n})\),可以通过此题。但由于洛谷上此题数据有所加强,无法使用莫队通过,需要用树状数组的方法来做,这里不再赘述这个问题。莫队的代码见下:

    想要测这道题的同学可以到这里来测试

    #include <bits/stdc++.h>
    
    using namespace std;
    
    const int maxn=5e4+10;
    
    struct pr {
    	int l,r,id,bel;
    }q[maxn*20];
    
    int ans[maxn*20],a[maxn*20],n,q,num[1000010],cur;
    
    bool cmp(pr x,pr y) {
    	if(x.bel!=y.bel) {
    		return x.bel<y.bel;
    	}
    	return x.r<y.r;
    }
    
    void add(int x) {
    	num[x] ++;
    	if (num[x] == 1)
    		cur ++;
    }
    
    void del(int x) {
    	num[x] --;
    	if (num[x] == 0)
    		cur --;
    }
    
    int main() {
    	
    	cin >> n;
    	int blocks = sqrt(n);
    	
    	for (int i = 1; i <= n; ++i) {
    		scanf("%d", &a[i]);
    	}
    	cin >> q;
    	for (int i = 1; i <= q; ++i) {
    		q[i].id = i;
    		scanf("%d %d", &q[i].l, &q[i].r);
    		q[i].bel = (q[i].l - 1) / blocks  + 1;//计算左端点所在的块
    	}
    	sort(q + 1, q + q + 1, cmp);
    	
    	
    	int l = 1, r = 0;
    	
    	for (int i = 1; i <= q; ++i) {
    		while (l < q[i].l) del(a[l++]);//这些是 l,r 的移动
    		while (l > q[i].l) add(a[--l]);
    		while (r < q[i].r) add(a[++r]);
    		while (r > q[i].r) del(a[r--]);
    		ans[q[i].id] = cur;//统计情况;
    	}
    	
    	for (int i =1 ; i <= q; ++i) {
    		printf("%d\n",ans[i]);
    	}
    	return 0;
    }
    

    但可能小伙伴们还有一些问题,比如说一下两个最有代表性:

    Q1: 为什么选取块的长度为 \(\sqrt{n}\)
    A1: 解答这个问题,我们要先搞清一点,\(q\sqrt{n},n\sqrt{n}\) 中的 \(\sqrt{n}\) 含义不同。一个是块长,一个是块的个数。我们现在设块长为 \(k\)。那么时间复杂度为 \(O(qk+n\cdot \frac{n}{k})\)。可以用基本不等式得到:\(qk+n\cdot \frac{n}{k}\geq 2\sqrt{qn^2}\),也可以知道当 \(qk=n\cdot \frac{n}{k}\) 时,即 \(k=\frac{n\sqrt{q}}{q}\) 时,可以取到最小值 \(2\sqrt{qn^2}\)。所以最理想的块长为 \(\frac{n\sqrt{q}}{q} \approx \sqrt{n}\),时间复杂度最小为 \(2\sqrt{qn^2}\approx n\sqrt{n}+q\sqrt{n}\)

    我们可以来测一下这两种的区别:

    \(\sqrt{n}\) 版:

    \(\frac{n\sqrt{q}}{q}\) 版:

    可见没什么大区别。所以我们一般采取长度为 \(\sqrt{n}\) 的块即可。

    Q2: 为什么要分块,为何不能以左端点的编号为第一关键字?
    A2: 我们可以这样想,以左端点为第一关键字的话,相当于块的长度 \(k=1\)。根据上面的推导知道这样时间复杂度是 \(O(q+n^2)\)。所以不可以的。

    [国家集训队]小Z的袜子

    那么我们再来看这样一道题。这道题明显是上一题的一个变形,只是我们要把上面的 \(\operatorname{add},\operatorname{del}\) 函数稍做调整。\(\operatorname{add}\) 就是把目前相同袜子颜色对答案贡献加上,把原来的减掉,\(\operatorname{del}\) 也一样。我们就可以得到以下代码:

    void add(int x) {
    	cur-=num[x]*(num[x]-1);
    	num[x] ++;
    	cur+=num[x]*(num[x]-1);
    }
    
    void del(int x) {
    	cur-=num[x]*(num[x]-1);
    	num[x] --;
    	cur+=num[x]*(num[x]-1);
    }
    

    注意分数要约分即可。

    带修莫队

    莫队这种算法,一定会有修改呀。如果我们在查询的时候有修改,怎么办呢?我们的询问顺序打乱了,怎么处理修改呢?

    [国家集训队]数颜色

    我们还是以一道例题引入。题目自己看一看,我把样例的图放上来,具体就不解释了:

    注意,我们这里改一改原来题目中的字母,在代码中不改。我们定义有 \(k\) 次操作,有 \(m\) 个修改操作,有 \(q\) 个询问操作。所以有 \(m+q=k\)

    其实我们很容易想到,我们记录一下每次询问前有多少次修改操作,因为我们不改变修改的顺序,这样,我们每次还要记录一下当前维护的区间进行了几次修改,然后我们只需要根据这些信息,可以把已进行的修改改回去,或者再继续改下去。

    这道题洛谷的数据有所加强,原数据评测请到这里

    慢着,这样就可以了吗?这样可以通过原数据,但是在洛谷上加强的数据无法通过。所以我们要想办法继续优化我们的算法。我们先来算一算时间复杂度。

    左端点和右端点的移动仍然相同。我们算一下修改的时间复杂度。我们完全可以设计一种数据,先进行 \(\frac{q}{2}\) 次询问,然后集中进行 \(m\) 次修改,最后在把之前的询问再问一遍,这样你每次就要来回的修改,时间复杂度为 \(O(qm)\)。总时间复杂度为 \(O(q\sqrt{n}+n\sqrt{n}+qm)\)。更极限情况下,\(q=m=\frac{n}{2}\)。这样时间复杂度为 \(O(\frac{3\cdot n\sqrt{n}}{2}+\frac{n^2}{4})\)。显然炸掉。

    我们的已解决办法就是来改一下块长,故技重施。我们还是取最极端的情况 \(q=m=\frac{n}{2}\)。此时设块长为 \(l\)\(k\) 已经有意义了,所以换一个参数),最好的 \(l=\sqrt[3]{n^2}\)。这样就可以通过此题。另外,此题最好以右端点 \(r\) 所在块为第二关键字。这样在本题中发挥更好。(最后就是要加上一个火车头,开个 O2 才能通过)。

    #pragma GCC diagnostic error "-std=c++11"
    #pragma GCC target("avx")
    #pragma GCC optimize(3)
    #pragma GCC optimize("Ofast")
    #pragma GCC optimize("inline")
    #pragma GCC optimize("-fgcse")
    #pragma GCC optimize("-fgcse-lm")
    #pragma GCC optimize("-fipa-sra")
    #pragma GCC optimize("-ftree-pre")
    #pragma GCC optimize("-ftree-vrp")
    #pragma GCC optimize("-fpeephole2")
    #pragma GCC optimize("-ffast-math")
    #pragma GCC optimize("-fsched-spec")
    #pragma GCC optimize("unroll-loops")
    #pragma GCC optimize("-falign-jumps")
    #pragma GCC optimize("-falign-loops")
    #pragma GCC optimize("-falign-labels")
    #pragma GCC optimize("-fdevirtualize")
    #pragma GCC optimize("-fcaller-saves")
    #pragma GCC optimize("-fcrossjumping")
    #pragma GCC optimize("-fthread-jumps")
    #pragma GCC optimize("-funroll-loops")
    #pragma GCC optimize("-fwhole-program")
    #pragma GCC optimize("-freorder-blocks")
    #pragma GCC optimize("-fschedule-insns")
    #pragma GCC optimize("inline-functions")
    #pragma GCC optimize("-ftree-tail-merge")
    #pragma GCC optimize("-fschedule-insns2")
    #pragma GCC optimize("-fstrict-aliasing")
    #pragma GCC optimize("-fstrict-overflow")
    #pragma GCC optimize("-falign-functions")
    #pragma GCC optimize("-fcse-skip-blocks")
    #pragma GCC optimize("-fcse-follow-jumps")
    #pragma GCC optimize("-fsched-interblock")
    #pragma GCC optimize("-fpartial-inlining")
    #pragma GCC optimize("no-stack-protector")
    #pragma GCC optimize("-freorder-functions")
    #pragma GCC optimize("-findirect-inlining")
    #pragma GCC optimize("-fhoist-adjacent-loads")
    #pragma GCC optimize("-frerun-cse-after-loop")
    #pragma GCC optimize("inline-small-functions")
    #pragma GCC optimize("-finline-small-functions")
    #pragma GCC optimize("-ftree-switch-conversion")
    #pragma GCC optimize("-foptimize-sibling-calls")
    #pragma GCC optimize("-fexpensive-optimizations")
    #pragma GCC optimize("-funsafe-loop-optimizations")
    #pragma GCC optimize("inline-functions-called-once")
    #pragma GCC optimize("-fdelete-null-pointer-checks")//壮观
    #include<bits/stdc++.h>
    #define int long long
    using namespace std;
    
    int read() {
    	char ch=getchar();
    	int f=1,x=0;
    	while(ch<'0'||ch>'9') {
    		if(ch=='-')
    			f=-1;
    		ch=getchar();
    	}
    	while(ch>='0'&&ch<='9') {
    		x=x*10+ch-'0';
    		ch=getchar();
    	}
    	return f*x;
    }
    
    const int maxn=133334;
    
    struct query {
    	int l,r,id,bell,belr,num;
    }q[maxn];
    
    struct modify {
    	int x,val,pre;
    }c[maxn]; 
    
    bool cmp(query x,query y) {
    	if(x.bell!=y.bell) {
    		return x.bell<y.bell;
    	}
    	if(x.belr!=y.belr)
    		return x.belr<y.belr;
    	return x.num<y.num;
    }
    
    int n,ans[maxn],m,a[maxn],cur,b[maxn],num[1000010];
    
    void add(int x) {
    	num[x]++;
    	if(num[x]==1) {
    		cur++;
    	}
    }
    void del(int x){
    	num[x]--;
    	if(num[x]==0) {
    		cur--;
    	}
    }
    
    signed main() {
    
    	n=read();m=read();
    	int block=pow(n,2.0/3),mnum=0,cntq=0,cntm=0;
    	fill(ans,ans+1+m,-1);
    	
    	for(int i=1;i<=n;i++) {
    		a[i]=read();
    		b[i]=a[i];
    	}
    	for(int i=1;i<=m;i++) {
    		char s;
    		cin>>s;
    		if(s=='Q') {
    			q[++cntq].l=read();
    			q[cntq].r=read();
    			q[cntq].bell=(q[cntq].l-1)/block+1;
    			q[cntq].belr=(q[cntq].r-1)/block+1;
    			q[cntq].id=i;
    			q[cntq].num=mnum;
    		}
    		else {
    			c[++cntm].x=read();
    			c[cntm].val=read();
    			c[cntm].pre=b[c[cntm].x];
    			b[c[cntm].x]=c[cntm].val;
    			mnum++;
    		}
    	}
    	int l=1,r=0,ch=0;
    	sort(q+1,q+cntq+1,cmp);
    	for(int i=1;i<=cntq;i++) {
    		while(l<q[i].l) del(a[l++]);
            while(l>q[i].l) add(a[--l]);
            while(r<q[i].r) add(a[++r]);
            while(r>q[i].r) del(a[r--]);
            while(ch>q[i].num) {
            	if(c[ch].x>=l&&c[ch].x<=r) {
            		del(c[ch].val);
            		add(c[ch].pre);
            	}
            	a[c[ch].x]=c[ch].pre;
            	ch--;
    		}
    		while(ch<q[i].num) {
    			ch++;
    			a[c[ch].x]=c[ch].val; 
    			if(c[ch].x>=l&&c[ch].x<=r) {
            		del(c[ch].pre);
            		add(c[ch].val);
            	}
    		}
            ans[q[i].id]=cur;
    	}
    	
    	for(int i=1;i<=m;i++) {
    		if(ans[i]!=-1) {
    			printf("%lld\n",ans[i]);
    		}
    	}
    	return 0;
    }
    
    

    树上莫队

  • 相关阅读:
    利用JQuery一步步打造无缝滚动新闻
    asp.net操作oracle存储过程的小问题
    C# Tostring() 格式大全 [转]
    Oracle存储过程总结【转】
    JQuery中text()、html()和val()的区别
    JQuery的JSON数据格式
    HDU 1065 I Think I Need a Houseboat
    POJ1251 Jungle Roads
    HDU3177 Crixalis's Equipment
    HDU1042 N!
  • 原文地址:https://www.cnblogs.com/huayucaiji/p/modui.html
Copyright © 2011-2022 走看看