【codeforces 402D】Upgrading Array

题目：Upgrading Array

题意：

       给出一个数列，可以进行一种操作将某一个前缀除去他们的gcd,有一个函数：

        

      

       其中p是坏素数，问经过若干操作后：

       

       以上的最大值。

解析：

       贪心+质数筛选+分解质因子。

       首先明确结果就是每个数的 f(x) 之和，f(x) 就是它的质因数当中的好素数的个数减去坏素数的个数。

       然后我们知道如果某个gcd中好素数的个数小于坏素数的个数，那么我们把它除掉一定可以得到更好的结果，因为它对答案的贡献是负的。那么如果从最后一个数开始的话，后面的数除gcd不会影响前面的数。而如果从前面开始枚举，那么进行一次修改后后面的前缀gcd都为1，无法对答案产生贡献。所以我们贪心的倒序处理即可。

       为什么贪心是对的？因为后面每次除之后对前面的答案没有影响，即无后效性，且除之后一定会使得答案更优。

 

代码：

#include <bits/stdc++.h>
#include <tr1/unordered_map>
using namespace std;

const int Max=5010;
int n,m,ans,tot;
int pri[16000],v[32000],num[Max],sum[Max];
tr1::unordered_map<int,bool>vis;

inline int get_int()
{
	int x=0,f=1;char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') f=-1,c=getchar();
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline void pre()
{
	for(int i=2;i<=31625;i++)
	{
	  if(v[i]) continue;
	  pri[++tot]=i;
	  for(int j;j<=31625/i;j++) v[i*j]=1;
	}
}

inline int calc(int x)
{
	int sum=0;
	for(int i=1;i<=tot&&pri[i]*pri[i]<=x;i++)
	{
	  if(x%pri[i]) continue;
	  int add=vis.find(pri[i])==vis.end()?1:-1;
	  while(x&&!(x%pri[i])) x/=pri[i],sum+=add;
	}
	if(x!=1) sum+=vis.find(x)==vis.end()?1:-1;
	return sum;
}

int main()
{
	pre();
	n=get_int(),m=get_int();
	for(int i=1;i<=n;i++) sum[i]=num[i]=get_int();
	for(int i=1;i<=n;i++) sum[i]=__gcd(sum[i-1],sum[i]);
	for(int i=1;i<=m;i++) vis[get_int()]=1;
	for(int i=1;i<=n;i++) ans+=calc(num[i]);
	int add=1;
	for(int i=n;i;i--)
	{
	  sum[i]/=add;
	  int x=calc(sum[i]);
	  if(x<0)
	  {
	  	add*=sum[i];
	  	ans-=i*x;
	  }
	}
	cout<<ans;
	return 0;
}