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

本文链接：https://blog.csdn.net/Lakersilk/article/details/80612999

题目描述

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 ，表示查询区间的两个数的乘积为完全平方数的对数。

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 7;
int ans[maxn];
int a[maxn];
int belong[maxn];
int block;
int n,q;
int Ans,zero;
int p[maxn];
struct node{
    int l,r,id;
    bool operator < (const node &a)const{
        if(belong[l] == belong[a.l]) return r < a.r;
        return l < a.l;
    }
}Q[maxn];
int cal(int x){
    return x * (x - 1) / 2;
}
void init(){
    for(int i=1;i<maxn;i++) p[i] = i;
    for(int i=2;i<maxn;i++){
        if(p[i] == i){
            for(int j=2*i;j<maxn;j+=i){
                while(p[j] % (i*i) == 0) p[j] /= i*i;
            }
        }
    }
}
int cnt1[maxn],cnt2[maxn];
void work(int x,int f){
    if(a[x] == 0){
        zero += f;
    }else if(a[x] > 0){
        Ans -= cal(cnt1[p[a[x]]]);
        cnt1[p[a[x]]] += f;
        Ans += cal(cnt1[p[a[x]]]);
    }else{
        Ans -= cal(cnt2[p[-a[x]]]);
        cnt2[p[-a[x]]] += f;
        Ans += cal(cnt2[p[-a[x]]]);
    }
}
int main(){
    cin >> n;
    block = sqrt(n);
    for(int i=1;i<=n;i++) cin >> a[i];
    cin >> q;
    for(int i=1;i<=q;i++){
        cin >> Q[i].l >> Q[i].r;
        belong[Q[i].l] = Q[i].l / block;
        Q[i].id = i;
    }
    init();
    sort(Q+1,Q+1+q);
    int l = 1,r = 0;
    for(int i=1;i<=q;i++){
        while(l < Q[i].l) work(l,-1),++l;
        while(l > Q[i].l) --l,work(l,1);
        while(r < Q[i].r) ++r,work(r,1);
        while(r > Q[i].r) work(r,-1),--r;
        //while(r < Q[i].r) ++r,work(r,1);
        ans[Q[i].id] = Ans + (r-l) * zero - cal(zero);
        //cout << ans[Q[i].id] << endl;
        //cout << Ans << " " << (r-l) * zero << " " << cal(zero) << endl;
    }
    for(int i=1;i<=q;i++) cout << ans[i] << endl;
    return 0;
}