1. 程式人生 > >RMQ預處理gcd

RMQ預處理gcd

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int MAXN=100009;
int st[MAXN][20];
map<int,LL>ans;
void RMQ(int n){                    //RMQ一般就是指ST表
    int Len=log2(n);                //得到了要維護的深度
    for(int i=1;i<=Len;i++){          //i深度,在輸入的時候st[j][0]已經存好,這裡i直接從1開始
        for(int j=1;j<=n;j++){        //j位置,注意輸入時存的下標是從1~N
            if(j-1+(1<<i)>n)break;    //退出條件:從j位起長為(1<<i)區間最後一位置j+(1<<i)-1,故>n跳出
            st[j][i]=__gcd(st[j][i-1],st[j+(1<<(i-1))][i-1]);//st[j][i]指第j位起長為2^i長度區間GCD
        }
    }
}
int gcd(int l,int r){            //傳入查詢區間
    int Len=log2(r-l+1);        //在第幾深度查詢
    return __gcd(st[l][Len],st[r-(1<<Len)+1][Len]);//左右區間GCD的GCD即為所求(區間可部分覆蓋)
}
//gcd從左端點開始要麼是一長段不變 , 要麼是直接減少一半以上
void init(int n){                 //就是用來更新ANS
    for(int i=1;i<=n;i++){        //列舉左界,二分求出每次變化的點
        int now=i;                //在以i為左界下,now是當前判到的變化點,即由i開始往右掃區間GCD有變化的點,左界i初始化為第一個
        while(now<=n){            //變化點未去到盡頭就要繼續,因為要把以I為左界的右邊的區間全部掃完
            int l=now,r=n,x=now;    //解釋x:在以i為左界下,以當前變化點now為右界的GCD記為G1
                                    //則x是在以i為左界下,能使GCD也是G1的最大右界
            while(l<=r){                    //二分未結束
                int mid=l+r>>1;             //二分割槽間中點
                if(gcd(i,mid)==gcd(i,now))  //如果i,mid區間與i,now區間GCD下相同
                    x=mid,l=mid+1;          //表明下一變化點在右半區間,則最大右界移到中點,然後左界右移
                else r=mid-1;               //否則下一變化點在左半區間,則右界左移,最大右界不動
            }
            ans[gcd(i,now)]+=x-now+1;       //一整段gcd都相同,最大右界減當前變化點的長度加進對應答案裡計數
            now=x+1;        //下一次變化點就是當前變化點下GCD不變的最大右界
        }
    }
}
int main(){
    int t,ca=0;cin>>t;
    while(t--){
        printf("Case #%d:\n",++ca);ans.clear();
        int n;cin>>n;for(int i=1;i<=n;i++)cin>>st[i][0];//輸入N個數,注意是從1開始存的
        RMQ(n);              //處理ST表
        init(n);             //處理GCD對映MAP(GCD值,GCD數量)
        int q;cin>>q;        //詢問數
        while(q--){
            int l,r;cin>>l>>r;int g=gcd(l,r);    //找出區間GCD
            printf("%d %lld\n",g,ans[g]);        //輸出區間找到的GCD及一樣GCD的區間數
        }
    }
} 
RMQ預處理gcd(hdu 5726 GCD)
題意:n的陣列 , q次詢問 , 每次詢問(l,r) , 求區間內所有數的gcd , 以及有多少個區間的gcd和這個區間相同(包括自己)
Sample Input
1
5
1 2 4 6 7
4
1 5
2 4
3 4
4 4
Sample Output
Case #1:
1 8
2 4
2 4
6 1