CF402D Upgrading Array(贪心+数学+线性素数筛)

来源:互联网 发布:云计算和云存储的区别 编辑:程序博客网 时间:2024/05/16 10:14

先预处理出g[i]表示a[1]…a[i]的gcd,我们贪心的发现,肯定是从后往前做,这样不会影响前面。每次我们分解这个gcd,看他是负贡献还是正贡献,决定是不是要除掉它。这样的操作之后,我们分解每一个剩下的数,统计答案即可。用线性素数筛提前把素数筛出来,复杂度应该是小于O(n sqrt(1e9))的。

#include <bits/stdc++.h>using namespace std;#define ll long long#define inf 0x3f3f3f3f#define N 5010inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}int n,m,prime[100000],tot=0,a[N],g[N],ans=0;bool notprime[100010];map<int,bool>bad;void Prime(){    notprime[1]=1;    for(int i=2;i<=100000;++i){        if(!notprime[i]) prime[++tot]=i;        for(int j=1;prime[j]*i<=100000;++j){            notprime[prime[j]*i]=1;            if(i%prime[j]==0) break;        }    }}inline int gcd(int x,int y){    return y==0?x:gcd(y,x%y);}int main(){//  freopen("a.in","r",stdin);    n=read();m=read();    for(int i=1;i<=n;++i) a[i]=read();g[1]=a[1];    for(int i=1;i<=m;++i) bad[read()]=1;    for(int i=2;i<=n;++i) g[i]=gcd(g[i-1],a[i]);    int tmp=1,res=0;    for(int i=n;i>=1;--i){        g[i]/=tmp;a[i]/=tmp;if(g[i]==1) continue;        int xx=g[i],num[2]={0,0};        for(int x=2;x*x<=g[i];++x){            if(notprime[x]||xx%x!=0) continue;            while(xx%x==0){                xx/=x;num[bad[x]]++;            }        }if(xx>1) num[bad[xx]]++;        if(num[1]>=num[0]) tmp*=g[i],a[i]/=g[i];    }for(int i=1;i<=n;++i){int xx=a[i];        for(int x=2;x*x<=a[i];++x){            if(notprime[x]||xx%x!=0) continue;            while(xx%x==0){                xx/=x;bad[x]?ans--:ans++;            }        }if(xx>1) bad[xx]?ans--:ans++;    }printf("%d\n",ans);    return 0;}