HDU 5936 折半枚举法

Difference

kk说是16年CCPC杭州赛的银牌题

给定$x,K(x<10^9,1\le K\le 9)$求满足$x=f(y,K)-y$的所有正整数y的数量。
$$f(y,K)=\sum _{z\text{ in every digits of }y}{z}^K\left(f(233,2)=2^2+3^2+3^2=22\right)$$

1. 定y的范围，$y<10^{10}$
2. 由于每个数字对f的贡献独立，所以可以折半求解。令$y=10^5\ast a+b$，则$x-(f(a,K)-10^5\ast a)=f(b,K)-b$。先求出1e5以内所有的$f(b,K)-b$，然后进行枚举高5位a，$val=x-(f(a,K)-10^5\ast a)$，二分查找对应的$f(b,K)-b$的数量。
3. 注意：当$x=0$时，会把0算进去。题目求正整数，减去1。

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
#define debug(_x) cout<<#_x<<": "<<_x<<endl;
typedef long long ll;
const int maxn=1e5+3,inf=1e9+1e5;
ll f[10][maxn],d[10][maxn];
ll qpow(int x,int k){
    ll res=1;
    while(k){
        if(k&1)res*=x;
        x=x*x;
        k>>=1;
    }
    return res;
}
void init(){
    int e5=100000;
    for(int j=1;j<=9;++j){
        for(int i=1;i<e5;++i){
            f[j][i] = f[j][i/10] + qpow(i%10,j);
            d[j][i] = f[j][i] - i;
        }
        sort(d[j],d[j]+e5);
    }
}
ll solve(int x,int k){
    int ans=0;
    ll val,e5=100000;
    for(int i=0;i<e5;++i){
        val = x - (f[k][i] - e5 * i);
        ans += upper_bound(d[k],d[k]+e5,val)-lower_bound(d[k],d[k]+e5,val);
    }
    return ans;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int T,x,k;
    init();
    cin>>T;
    for(int casei=1;casei<=T;++casei){
        cin>>x>>k;
        cout<<"Case #"<<casei<<": "<<solve(x,k)+(x ? 0 : -1)<<endl;
    }
}