Codeforces Round #236 (Div. 2) D. Upgrading Array

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f

int p[5050];
int g[5050];
set<int> b;
int gcd(int a,int b){
return b?gcd(b,a%b):a;
}
int get(int x)
{
int i,j;
map<int,int> m;
for(i=2;i*i<=x;i++)
{
if(x%i==0)
{
while(x%i==0)
{
m[i]++;
x/=i;
}
}
}
if(x!=1) m[x]++;
map<int,int>::iterator it;
int ans=0;
for(it=m.begin();it!=m.end();it++)
{
if(b.find((*it).first)!=b.end())
ans-=(*it).second;
else
ans+=(*it).second;
}
return ans;
}
int main()
{
int n,m,i,j,k;
scanf("%d%d",&n,&m);
for(i=0;i<n;i++)
scanf("%d",&p[i]);
for(i=0;i<m;i++)
{
scanf("%d",&k);
b.insert(k);
}
int sum=0;
for(i=0;i<n;i++)
sum+=get(p[i]);
g[0]=p[0];
for(i=1;i<n;i++)
g[i]=gcd(g[i-1],p[i]);
int h=1,t,res;
for(i=n-1;i>=0;i--)
{
t=g[i]/h;
res=get(t);
if(res<0){
sum-=res*(i+1);
h*=t;
}
}
printf("%d\n",sum);
//system("pause");
}