BNUOJ 52325 Increasing or Decreasing 數位dp

傳送門:BNUOJ 52325 Increasing or Decreasing
題意:求[l,r]非遞增和非遞減序列的個數
思路:數位dp,dp[pos][pre][status]
  1. pos:處理到第幾位
  2. pre:前一位是什麼
  3. status:是否有前導零

遞增遞減差很少思路,不過他們計算的過程當中像5555,444 這樣的重複串會多算,因此要剪掉。個數是(pos-1)*9+digit[最高位],好比一位重複子串是:1,2,3,4...9,9個,二位重複子串:11,22,33,44,...,99,9個;同理,其餘類推;php

不過這個題若是dp值每算完一個[l,r]就清零,會超時。那麼咱們這麼分析,算[l1,r1],[l2,r2]這兩個區間時,dp是否真的有必要清零呢,答案是否認的,記憶化搜索的過程當中記錄的dp值若是計算過,那麼當其餘值算到他時,這個值是能夠用的。具體的本身想一想就行了ios

/**************************************************************
    Problem:BNUOJ 52325 Increasing or Decreasing
    User: youmi
    Language: C++
    Result: Accepted
    Time:    380 ms
    Memory:    1632 KB
****************************************************************/
//#pragma comment(linker, "/STACK:1024000000,1024000000")
//#include<bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <cmath>
#include <queue>
#include <deque>
#include <string>
#include <vector>
#define zeros(a) memset(a,0,sizeof(a))
#define ones(a) memset(a,-1,sizeof(a))
#define sc(a) scanf("%d",&a)
#define sc2(a,b) scanf("%d%d",&a,&b)
#define sc3(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define scs(a) scanf("%s",a)
#define sclld(a) scanf("%lld",&a)
#define pt(a) printf("%d\n",a)
#define ptlld(a) printf("%lld\n",a)
#define rep(i,from,to) for(int i=from;i<=to;i++)
#define irep(i,to,from) for(int i=to;i>=from;i--)
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
#define lson (step<<1)
#define rson (lson+1)
#define eps 1e-6
#define oo 0x3fffffff
#define TEST cout<<"*************************"<<endl
const double pi=4*atan(1.0);

using namespace std;
typedef long long ll;
template <class T> inline void read(T &n)
{
    char c; int flag = 1;
    for (c = getchar(); !(c >= '0' && c <= '9' || c == '-'); c = getchar()); if (c == '-') flag = -1, n = 0; else n = c - '0';
    for (c = getchar(); c >= '0' && c <= '9'; c = getchar()) n = n * 10 + c - '0'; n *= flag;
}
ll Pow(ll base, ll n, ll mo)
{
    ll res=1;
    while(n)
    {
        if(n&1)
            res=res*base%mo;
        n>>=1;
        base=base*base%mo;
    }
    return res;
}
//***************************

int n;
const int maxn=100000+10;
const ll mod=1000000007;
int digit[30];
ll dp0[20][20][2];
ll dp1[20][20][2];
int tot=0;
ll dfs0(int pos,int pre,int status,int limit)
{
    if(pos<0)
        return status;
    if(!limit&&dp0[pos][pre][status]!=-1)
        return dp0[pos][pre][status];
    int ed=limit?digit[pos]:9;
    ll res=0;
    if(status==0)
    {
        for(int i=0;i<=min(pre,ed);i++)
        {
            if(i==0)
                res+=dfs0(pos-1,10,0,limit&&(i==ed));
            else
                res+=dfs0(pos-1,i,1,limit&&(i==ed));
        }
    }
    else
    {
        for(int i=0;i<=min(ed,pre);i++)
            res+=dfs0(pos-1,i,status,limit&&(i==ed));
    }
    if(!limit)
        dp0[pos][pre][status]=res;
    return res;
}
ll dfs1(int pos,int pre,int status,int limit)
{
    if(pos<0)
        return status;
    if(!limit&&dp1[pos][pre][status]!=-1)
        return dp1[pos][pre][status];
    int ed=limit?digit[pos]:9;
    ll res=0;
    for(int i=pre;i<=ed;i++)
        res+=dfs1(pos-1,i,status||i,limit&&(i==ed));
    if(!limit)
        dp1[pos][pre][status]=res;
    return res;
}
void work(ll num)
{
    tot=0;
    while(num)
    {
        digit[tot++]=num%10;
        num/=10;
    }
}
ll solve(ll num)
{
    if(num==0)
        return 0;
    ll ans=(tot-1)*9+digit[tot-1];
    ll temp=0;
    int tt=0;
    while(tt<tot)
        temp=temp*10+digit[tot-1],tt++;
    if(temp>num)
        ans--;
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int T_T;
    scanf("%d",&T_T);
    ones(dp0);
    ones(dp1);
    for(int kase=1;kase<=T_T;kase++)
    {
        ll num;
        read(num);
        num--;
        work(num);
        ll temp0=dfs0(tot-1,10,0,1);
        temp0+=dfs1(tot-1,0,0,1);
        temp0-=solve(num);
        read(num);
        work(num);
        ll temp1=dfs0(tot-1,10,0,1);
        temp1+=dfs1(tot-1,0,0,1);
        temp1-=solve(num);
        ptlld(temp1-temp0);
    }
    return 0;
}
相關文章
相關標籤/搜索