2020牛客暑期多校训练营（第六场）H-Harmony Pairs (数位 dp)

最新推荐文章于 2022-05-29 11:07:30 发布

stduy_ing

最新推荐文章于 2022-05-29 11:07:30 发布

阅读量375

点赞数 1

分类专栏： dp 数位 dp

本文链接：https://blog.csdn.net/hddddh/article/details/107648814

版权

dp 同时被 2 个专栏收录

19 篇文章 1 订阅

订阅专栏

数位 dp

8 篇文章 1 订阅

订阅专栏

链接： H-Harmony Pairs
在这里插入图片描述
题意：
求 0 <= A <= B 中 S(A) > S(B)的 A，B 的对数，S(A)是 A的各位之和。
思路：
首先 S（A）> S(B),很好解决，dfs维护一个 A B 的当前数位差就好了，还有就是 A <= B,再来一个状态表示当前所有位 A和B的每一位是不是都相同就好了，若都相同 B的当前位必须大于等于 A的当前位，否则就随便取。所以 dp记录当前位置，数位差，A是否等于 B, 再记录一下两个 limit （一开始没记录 T 了，不知道为啥）

#include <iostream>
#include <cstdio>
#include <map>
#include <math.h>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e2 + 7;
const int mod=1e9 + 7;
ll dp[maxn][2000+7][3][3][3];
int a[maxn];
char s[maxn];
ll dfs(int pos, int d, int limit1, int limit2,int statue){
    if(pos == -1){
       return d > 0 ;
    }
    if(dp[pos][d + 1000][statue][limit1][limit2]!=-1) {
        return dp[pos][d+1000][statue][limit1][limit2] % mod;
    }
    int up1 = limit1 ? a[pos] : 9 ;
    int up2 = limit2 ? a[pos] : 9 ;
    ll temp = 0;
    for(int i = 0; i <= up1; i++){
        int l = statue ? i : 0 ;
        for(int j = l; j <= up2; j++){
            int dd = d + i - j;
            temp=(temp + dfs(pos - 1, dd, limit1&&(i==a[pos]),limit2&&(j==a[pos]),statue&&(i==j))) % mod;
        }
    }
    dp[pos][d+1000][statue][limit1][limit2] = temp % mod;
    return temp;
}

int main(){
    scanf("%s",&s);
    memset(dp,-1,sizeof(dp));
    int len=strlen(s);
    for(int i = 0; i < len; i ++){
       a[i] = s[len - i -1] - '0';
    }
    ll ans = dfs(len-1, 0, 1, 1, 1);
    printf ("%lld\n",ans % mod); 
}