[JZOJ5023]Sequence

最新推荐文章于 2020-02-02 21:41:22 发布

a_crazy_czy

最新推荐文章于 2020-02-02 21:41:22 发布

阅读量519

点赞数

分类专栏：莫队算法筛法纪中OJ 文章标签： OI 莫队线筛

本文链接：https://blog.csdn.net/a_crazy_czy/article/details/63256102

版权

纪中OJ 同时被 3 个专栏收录

127 篇文章 1 订阅

订阅专栏

筛法

11 篇文章 0 订阅

订阅专栏

莫队算法

5 篇文章 0 订阅

订阅专栏

题目大意

有一个长度为 $n$ 的序列 $\{A_n\}$ 。一共有 $q$ 个询问，每个询问有两个参数 $l,r$ ，表示询问

\sum i = l r \sum j = l r ω (gcd (A i, A j))

$\sum_{i=l}^r\sum_{j=l}^r\omega\left(\gcd(A_i,A_j)\right)$
其中

ω(x) $\omega(x)$ 为

x $x$ 的不同质因子个数。

$1\le n,q\le 10^5,1\le A_i\le10^7$

题目分析

对每一个质因子分开来考虑，如果一个质因子出现在这个区间的 $x$ 个数中，那么其贡献就是 $x^2$ 。那么我们直接使用莫队维护就好了，因为每一个 $A_i$ 最多含有 $8$ 个不同的质因子，于是复杂度近似为 $\mathrm O(n\sqrt n)$ 。

代码实现

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cctype>
#include <cmath>

using namespace std;

typedef long long LL;

int read()
{
    int x=0,f=1;
    char ch=getchar();
    while (!isdigit(ch)) f=ch=='-'?-1:f,ch=getchar();
    while (isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return x*f;
}

int buf[30];

void write(LL x)
{
    if (x<0) putchar('-'),x=-x;
    for (;x;x/=10) buf[++buf[0]]=x%10;
    if (!buf[0]) buf[++buf[0]]=0;
    for (;buf[0];putchar('0'+buf[buf[0]--]));
}

LL sqr(int x){return 1ll*x*x;}

const int A=10000000;
const int N=100050;
const int Q=100050;

struct query
{
    int l,r,id;
}qy[Q+5];

int pri[A+5],f[A+5],g[A+5],cnt[A+5];
int n,q,bs,lcur,rcur;
LL ans[N];
int a[N];
LL cur;

bool operator<(query a,query b){return (a.l-1)/bs<(b.l-1)/bs||(a.l-1)/bs==(b.l-1)/bs&&a.r<b.r;}

void pre()
{
    f[1]=g[1]=1;
    for (int i=2;i<=A;++i)
    {
        if (!f[i]) pri[++pri[0]]=f[i]=g[i]=i;
        for (int j=1,x;j<=pri[0];++j)
        {
            if (1ll*pri[j]*i>A) break;
            f[x=pri[j]*i]=pri[j];
            g[x]=(f[i]==f[x]?g[i]:1)*pri[j];
            if (!(i%pri[j])) break;
        }
    }
}

void add(int id){for (int x=a[id];x!=1;x/=g[x]) cur-=sqr(cnt[f[x]]),cur+=sqr(++cnt[f[x]]);}
void erase(int id){for (int x=a[id];x!=1;x/=g[x]) cur-=sqr(cnt[f[x]]),cur+=sqr(--cnt[f[x]]);}

void Mo_Algorithm()
{
    cur=0;
    for (lcur=qy[1].l,rcur=lcur-1;rcur<qy[1].r;add(++rcur));
    ans[qy[1].id]=cur;
    for (int i=2;i<=q;++i)
    {
        for (;lcur<qy[i].l;erase(lcur++));
        for (;lcur>qy[i].l;add(--lcur));
        for (;rcur<qy[i].r;add(++rcur));
        for (;rcur>qy[i].r;erase(rcur--));
        ans[qy[i].id]=cur;
    }
}

int main()
{
    pre();
    freopen("sequence.in","r",stdin),freopen("sequence.out","w",stdout);
    n=read(),q=read();
    for (int i=1;i<=n;++i) a[i]=read();
    for (int i=1;i<=q;++i) qy[i].l=read(),qy[i].r=read(),qy[i].id=i;
    bs=trunc(sqrt(n)+1),sort(qy+1,qy+1+q);
    Mo_Algorithm();
    for (int i=1;i<=q;++i) write(ans[i]),putchar('\n');
    fclose(stdin),fclose(stdout);
    return 0;
}