【模板】 莫隊算法

莫隊算法??

這個算法是由之前的國家隊隊長莫濤巨神(Orz….%%%)發明的,所以尊稱莫隊算法。

莫隊是啥??

如果我們知道區間[L,R],就能在O(1)求出[L−1,R],[L+1,R],[L,R−1],[L,R+1]的話,那就可以用莫隊算法了。

莫隊咋搞??

1):排序,以左段點所在的塊爲第一關鍵字,以右端點爲第二關鍵字

2):從左往右處理詢問(離線)

3):不斷調整l,r的位置並同時修改

莫隊代碼??

題目:HH的項鍊 基本是模板題……

#include <bits/stdc++.h>
using namespace std;
struct question{
    int l;
    int r;
    int id;
}qst[1110000];           //由於是離線做法,所以要開結構體存一下
int comp(const question & a,const question & b)
{
    if(a.l==b.l)    return a.r<b.r;
    return a.l<b.l;
}                        //結構體排序
int sum,num[1110000],l=0,r=0,ans[1110000];
void pls(int A)
{
    num[A]++;
    if(num[A]==1)
    {
        sum++;
    }
}                     
void mns(int B)
{
    num[B]--;
    if(num[B]==0)
    {
        sum--;
    }
}
int str[1111000];
int main()
{
    int m,n,k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&str[i]);

    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&qst[i].l,&qst[i].r);
        qst[i].id=i;
    }
    sort(qst+1,qst+m+1,comp);           //以l爲第一關鍵字,以r爲第二關鍵字 排序

    for(int i=1;i<=m;i++)
    {
        while(r<qst[i].r)
            r++,pls(str[r]);

        while(r>qst[i].r)
            mns(str[r]),r--;

        while(l<qst[i].l)
            mns(str[l]),l++;

        while(l>qst[i].l)
            l--,pls(str[l]);
        //因爲r和l的初始值的原因,while循環內部順序不可改變
        ans[qst[i].id]=sum;
    }
    for(int i=1;i<=m;i++)
        cout<<ans[i]<<endl;;
    return 0;
}

莫隊!!

莫隊算法在解決離線區間詢問幾乎是無敵的。
恩,幾乎只要能離線,用分塊的莫隊算法都能取得一個令人滿意的的解法。
所以就有很多擴展(解決線段樹等數據結構由於需要區間加和性而不能解決的問題),如區間衆數,平均數什麼的。
恩。棒!

發佈了51 篇原創文章 · 獲贊 7 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章