莫队算法

莫队算法

用ql和qr表示正在询问的区间，当询问区间[l,r]时，移动ql和qr并随着移动增加或减少颜色的数量，比如现在区间[ql,qr]（红色为1，蓝色为2，绿色为3）,cnt_1=3，cnt_2=2,cnt_3=1，当ql向左移动一位，cnt_2变化为1，当如果区间间隔太大移动次数多时间复杂度也随着增加，所以可以给区间排序，按左端点升序，相同按右断点升序，但是如果出现数据（[1,1000],[2,3],[3,1000],[4,2]）指针跳来跳去时间复杂度也不小。

重点来了！！！！(分块)

对询问进行分块，我们知道m个询问，L和r的范围都在n以内，我们根据L和r的大小来对询问分块。
比如n=9，有以下的询问：
2 3
1 4
4 5
1 6
7 9
8 9
5 8
6 8
对于n=9，我们以根号n为每个块block的大小，这里block=3.
那么我们把1~ 3分成一组，4~ 6，7~9.
对于每一个询问（L，r），我们以L的范围来决定这个询问在哪一个块。
然后每一个独自的块内，我们让询问r更小的排在更前面。
那么上面的询问就可以分组成：
(2,3)/(1,4)/(1,6)和
(4,5)/(5,8)/(6,8)和
(7,9)/(8,9)

超重点来了！！！！(奇偶性排序)
正常排序时，由于每个块的右端点都是按照从小到大排序的，当指针跳回左边后处理下一个块又要跳回右边，增加了不必要的移动，因此，此时可以按照奇偶性排序进行优化：当左端点的块为奇数时，右端点按照从小到大排；当左端点的块偶数时，右端点按照从大到小排。这样可以保证指针移到右边不用再跳回左边，减少一半的操作，理论上可以快一倍。

更详细的解说→莫队解说

bool cmp2(node a , node b){    //正常排序（分块）
    if (a.l/block == b.l/block)    return a.r < b.r ;
    else    return a.l/block < b.l/block ;  
}
bool cmp(node a , node b){    //奇偶性排序
/*
    if (a.l/block == b.l/block){
        if ((a.l/block)%2 == 1)    //左端点块为奇数时
            return a.r < b.r ;
        else    //左端点块为偶数时
            return a.r > b.r ;
    }
    else{    //当左端点块不位于同一块时
        return a.l < b.l ; //按块的位置排序
    }
    */
    // 本函数可用以下语句完成
    return (a.l/block)^(b.l/block) ? a.l <b.l :(((a.l/block) & 1) ? a.r<b.r : a.r>b.r) ;
}

下题为牛客小白赛17 G题
链接：https://ac.nowcoder.com/acm/contest/1085/G

#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std ;
const int N = 1e5 + 5 ;
struct node{
    long long l , r , id ;
}q[N] ;
long long num[N] , cnt[N] , res[N] , block , ans ;
bool cmp (node a , node b){
   return (a.l/block)^(b.l/block) ? a.l <b.l :(((a.l/block) & 1) ? a.r<b.r : a.r>b.r) ;
}
void add(long long x){
    ans+=(2*cnt[num[x]]+1) *num[x];
    cnt[num[x]]++;
}
void del(long long x){
    ans-=(2*cnt[num[x]]-1) *num[x];
    cnt[num[x]]--;
}
inline long long read(){
    char ch = getchar() ;
    long long x = 0 , f = 1 ;
    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 ;
}
inline void print(long long x){
    char f[200] ;
    if (x < 0) {
        x = -x ;
        putchar('-') ;
    }
    int len = 0 ;
    while(x){
        f[len++] = x%10 + '0' ;
        x /= 10 ;
    }
    while(len > 0)    putchar(f[--len]) ;
}
int main(){
    long long n , m ;
    n = read() ;
    m = read() ;
    for (int i = 1 ; i <= n ; i ++)
        num[i] = read() ;
    for(int i = 1 ; i <= m ; i ++){
        q[i].l = read() ;
        q[i].r = read() ;
        q[i].id = i ;
    }
    block = n/sqrt(2*m/3) ;
    sort(q+1,q+1+m,cmp) ;
    long long l = 0 , r = 0 ;
    for (long long i = 1 ; i <= m ; i ++){
        long long ql = q[i].l , qr = q[i].r ;
        while(l < ql)    del(l ++) ;
        while(l > ql)    add(-- l) ;
        while(r < qr)    add(++ r) ;
        while(r > qr)    del(r --) ;
        res[q[i].id] = ans ;
    }
    print (res[1]);
    for (long long i = 2 ; i <= m ; i ++){
        putchar('\n') ;
        print(res[i]) ;
    }
    putchar('\n') ;
    return 0 ;
}

ps：一般莫队题的输入输出较多，用scanf和printf也可能会慢，所以有时候会需要快读或快写函数。