hdu 4710 Balls Rearrangement（）

http://acm.hdu.edu.cn/showproblem.php?pid=4710

【code】：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>

using namespace std;
#define N 1000151

int prim[N+10];
int hash[1000000];
int mark[1000010];
int cjsb[1000010];
int hash_cnt=0;
int lowbit(int i)
{
    return i&-i;
}
void add(int i,int a)
{
    while(i<=N)
    {
        cjsb[i]+=a;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=cjsb[i];
        i-=lowbit(i);
    }
    return s;
}
int main()
{
    int i,j;
    hash_cnt=0;
    for(i=2;i<=N;i++)
    {
        if(!prim[i])
        {
            hash[hash_cnt++]=i;
            for(j=2;j*i<=N;j++)
            {
                prim[j*i]=1;
            }
        }
    }
   // cout<<hash_cnt<<" "<<hash[hash_cnt-1]<<endl;
    int tou=0;
    memset(cjsb,0,sizeof(cjsb));
    for(i=1;i<hash_cnt;i++)
    {
        for(j=1;j<=i-1;j++)
        {
            int temp = hash[i]-hash[j];
            if(temp<=tou)
                break;
            if(mark[temp])  continue;
            else
            {
                mark[temp] = hash[i];
                add(temp,1);
                if(sum(temp)==temp/2)
                    tou=temp;
            }
        }
    }
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int m;
        scanf("%d",&m);
        if(m==0)
        {
            puts("2 2");
            continue;
        }
        if(m<0)
        {
            m=-m;
            printf("%d %d
",mark[m]-m,mark[m]);
        }
        else
        {
            printf("%d %d
",mark[m],mark[m]-m);
        }
    }
    return 0;
}