2018.10.27 codeforces402D. Upgrading Array(數論+貪心)
阿新 • • 發佈:2018-10-31
傳送門
唉我覺得這題資料範圍1e5都能做啊。。。
居然只出了2000
考完聽
說我的貪心可以卡但過了?
可能今天本來是
只是運氣好
騙了
吧233.
下面講講貪心:
令
令
那麼
令
考慮
有
令
那麼有
所以如果先在
那個地方進行一次操作,無論怎麼樣都不會令結果變小。
於是我們從後向前算出所有有正的貢獻的
加到答案裡面就行了。
程式碼:
#include<bits/stdc++.h>
#include<tr1/unordered_map>
using namespace std;
const int N=32005;
int n,m,pri[N],a[5005],b[5005],tot=0,sum=0,ans=0;
bool vis[N];
tr1::unordered_map<int,int>S;
inline int read(){
int ans=0;
char ch=getchar();
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
return ans;
}
inline void init(){
for(int i=2;i<=N-5;++i){
if(!vis[i])pri[++tot]=i;
for(int j=1;j<=tot&&i*pri[j]<=N-5;++j){
int k=i*pri[j];
vis[k]=1;
if(i%pri[j]==0)break;
}
}
}
inline int calc(int x){
int ret=0;
for(int i=1;i<=tot&&pri[i]*pri[i]<=x;++i){
if(x==1)break;
if(x!=x/pri[i]*pri[i])continue;
int cnt=0;
while(x==x/pri[i]*pri[i])++cnt,x/=pri[i];
if(S[pri[i]])ret-=cnt;
else ret+=cnt;
}
if(x^1){
if(S[x])--ret;
else ++ret;
}
return ret;
}
int main(){
init();
n=read(),m=read(),S.clear();
for(int i=1;i<=n;++i)a[i]=read();
for(int i=1;i<=m;++i)S[read()]=1;
for(int i=1;i<=n;++i){
sum+=calc(a[i]);
if(i^1)a[i]=__gcd(a[i],a[i-1]);
b[i]=-calc(a[i]);
}
int pos=n,delta=0;
while(pos){
if(b[pos]-delta>0)sum+=(b[pos]-delta)*pos,delta+=b[pos]-delta;
--pos;
}
cout<<sum;
return 0;
}