Balanced Number HDU - 3709 （数位DP）

Balanced Number

题目链接： HDU - 3709

题意：若能在一个数中找到一个位置x， 使得该数其他位置得数以x为界限分为左右两组， 并每组的价值等于每个数位上的数乘以他到x的距离之和；如4129：以2的位置为界限， 左边为：4*2+1*1， 右边为9*1；当左右价值相同时称之为平衡数， 求[x, y]有多少平衡数；

若一个数为平衡数， 则位置x是固定不变的， 即只能找到一个x；

那么数位DP时我们需要将x的所有位置都找一下，最后减去找的x的位置个数len-1， 因为当全是0时相当于找了len次， 多找了len-1次；

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <queue>
using namespace std;
long long dp[20][20][2000];
int digit[20];
long long dfs(int pre, int index, int sum, int limit){
	if(pre==-1) return sum==0?1:0;
	if(sum<0) return 0;
	if(!limit&&dp[pre][index][sum]!=-1) return dp[pre][index][sum];
	int up=limit?digit[pre]:9;
	long long ans=0;
	for(int i=0; i<=up; i++){
		ans+=dfs(pre-1, index, sum+i*(pre-index), limit&&i==digit[pre]);
	}
	if(!limit) dp[pre][index][sum]=ans;
	return ans;
}
long long solve(long long x){
	int pre=0;
	while(x){
		digit[pre++]=x%10;
		x/=10;
	}
	long long ans=0;
	for(int i=0; i<pre; i++){
		ans+=dfs(pre-1, i, 0, 1);
	}
	return ans-(pre-1);
}
int main(){
	int T;
	long long x, y; 
	cin >> T;
	memset(dp, -1, sizeof(dp));
	while(T--){
		cin >> x >> y;
		cout << solve(y)-solve(x-1) << endl;
	}
	return 0;
}