BZOJ1799 [self 同类分布]

最新推荐文章于 2019-07-22 09:24:45 发布

Venishel

最新推荐文章于 2019-07-22 09:24:45 发布

阅读量489

点赞数

分类专栏：数位dp 文章标签： dp bzoj bzoj1799 数位DP oj

本文链接：https://blog.csdn.net/qq_37321281/article/details/74427157

版权

数位dp 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

Description

给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。

Sample Input

10 19

Sample Output

3

HINT

【约束条件】1 ≤ a ≤ b ≤ 10^18

题解：数位DP

枚举数字之和

dp[pos][sta1][sta2]表示在第pos位的数为sta2,膜为sta1时累计的数的个数

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
#define LL long long
LL a[20], dp[20][180][180];

LL dfs( int pos, int p, int sta1, int sta2, int flg ){
   if ( pos == -1 ) return (sta1 == 0 && sta2 == p );
   if ( sta2 > p || sta2 + (pos+1)*9 < p ) return 0;
   if ( !flg && dp[pos][sta1][sta2] != -1 ) return dp[pos][sta1][sta2];
   int up = flg ? a[pos] : 9;
   LL ret = 0;
   for ( int i = 0; i <= up; i++) ret += dfs( pos-1, p, (sta1*10+i)%p, sta2 + i, flg && ( i == up ) );
   if ( !flg ) dp[pos][sta1][sta2] = ret;
   return ret;
}
LL solve( LL x ){
    int pos = 0;
    for( ; x; a[pos++] = x % 10, x /= 10);
    if( !pos ) return 0ll;
    LL ans = 0;
    for ( int i = 1; i <= pos*9; i++){
        memset( dp, -1, sizeof(dp)); 
         ans += dfs( pos-1, i, 0, 0, 1);
    }
    return ans;
}
int main(){
   LL l, r;
    cin>>l>>r;
    cout<<solve(r)-solve(l-1)<<endl;    
}