NOIP提高组【JZOJ4792】整除

Description

麦克雷有一个1~n的排列，他想知道对于一些区间，有多少对区间内的数（x，y），满足x能被y整除。

Data Constraint

30%：1<=n,m<=100
100%：1<=n,m<=2*10^5,1<=pi<=n

Solution

我们对于这种题考虑一下扫描线。我们先处理询问中（a[j]=k*a[i]，i>j,k>1）我们将询问先按右端点排序，然后从1到n往右做，显然对于一个新加入的点a[i]，我们枚举k*a[i]（k>1）的位置，假如在i的左侧，我们在树状数组中k *a[i]的位置加1，那么对于一次询问（x,y），它的答案显然是树状数组中区间（x,y）的和。那么对于询问中（a[j]=k*a[i]，j>=i,k>=1）的情况,只需将询问按左端点排序，从n到1倒着做即可。最后统计答案。总时间复杂度为O(N$(logN)^2$)。

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=200005,maxn1=500;
struct code{
    int a,b,c;
}b[maxn];
int a[maxn],bz[maxn],f[maxn],n,m,i,t,j,k,l,x,y,z,r,p,ans[maxn];
bool cmp(code x,code y){
    return x.b<y.b;
}
bool cmp1(code x,code y){
    return x.a<y.a;
}
int lowbit(int x){
    return x&(-x);
}
void insert(int x){
    if (x>n) return;
    f[x]++;
    insert(x+lowbit(x));
}
int find(int x){
    if (x<1) return 0;
    return f[x]+find(x-lowbit(x));
}
int main(){
//  freopen("data.in","r",stdin);
    scanf("%d%d",&n,&m);t=sqrt(n);
    for (i=1;i<=n;i++)
        scanf("%d",&a[i]),bz[a[i]]=i;
    for (i=1;i<=m;i++)
        scanf("%d%d",&b[i].a,&b[i].b),b[i].c=i;
    sort(b+1,b+m+1,cmp);
    for (i=1;i<=m;i++){
        for (j=b[i-1].b+1;j<=b[i].b;j++){
            x=n/a[j];
            for (k=2;k<=x;k++)
                if (bz[k*a[j]]<j) insert(bz[k*a[j]]);
        }
        ans[b[i].c]=find(b[i].b)-find(b[i].a-1);
    }
    memset(f,0,sizeof(f));
    sort(b+1,b+m+1,cmp1);
    b[m+1].a=n+1;
    for (i=m;i>=1;i--){
        for (j=b[i+1].a-1;j>=b[i].a;j--){
            x=n/a[j];
            for (k=2;k<=x;k++)
                if (bz[k*a[j]]>j) insert(bz[k*a[j]]);
        }
        ans[b[i].c]+=find(b[i].b)-find(b[i].a-1)+b[i].b-b[i].a+1;
    }
    for (i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}