Balanced Number 数位dp

2019-04-16 20:40:32 阅读：269 来源： 互联网

标签：int scanf d% Number long solve Balanced dp define

题意：　　给出求ab之间有多少个平衡数 4139为平衡数以3为轴 1*1+4*2==9*1

思路很好想但是一直wa ：

注意要减去前导零的情况 0 00 000 0000 不能反复计算

#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define inf 0x3f3f3f3f
#define N 3700+5
#define MID N/2
ll dp[20][20][N];
ll a[20];
ll dfs(int pos,int t,int state,bool lead,bool limit  )
{
    if(!pos)return state==0;

    if(!limit&&!lead&&dp[pos][t][state+MID]!=-1)return dp[pos][t][state+MID];
    ll ans=0;
    int up=limit?a[pos]:9;
    rep(i,0,up)
    {
        ans+=dfs(pos-1,t,state+i*(pos-t), lead&&i==0,limit&&i==a[pos]);
    }
    if(!limit&&!lead)dp[pos][t][state+MID]=ans;
    return ans;
}

ll solve(ll x)
{
    int pos=0;
    ll ans=0;
    while(x)
    {
        a[++pos]=x%10;
        x/=10;
    }
    rep(i,1,pos)
    ans+=dfs(pos,i,0,true,true);
    return ans-pos+1;//去除前导零
}
int main()
{
    int cas;
    CLR(dp,-1);
    RI(cas);
    while(cas--)
    {
        ll a,b;
        cin>>a>>b;
        printf("%lld\n",solve(b)-solve(a-1));
    }
    return 0;
}

View Code

或者

#include<bits/stdc++.h>
using namespace std;
//input by bxd
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m)
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define ll long long
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define inf 0x3f3f3f3f
#define N 3700+5
#define MID N/2
ll dp[20][20][N];
ll a[20];
ll dfs(int pos,int t,int state,bool lead,bool limit  )
{
    if(!pos)return !lead&&state==0;

    if(!limit&&!lead&&dp[pos][t][state+MID]!=-1)return dp[pos][t][state+MID];
    ll ans=0;
    int up=limit?a[pos]:9;
    rep(i,0,up)
    {
        ans+=dfs(pos-1,t,state+i*(pos-t), lead&&i==0,limit&&i==a[pos]);
    }
    if(!limit&&!lead)dp[pos][t][state+MID]=ans;
    return ans;
}

ll solve(ll x)
{
    if(x<0)return 0;
    if(x==0)return 1;
    int pos=0;
    ll ans=0;
    while(x)
    {
        a[++pos]=x%10;
        x/=10;
    }
    rep(i,1,pos)
    ans+=dfs(pos,i,0,true,true);
    return ans+1;//去除前导零
}
int main()
{
    int cas;
    CLR(dp,-1);
    RI(cas);
    while(cas--)
    {
        ll a,b;
        cin>>a>>b;
        printf("%lld\n",solve(b)-solve(a-1));
    }
    return 0;
}

View Code

标签：int,scanf,d%,Number,long,solve,Balanced,dp,define
来源： https://www.cnblogs.com/bxd123/p/10719842.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

Balanced Number 数位dp