FZU 2113 Jason的特殊愛好 [數位DP]

題意:輸出a到b之間的整數包含多少個1。

解法:數位DP,滿足區間減法,所以可以考慮getsum(R)-getsum(L-1)。

因爲直接DP想法不是很直接,所以一般用dfs+記憶化來寫比較舒服,板子一般是這樣的:

pos代表搜索到第幾位(高到低)

state是一些關於題目所求的參數(可以多個)

limit是用來限制上下界的,一些不滿足區間減法的題需要兩個limit,一個限制上界,一個限制下界)

dfs(int pos,int state,int limit){    //這裏單獨一個limit,表示上界

if(pos==0)return [state]

if( ! limit && ~ dp[pos][state]) return dp[pos][state]   //DP初值爲-1,而這樣返回DP,可以讓DP數組反覆利用(一般情況下)

        int d=9,sum=0;

        if(limit)d=num[pos];

        rep(i,0,d){

              sum+=dfs(pos-1,new state, limit && i==d) ;

               // 關於limit參數:如果前一位達到上界且這一位也是上界,那麼之後的數不能隨便取,否則可以取0-9 

        }

       if( ! limit) dp[pos][state] =sum; //沒有被limit限制纔可以賦值給dp,也就是一般情況,沒有上界限制的情況下

       return sum;

}

對於此題,只需要將state表示成1的個數即可。

代碼:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=int(a);i<=int(b);i++)
#define drep(i,a,b) for(int i=int(a);i>=int(b);i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef unsigned __int64 ll;
typedef pair<int,int> pii;

ll dp[20][20][2];
ll num[20];
ll dfs(ll pos,ll pre,bool lim){
    if(pos==0)return pre;
    if(~dp[pos][pre][lim])return dp[pos][pre][lim];
    ll sum=0;
    ll d=9;

    if(lim)d=num[pos];
    rep(i,0,d){
        if(i==1)sum+=dfs(pos-1,pre+1,lim&&i==d);
        else sum+=dfs(pos-1,pre,lim&&i==d);
    }
    return dp[pos][pre][lim]=sum;
}
ll solve(ll x){
    ll len=0;
    ll t=x;
    while(t){
        num[++len]=t%10;
        t/=10;
    }
    mem(dp,-1);
    return dfs(len,0,1);
}

ll l,r;
int main(){
    while(scanf("%I64u%I64u",&l,&r)!=EOF){
        mem(dp,-1);
        if(l>r)swap(l,r);
        printf("%I64u\n",solve(r)-solve(l-1));
    }
    return 0;
}




發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章