1145 Hashing - Average Search Time (25 分)

/*
    题目大意：给定一个序列，用平方探测法解决哈希冲突，然后给出m个数字，如果这个数字不能够被插入就输出”X cannot be inserted.”，
    然后输出这m个数字的平均查找时间

分析：先找到大于tsize的最小的素数为真正的tsize，然后建立一个tsize长度的数组。
首先用平方探测法插入数字a，每次pos = (a + j * j) % tsize，j是从0～tsize-1的数字，
如果当前位置可以插入就将a赋值给v[pos]，如果一次都没有能够插入成功就输出”X cannot be inserted.”。
其次计算平均查找时间，每次计算pos = (a + j * j) % tsize，其中j <= tsize，
如果v[pos]处正是a则查找到了，则退出循环，如果v[pos]处不存在数字表示没查找到，
那么也要退出循环。每次查找的时候，退出循环之前的j就是这个数字的查找长度。
最后ans除以m得到平均查找时间然后输出～

此题似乎并不够严谨。。。 大概懂了就可以
*/

#include <iostream>
using namespace std;
#include <vector>
#include <iomanip>

bool isprime(int n){
    if(n <= 1) return false;
    for(int i=2; i*i<=n; i++){
        if(n % i == 0){
            return false;
        }
    }
    return true;
}
int main(){
    int tsize, n, m, a;
    cin >> tsize >> n >> m;
    while(!isprime(tsize)) tsize++; //如果tsize不是质数，那么重置tsize为 大于tsize的最小素数
    
    vector<int> v(tsize);
    for(int i=0; i<n; i++){
        cin >> a;
        int flag = 0;
        for(int j=0; j<tsize; j++){
            int pos = (a + j*j) % tsize;
            if(v[pos] == 0){
                v[pos] = a;
                flag = 1;
                break;
            }
        }
        if(!flag) cout << a << " cannot be inserted." << endl;
    }
    
    int ans = 0;
    for(int i=0; i<m; i++){
        cin >> a;
        for(int j=0; j<=tsize; j++){
            ans++; //每次进来 记录一次查找的次数
            int pos = (a + j*j) % tsize;
            //在某个位置 查找到 a 直接break
            //因为题目要求输入正数，因此某个位置等于0时候说明此位置没有插入元素
            //即此元素不存在表中 break即可
            if(v[pos] == a || v[pos] == 0) break;
        }
    }
    
    cout << fixed << setprecision(1) << ans * 1.0 / m << endl;
    return 0;
}