2015年ALPC暑期代码能力练习I D Upgrading Array

题目链接：http://codeforces.com/problemset/problem/402/D

又WR了N次，卡了2天，最后问别人才直到错误在哪里。在初始化素数数组时是i*i<MAX_INT，如果输入的数是一个素数且>素数数组中最大的素数，那么这个数没有被统计。大神们一看就知道错误在哪里，可怜我卡了这么久。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<string>
#include<map>
#include<set>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
/*********************************************************************/
const int MAX_NUM = 5000+500;
const int MAX_INT = 1000000000+10;
int a[MAX_NUM];
int allGcd[MAX_NUM];
vector<int> badPrime;
vector<int> allPrime;

void swap (int& a,int& b){a^=b;b^=a;a^=b;}
int min_2 (int a,int b){return a<b?a:b;}
int max_2 (int a,int b){return a>b?a:b;}
int gcd (int a,int b){return b==0?a:gcd(b,a%b);}

void InitGoodPrime ()
{
  allPrime.push_back (2);
  int sign=1;
  for (int i=3;i*i<MAX_INT;++i){
	sign=1;
	for (int j=0;j<allPrime.size ();++j)
	  if (i%allPrime[j]==0){sign=0;break;} 
	if (sign)
	  allPrime.push_back (i);
  }
}
int JudgeGcd (int x)
{
  int sum=0;
  for (int i = 0; i<badPrime.size (); i++){//统计坏素数的个数
	while (x%badPrime[i]==0) {
	  x/=badPrime[i];
	  --sum;
	}
  }
  for (int i = 0; i<allPrime.size () && allPrime[i]<=x; i++){//统计好素数的个数，坏素数已经被除尽
	while (x%allPrime[i]==0) {
	  x/=allPrime[i];
	  ++sum;
	}
  }
  if (x!=1)//素数只统计到sqrt(MAX)，如果x!=1就是大于allPrime中的最大素数，而且x一定是素数
	++sum;
  return sum;
}
/*********************************************************************/
int main()
{
  int n,m;
  allPrime.clear ();
  InitGoodPrime ();
  while (scanf ("%d %d",&n,&m)!=EOF)
  {
	for (int i=0;i<n;i++)
	  scanf ("%d",&a[i]); 
	badPrime.clear ();
	int bad;
	for (int i=0;i<m;i++){
	  scanf ("%d",&bad);
	  badPrime.push_back(bad);
	}
	allGcd[0]=a[0];
	for (int i=1;i<n;i++)
	  allGcd[i]=gcd (allGcd[i-1],a[i]);
	int div[MAX_NUM];
	div[n]=1;
	for (int i=n-1;i>=0;i--){
	  if (JudgeGcd (allGcd[i]/div[i+1])<0)//考虑后面已经除了。如果再除的数，坏素数多，就除以gcd
		div[i]=allGcd[i];
	  else
		div[i]=div[i+1];//否则就除以后面那个数的除数
	  a[i]/=div[i];
	}
	int source=0; 
	for (int i=0;i<n;i++)
	  source+=JudgeGcd (a[i]);
	printf ("%d\n",source);
  }
  return 0;
}