HDU 4777 Rabbit Kingdom --容斥原理+树状数组

题意： 给一个数的序列，询问一些区间，问区间内与区间其他所有的数都互质的数有多少个。

解法： 直接搞有点难， 所谓正难则反，我们求区间内与其他随便某个数不互质的数有多少个，然后区间长度减去它就是答案了。

那么怎么求区间内与区间其他某个数不互质的数的个数（记为cnt）呢？ 我们用L[i],R[i]表示在整个序列中左边与 i 最近的与 i 不互质的数的位置，R[i]表示右边的，L[i],R[i]我们可以正反扫一遍顺便分解因子，用个pos[]记录很方便地求出。那么区间内的cnt为L[i]或R[i]在区间内的 i 的个数。

令事件 A：i 的 L[i]在区间内    B：i 的 R[i]在区间内， 则答案为

即 cnt = |A|+|B|-|A∩B|

那么

事件A 记为 [L[i],i] 在区间内

事件B 记为 [i,R[i] 在区间内

事件|A∩B| 记为 [L[i],R[i]]在区间内

用三个vector分别存下三种区间。

解决区间内有多少个区间可以用离线树状数组做。

注意： sort 结构体vector时务必在结构体中内嵌比较函数，手写cmp函数再 sort(v.begin(),v.end(),cmp) 会超时。我也不知道为啥。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;
#define N 200007

struct node{
    int l,r,ind;
    node(int _l,int _r,int _ind):l(_l),r(_r),ind(_ind){}
    node(){}
    bool operator<(const node &B)const{ return r<B.r; }
}Q[N];
vector<node> AB[3];
int ans[3][N],T[N];
int a[N],pos[N],maxi,L[N],R[N],n,m,c[N];

int cmp(node ka,node kb) { return ka.r < kb.r; }
int lowbit(int x) { return x&-x; }

void modify(int x)
{
    if(x <= 0) return;
    while(x <= n) { c[x]++; x += lowbit(x); }
}

int getsum(int x)
{
    int res = 0;
    while(x > 0) { res += c[x]; x -= lowbit(x); }
    return res;
}

void init()
{
    int i,j;
    for(i=0;i<=maxi;i++) pos[i] = 0;
    for(i=1;i<=n;i++)
    {
        int tmp = a[i];
        for(j=2;j*j<=tmp;j++)
        {
            if(tmp%j == 0)
            {
                L[i] = max(L[i],pos[j]);
                pos[j] = i;
                while(tmp%j == 0) tmp/=j;
            }
        }
        if(tmp != 1)
        {
            L[i]=max(L[i],pos[tmp]);
            pos[tmp]=i;
        }
    }
    for(i=0;i<=maxi;i++) pos[i] = n+1;
    for(i=n;i>=1;i--)
    {
        int tmp = a[i];
        for(j=2;j*j<=tmp;j++)
        {
            if(tmp%j == 0)
            {
                R[i] = min(R[i],pos[j]);
                pos[j] = i;
                while(tmp%j == 0) tmp/=j;
            }
        }
        if(tmp != 1)
        {
            R[i]=min(R[i],pos[tmp]);
            pos[tmp]=i;
        }
    }
}

void GET(int k)
{
    memset(c,0,sizeof(c));
    int i,j = 0;
    for(i=1;i<=m;i++)
    {
        int L = Q[i].l;
        int R = Q[i].r;
        int ind = Q[i].ind;
        while(j < n && AB[k][j].r <= R)
            modify(AB[k][j].l), j++;
        ans[k][ind] = getsum(R)-getsum(L-1);
    }
}

int main()
{
    int i,j;
    while(scanf("%d%d",&n,&m)!=EOF && n+m)
    {
        maxi = 0;
        memset(ans,0,sizeof(ans));
        for(i=0;i<3;i++) AB[i].clear();
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]), maxi = max(maxi,a[i]);
            L[i] = 0, R[i] = n+1;
        }
        init();
        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&Q[i].l,&Q[i].r);
            Q[i].ind = i;
            T[i] = Q[i].r-Q[i].l+1;
        }
        sort(Q+1,Q+m+1);
        for(i=1;i<=n;i++)
        {
            AB[0].push_back(node(L[i],i,0));       //A : L[i]在区间内的数的个数
            AB[1].push_back(node(i,R[i],0));       //B : R[i]在区间内的数的个数
            AB[2].push_back(node(L[i],R[i],0));    //A交B
        }
        for(i=0;i<3;i++)
        {
            sort(AB[i].begin(),AB[i].end());
            GET(i);
        }
        for(i=1;i<=m;i++)
            printf("%d\n",T[i]-ans[0][i]-ans[1][i]+ans[2][i]);  //容斥原理
    }
    return 0;
}