HDU 5085/BC 15D Counting problem

题意：设函数f(x,k)为x的每一位的k次方的和，求a到b中有多少i满足,f(i,k)=S   (a,b,k,S给定)

题解：

自己想的比较复杂 首先k为1 2时可以特殊处理 因为此时的S也很小，正常的数位dp就可以了

而k>2时 可以先拆分S 这时S的拆分种数较少，对每一种再进行计算即可。

但这样不好敲啊=。= 忍不住看了题解，果然机智，分治即可，预处理储存S的后几位，再枚举S的前几位既可。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#include<list>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<deque>
#define mem(x,y) memset(x,y,sizeof(x))
#define pb push_back
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,double> pii;
#define bug puts("===========");
#define zjc puts("");
const double pi=(acos(-1.0));
const double eps=1e-8;
const ll INF=1e18+10;
const ll inf=1e9+10;
const int mod=1e9+7;
const int maxn=1000+10;
const int maxm=1e5+5;
ll p[15][20];
ll fuck[maxm][17];
map<ll,ll>mp;
ll cnt(ll n,ll k){
    ll x=0;
    while(n){
        x += p[n%10][k];
        n/=10;
    }
    return x;
}
ll go(ll n,ll k,ll s){
    if(n<=1e6){
        ll ans=0;
        for(ll i=1;i<=n;i++)
            if(cnt(i,k)==s) ans++;
        return ans;
    }
    mp.clear();
    ll len=1e4;
    for(ll i=0;i<len;i++){
        ll now=fuck[i][k];
        mp[now]++;
    }
    ll nn=n/len,ans=mp[s];
    for(ll i=1;i<nn;i++){
        ll now=fuck[i][k];
        if(now<=s&&mp.count(s-now))
        ans+=mp[s-now];
    }
    ll z=cnt(nn,k),nnn=n%len;
    for(ll j=0;j<=nnn;j++){
        if(fuck[j][k]==s-z) ans++;
    }
    return ans;
}
int main()
{
    for(ll i=1;i<=9;i++){
        p[i][0]=1;
        for(ll j=1;j<16;j++){
            p[i][j]=p[i][j-1]*i;
        }
    }
    int len=1e5;
    for(int k=1;k<=15;k++)
    for(int i=1;i<=len;i++)
        fuck[i][k]=cnt(i,k);
    ll a,b,k,s;
    while(~scanf("%I64d%I64d%I64d%I64d",&a,&b,&k,&s)){
       ll ans=go(b,k,s)-go(a-1,k,s);
        printf("%I64d\n",ans);
    }
}