1145 Hashing - Average Search Time (25 分)

题目链接
平方探测k等于哈希表长度时，视为失败。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
int H[maxn]={0};
int arr[maxn]={0};
bool isPrime(int x){
	if(x<=1) return false;
	int sqr = sqrt(x);
	for(int i=2;i<=sqr;i++){
		if(x%i==0) return false;
	}
	return true;
}
int main(){
	int m,n,x,s;
	cin>>s>>n>>m;
	while(isPrime(s)==false) s++;
	while(n--){
		cin>>x;
		bool flag = false;
		for(int k=0;k<=s;k++){
			int now = (x+k*k)%s;
			if(H[now] == 0){
				H[now]=1;
				arr[now]=x;
				flag=true;
				break;
			}
		}
		if(flag==false) cout<<x<<" cannot be inserted."<<endl;
	}
	int cnt=0;
	for(int j=0;j<m;j++){ //m个查询
		cin>>x;
		int k;
		bool flag = false;
		for(k=0;k<s;k++){
			int now = (x+k*k)%s;
			if(H[now]==1&&arr[now]==x) break;
			else if(H[now]==0) break;
		}
		cnt += k+1;
	}
	printf("%.1f\n",cnt*1.0/m);
	return 0;
}