分解质因数 - Prime Game - ICPC 2018 南京

分解质因数 - Prime Game - ICPC 2018 南京

题意：

$给定一个长度为n的整数序列a_i。$

$令$

$$mul(l,r)=\prod _{i=l}^r{a}_i，fac(l,r)表示mul(l,r)本质不同的质因子的个数$$

$请计算：$

$$\sum _{i=1}^n\sum _{j=i}^{n}fac(i,j)$$

输入：

$首行输入一个正整数n(1\le n\le 10^6)，表示序列的长度。$

$第二行包含n个整数a_i(1\le i\le n,1\le a_i\le 10^6)，为这个整数序列。$

输出：

$输出一个整数表示答案。$

Examples
standard input

10
99 62 10 47 53 9 83 33 15 24

standard output

248

standard input

10
6 7 5 5 4 9 9 1 8 12

standard output

134

---

分析：

$正难则反，考虑每个质因数对每个区间的贡献。$

$因为一个区间中可能有多个相同的质因子，$

$本题难点在于，如何用一个合理的计算方法，剔除掉同一个质因子对区间的重复贡献。$

思路如下：

$①、首先，我们对每个数分解质因数，对于每一个质因数p，记录下包含质因子p的所有数a_i的位置i。$

$②、接着，对于每一个质因子p，我们遍历位置数组pos，pos[p][j]表示质因子p第j次出现的位置。$

$③、对于每一个位置pos[p][j]，我们计算这个位置的数对答案的贡献，$

$Ⅰ、从这个位置向左(包含这个位置)，剔除掉上一个位置pos[p][j-1],$

$我们共有pos[p][j]-pos[p][j-1]个不同的选择，即不同的区间左端点$

$Ⅱ、从这个位置向右，一直到末尾n，$

$我们共有n-pos[p][j]+1种不同的选择，即不同的区间右端点。$

$仔细思考这个策略，计算每个位置的数的贡献时，$

$我们总是剔除了前一个位置的具有相同质因子的数的影响。$

$我们不需要考虑后面的位置的数的影响，因为我们统计的是包含当前位置的数的区间，这部分区间的贡献。$

$而因为前面的数也未考虑后面的数对其贡献的影响，所以我们要剔除它。$

$故最终答案为：$

$$Ans=\sum _{i=0}^{p_i\le ma{x}_p}\sum _{j=0}^{pos[p].size()}[pos[p][j]-(j?pos[p][j-1]:0)]\times (n-pos[p][j]+1)$$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>

using namespace std;

const double eps = 1e-5;
const int mod = 1e9+7;
const int N = 1e6+10;

#define P pair<double,double>
#define x first
#define y second
#define ll long long

int n;
int primes[N], cnt;
bool st[N];
int num[N];
vector<int> pos[N];
int maxp;

void get_prime(int n)
{ 
    for(int i=2;i<=n;i++)
    {
        if(!st[i]) primes[cnt++]=i;
        for(int j=0;primes[j]*i<=n;j++)
        {
            st[i*primes[j]]=true;
            if(i%primes[j]==0) break;
        }
    }
}

void solve(int id)
{
    int x = num[id];
    int sqx = sqrt(x);
    for(int i=0;primes[i]<=sqx;i++)
    {
        int p = primes[i];
        if(x%p==0)
        {
            maxp = max(maxp,p);
            pos[p].push_back(id);
            while(x%p==0) x/=p;
        }
    }
    if(x>1) 
    {
        maxp=max(maxp,x);
        pos[x].push_back(id);
    }
}

ll cal()
{
    ll res = 0;
    for(int i=0;primes[i]<=maxp;i++)
    {
        int p = primes[i];
        for(int j=0;j<pos[p].size();j++)
            res += (ll)(pos[p][j] - (j ? pos[p][j-1] : 0)) * (n - pos[p][j] + 1);
    }
    return res;
}

int main()
{
    get_prime(N-1);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&num[i]);
    for(int i=1;i<=n;i++) solve(i);
    printf("%lld\n",cal());
    return 0;
}