湖南师范大学2018年大学生程序设计竞赛新生赛-J： 名哥的完全平方数（莫队）

链接：https://www.nowcoder.com/acm/contest/127/J
来源：牛客网

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

511 CF第一人名哥不上紫名不实习！
这天，名哥上CF刷了一道有趣的题（CF 480D），意犹未尽！
跟数学大佬浩佬吹嘘，浩佬看了题目：”这太简单了！我改一下，看你能做出来吗? 给一个长度为n的数组，做q次询问，每次询问区间[l,r]里有多少对数的乘积为完全平方数？”
名哥呆了，您能帮名哥解决吗？

输入描述:

第一行输入 n(1≤n≤3*10^5) ，表示数组的长度。
第二行输入a1……an(-1000000≤ai≤1000000)
第三行输入q(1≤n≤3*10^5) ， 表示询问的次数
下面q行，每行两个整数表示查询的区间：l,r(1≤l≤r≤n)

输出描述:

对每次查询输出一行：输出1个整数 ans ，表示查询区间的两个数的乘积为完全平方数的对数。

示例1

输入

复制

5
1 -4 -36 2 0
4
1 2
2 3
3 5
4 4

输出

复制

0
1
2
0

思路：对于一对数x，y。当x*y是个完全平方数时，那么x的素因子和y的素因子是相同的，且个数和为偶数。那么对a数组进行预处理，对于a[i]，把所有素因子的平方值从中除去，即使得所有a[i]全部由若干个质数组成。那么此时要使得a[i]*a[j]是个完全平方数，只有a[i]=a[j]时才成立。

那么接下来的处理方法就是莫队离线处理了。

注意：对于a[i]=0的情况，可以单独拿出来处理！

#include<bits/stdc++.h>
using namespace std;
const int MAX=1e6+10;
typedef long long ll;
struct lenka
{
    int l,r;
    int index;
    int th;
}b[MAX];
int cmp(const lenka& p,const lenka& q)
{
    if(p.th==q.th)return p.r<q.r;
    return p.th<q.th;
}
int a[MAX];
ll ans[MAX],num[MAX];//num[i]表示a[1...i]中0的个数
ll z[MAX],f[MAX];    //z[i]记录i个数,f[i]记录-i的个数
ll solve(ll sum,int x,int y,int l,int r)
{
    for(int i=x;i<l;i++)
    {
        if(a[i]>0)z[a[i]]--,sum-=z[a[i]];
        if(a[i]<0)f[-a[i]]--,sum-=f[-a[i]];
    }
    for(int i=l;i<x;i++)
    {
        if(a[i]>0)sum+=z[a[i]],z[a[i]]++;
        if(a[i]<0)sum+=f[-a[i]],f[-a[i]]++;
    }
    for(int i=y+1;i<=r;i++)
    {
        if(a[i]>0)sum+=z[a[i]],z[a[i]]++;
        if(a[i]<0)sum+=f[-a[i]],f[-a[i]]++;
    }
    for(int i=r+1;i<=y;i++)
    {
        if(a[i]>0)z[a[i]]--,sum-=z[a[i]];
        if(a[i]<0)f[-a[i]]--,sum-=f[-a[i]];
    }
    return sum;
}
int main()
{
    int n;
    cin>>n;
    num[0]=0;
    for(int i=1;i<=n;i++)
    {
        num[i]=num[i-1];
        scanf("%d",&a[i]);
        for(int j=2;j*j<=abs(a[i]);j++)
        {
            while(a[i]%(j*j)==0)a[i]/=j*j;
        }
        num[i]+=a[i]==0;
    }
    int m;
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&b[i].l,&b[i].r);
        b[i].index=i;
        b[i].th=b[i].l/(int)sqrt(n);
    }
    sort(b+1,b+m+1,cmp);
    memset(ans,0,sizeof ans);
    memset(z,0,sizeof z);
    memset(f,0,sizeof f);
    for(int i=1;i<=m;i++)
    {
        if(i==1)
        {
            for(int j=b[i].l;j<=b[i].r;j++)
            {
                if(a[j]<0)ans[b[i].index]+=f[-a[j]],f[-a[j]]++;
                if(a[j]>0)ans[b[i].index]+=z[a[j]],z[a[j]]++;
            }
            continue;
        }
        ans[b[i].index]=solve(ans[b[i-1].index],b[i-1].l,b[i-1].r,b[i].l,b[i].r);
    }
    for(int i=1;i<=m;i++)//对于0，单独拿出来处理
    {
        ll tot=num[b[i].r]-num[b[i].l-1];
        ans[b[i].index]+=tot*(b[i].r-b[i].l)-tot*(tot-1)/2;
    }
    for(int i=1;i<=m;i++)printf("%lld\n",ans[i]);
    return 0;
}