PAT AHashing - Average Search Time

这个题我的代码写的好复杂。。。。

贴一下 明天再优化代码

#include<iostream>
using namespace std;
const int N=10010;
int h[N],n,m,k,cnt;
bool is_prinme(int x)
{
    if(x==1) return false;
    for(int i=2;i<=x/i;++i)
        if(x%i==0) return false;
    return true;
}
int hashn(int x,int p)
{
    int flag=true;
    for(int i=0;i<m;++i)
    {
        int t=(i*i+x)%m;
        if(h[t]==x||!h[t]&&p==1) 
        {
            cnt+=i+1,flag=false;
            break;
        }
        if(!h[t]&&p==0) 
        {
            h[t]=x,flag=false;
            return t;
        }
    }
    if(p==1&&flag) cnt+=m+1;
    if(p==0&&flag)
        return -1;
}
int main()
{
    cin>>m>>n>>k;
    while(!is_prinme(m)) m++;
    while(n--)
    {
        int x;cin>>x;
        int t=hashn(x,0);
        if(t==-1) printf("%d cannot be inserted.\n",x);
    }
    for(int i=0;i<k;++i)
    {
        int x;cin>>x;
        hashn(x,1);
    }
    //cout<<cnt;
    printf("%.1lf",(double)cnt/k);
    return 0;
}