題目連結

題意 : 給出一個 x 和 k 問有多少個 y 使得 x = f(y, k) - y 、f(y, k) 為 y 中每個位的數的 k 次方之和、x ≥ 0

分析 :

f(y, k) - y = x ≥ 0

可得 x > y

所以滿足條件的 y 最多不超過 10 位

變化一下式子

x = f(a, k) - a + f(b, k) - b * (1e5)

x - f(a, k) + a = f(b, k) - b * (1e5)

那麼先處理出等式右邊的 f(b,k) - b*(1e5)

這個的複雜度是大於但是不怎麼大於 O(1e5 * 9)

然後再從 0 ~ 1e5 列舉 a 用二分查找出滿足條件的數的個數

#include<bits/stdc++.h>
#define LL long long
using namespace std;

const int maxn = 1e5 + 10;
LL F[maxn][15];
vector<LL> num;
LL x, k;

LL Qpow(LL a, int b)
{
    LL ret = 1;
    while(b){
        if(b & 1) ret = ret * a;
        a = a * a;
        b 
 
>>= 1;
    }return ret;
}

LL GetF(int i, int j)
{
    LL ret = 0;
    while(i){
        ret += Qpow((LL)(i%10), j);
        i /= 10;
    }return ret;
}

inline void init()
{
    for(int i=0; i<=(int)1e5; i++)
        for(int j=1; j<=9; j++)
            F[i][j] = GetF(i, j);
}

LL Solve()
{
    num.clear();

    
 
for(int i=0; i<=(int)1e5; i++) num.push_back(F[i][k] - 1LL * i * (LL)1e5);

    sort(num.begin(), num.end());

    LL ret = 0;
    for(int i=0; i<=(int)1e5; i++)
        ret += upper_bound(num.begin(), num.end(), x - F[i][k] + i) -
               lower_bound(num.begin(), num.end(), x - F[i][k] + i);

    return ret - (LL)(x == 0);
}

int main(void)
{
    init();

    int nCase;
    scanf("%d", &nCase);

    for(int T=1; T<=nCase; T++){

        scanf("%lld %lld", &x, &k);

        printf("Case #%d: %lld\n", T, Solve());
    }
    return 0;
}