寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

アルゴ式(beta版): C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. 残った数を K の倍数に

アルゴ式(beta版)の「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。

C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. 残った数を K の倍数に


僕が作成、提出したコードは、以下のとおりです。

Q. 残った数を K の倍数に

algo-method.com

/*
アルゴ式(beta版): C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. 残った数を K の倍数に
Q. 残った数を K の倍数に
https://algo-method.com/tasks/1110
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const ll mod=1e9+7;
int main(void){
    ll n,k;
    cin>>n>>k;
    string s;
    cin>>s;
    vector<vector<ll>> dp(n+1,vector<ll>(k,0));
    dp[0][0]=1;
    for(ll i=0;i<n;i++){
        ll j=s[i]-48;
        for(ll l=0;l<k;l++){
            dp[i+1][l]+=dp[i][l];
            dp[i+1][l]%=mod;
            ll m=(l*10+j)%k;
            dp[i+1][m]+=dp[i][l];
            dp[i+1][l]%=mod;
        }
    }
    ll ans=dp[n][0]-1;
    ans=(ans+mod)%mod;
    cout<<ans<<endl;
    return 0;
}

設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com