HDU - 3709(数位DP)

题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=3709

依据题目意思,建立DP三维状态,dp[pos][mid][sum]表示第pos位以mid为中点建立的力矩值为sum的个数,

直接枚举每一位为中心点的情况即可,直接上个板子。

(但是莽夫做题选手注意一下全为0的情况,需要减去)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<queue>
#define MAX_len 50100*4
using namespace std;
typedef long long ll;
const ll mod=1000000007;
int a[70];
ll dp[25][25][2000];
ll dfs(int pos,int mid,int sum,bool limit)
{
	if(pos<1)
	{
		if(sum==0)
		return 1;
		return 0;
	}
    if(!limit&&dp[pos][mid][sum]!=-1)
	return dp[pos][mid][sum];
    int up=limit?a[pos]:9;
   	ll temp=0;
    for(int i=0;i<=up;i++)
    {
    	temp+=dfs(pos-1,mid,sum+(pos-mid)*i,limit&&i==up);
    }
    if(!limit)
	dp[pos][mid][sum]=temp;
    return temp;
}
ll solve(ll n)
{ 
	if(n<0)
	return 0; 
	int len=1;
	while(n)
	{
		a[len++]=n%10;
		n/=10;
	}
	ll ans=0;
	for(int i=1;i<len;i++)
	ans+=dfs(len-1,i,0,true);
	return ans-(len-1)+1;
}
int main()
{
	memset(dp,-1,sizeof(dp));
	int T;
	scanf("%d",&T);
	while(T--)
	{
		ll l,r;
		scanf("%I64d %I64d",&l,&r);
		printf("%I64d\n",solve(r)-solve(l-1));
	}
	return 0;
}
发布了56 篇原创文章 · 获赞 17 · 访问量 2336

猜你喜欢

转载自blog.csdn.net/weixin_43958964/article/details/100159516
今日推荐