【HDU3709】 Balanced Number

Problem

http://acm.hdu.edu.cn/showproblem.php?pid=3709

Solution

一开始想到的是f[len][pivot][j][k]记录到len，支点为pivot，左边j，右边k的值
看过题解后发现可以直接记录j-k为power_matrix，这样可以少一维，还可以在为负数时剪枝。

CODE

/*
 * @key word:digit
 * @tesed on:HDU3709 15ms
 * @Author: lhq 
 * @Date: 2017-08-16 19:31:38 
 * @Last Modified by: lhq
 * @Last Modified time: 2017-08-16 19:33:09
 */
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
ll dp[20][20][1500];
int a[20];

ll dfs(int len,int pos,int sum,bool limit)
{
    if (len==0) return sum==0?1:0;
    if (sum<0) return 0;
    if (!limit && dp[len][pos][sum]!=-1) return dp[len][pos][sum];
    int up=limit?a[len]:9;
    ll ans=0;
    for (int i=0;i<=up;i++)
        ans+=dfs(len-1,pos,sum+i*(len-pos),limit && i==up);
    if (!limit) dp[len][pos][sum]=ans;
    return ans;
}
ll solve(ll x)
{
    if (x<0) return 0;
    int len=0;
    while (x)
    {
        a[++len]=x%10;
        x/=10;
    }
    ll ans=0;
    for (int i=1;i<=len;i++)
        ans+=dfs(len,i,0,true);
    return ans-len+1;
}
int main()
{
    int T;
    cin>>T;
    memset(dp,-1,sizeof(dp));
    while (T--)
    {
        ll x,y;
        cin>>x>>y;
        cout<<solve(y)-solve(x-1)<<endl;
    }
    return 0;
}