Problem H. Curious （莫比烏斯反演）

H - Curious

Gym - 102800H 

題意：給你T組樣例，給n, m, k， n代表陣列元素個數，m代表陣列元素值最大值，k代表詢問個數，然後給你n個元素值a[i], k層詢問，每層詢問給你一個x 問 個數

思路:

 

 

h(x) 就是 1~n 中的元素值取餘x的個數。f(x) = h^2(x), 比如3個元素2 4 6 ，d為2， 那f(d) = 9 （2和2匹配,2和4，2和6，4和2，4和4，4和6，6 和2，6和4，6和6）。

所以 ?(?) = ∑ ?|? ? (?)?( ? ? )。

#include<bits/stdc++.h>
#define up(i, a, b) for(int i = a; i <= b; i++)
#define down(i, a, b)for(int i = a;i >= b; i--)
#define mem(a, b)memset(a, b, sizeof(a))
#define inf 0x3f3f3f3f
#define int long long
using namespace std;
const int maxn = 1e5 + 700;
int mu[maxn], prim[maxn], vis[maxn] = {1,1}, prime_sum = 0;
void init(int n){
    mu[1] = 1;
    up(i, 2, n){
        if(vis[i]==0)prim[++prime_sum] = i, mu[i] = -1;
        for(int j = 1;prim[j]*i <= n && j <= prime_sum; j++){
            vis[prim[j]*i] = 1;
            if(i%prim[j] == 0){
                break;
            }
            else mu[prim[j]*i] = -mu[i];
        }
    }
}
vector<int> v[maxn];
void get(int n){
    up(i, 1, n){
        for(int j = 1; j * j <= i; j++){
            if(i % j == 0){
                v[i].push_back(j);
                if(j*j != i)v[i].push_back(i / j);
            }
        }
    }
}
int arr[maxn], exist[maxn], num[maxn], dp[maxn];
signed main()
{
    init(100005);
    get(100005);

    int T;
    scanf("%lld", &T);
    while(T--){
        int n, m, k;
        scanf("%lld%lld%lld", &n, &m, &k);
        up(i, 1, 100300)exist[i] = 0, dp[i] = -1;
        up(i, 1, n){
            scanf("%lld", &arr[i]);
            exist[arr[i]]++;
        }
        int x;
        while(k--){
            scanf("%lld", &x);
            if(dp[x] != -1){printf("%lld\n", dp[x]);continue;}
            int len = 0, res = 0;
            for(int j = 1;j <= m; j++){
                int X = j*x;
                if(X > m)break;
                len = j;
                for(auto g : v[j]){
                    num[g] += exist[X];
                }
            }
            //求h(x)這裡比較巧妙，
            //j的因子必被  j*x 整除，所以 看j*x有多少arr值，即arr % (j*x)有多少個,一直累加
            //
            up(i, 1, len)res += num[i]*num[i]*mu[i], num[i] = 0;
            //1倍到len倍
            dp[x] = res;//記憶化
            printf("%lld\n", res);
        }
    }

    return 0;
}
/*
2
5 5 5
1 2 3 4 5
1
2
3
4
5
5 5 5
1 2 3 4 5
1
2
3
4
5
*/