Hướng dẫn giải của Atcoder Educational DP Contest S - Digit Sum
Chỉ dùng lời giải này khi không có ý tưởng, và đừng copy-paste code từ lời giải này. Hãy tôn trọng người ra đề và người viết lời giải.
Nộp một lời giải chính thức trước khi tự giải là một hành động có thể bị ban.
Nộp một lời giải chính thức trước khi tự giải là một hành động có thể bị ban.
Tác giả:
Độ phức tạp thời gian:
Ta có thể dùng quy hoạch động chữ số để giải bài toán này. Gọi
Để tính đáp án, ta duyệt qua tất cả các tiền tố của xâu
Copy#include <bits/stdc++.h> using namespace std; const int MAXL = 10000; const int MAXD = 100; const int MOD = 1e9 + 7; int dp[MAXL + 1][MAXD]; int add(int i, int j) { if ((i += j) >= MOD) i -= MOD; return i; } void init(int l, int d) { dp[l][0] = 1; for (int i = l - 1; i >= 0; --i) for (int j = 0; j < d; ++j) for (int k = 0; k < 10; ++k) dp[i][j] = add(dp[i][j], dp[i + 1][(j + k) % d]); } int query(const string& s, int d) { int ans = 0, carry = 0; for (int i = 0; i < (int) s.length(); ++i) { for (int j = 0; j < (s[i] - '0'); ++j) ans = add(ans, dp[i + 1][(carry + j) % d]); carry = (carry + (s[i] - '0')) % d; } if (carry == 0) ans++; if (--ans < 0) ans += MOD; return ans; } int main() { cin.tie(0)->sync_with_stdio(0); string s; int d; cin >> s >> d; init((int) s.size(), d); cout << query(s, d) << '\n'; }
Bình luận