結果
問題 |
No.1085 桁和の桁和
|
ユーザー |
![]() |
提出日時 | 2020-06-19 22:59:14 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 3,691 bytes |
コンパイル時間 | 1,855 ms |
コンパイル使用メモリ | 174,972 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-06 17:39:22 |
合計ジャッジ時間 | 3,066 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> #include <iomanip> using namespace std; #define reps(i,s,n) for(int i = s; i < n; i++) #define rep(i,n) reps(i,0,n) #define Rreps(i,n,e) for(int i = n - 1; i >= e; --i) #define Rrep(i,n) Rreps(i,n,0) #define ALL(a) a.begin(), a.end() #define fi first #define se second typedef long long ll; typedef vector<ll> vec; typedef vector<vec> mat; ll N,M,H,W,Q,K,A,B; string S; const ll MOD = 998244353; //const ll MOD = (1e+9) + 7; typedef pair<ll, ll> P; const ll INF = (1LL<<58); class mint{ public: ll x; static unsigned long long mod; static unsigned long long mod_plus; mint(){x = 0;} mint(ll _x) : x((_x < 0 ? ((_x += mod_plus) < 0 ? _x + mod_plus : _x) : _x)%mod){} /*mint& operator=(const ll &a){ x = a; return *this; }*/ mint operator-(){ return x == 0 ? 0 : mod - x; } mint& operator+=(const mint& a){ if((x += a.x) >= mod) x -= mod; return *this; } mint operator+(const mint& a) const{ mint res(*this); return res += a; } mint& operator-=(const mint& a){ if((x -= a.x) < 0) x += mod; return *this; } mint operator-(const mint& a) const{ mint res(*this); return res -= a; } mint& operator*=(const mint& a){ (x *= a.x)%=mod; return *this; } /*mint& operator*=(const ll& a){ mint temp(a); return *this *= temp; }*/ mint operator*(const mint& a) const{ mint res(*this); return res *= a; } /*mint operator*(const ll& a) const{ mint res(*this); return res *= a; }*/ mint pow(unsigned long long pw) const{ mint res(1), comp(*this); while(pw){ if(pw&1) res *= comp; comp *= comp; pw >>= 1; } return res; } //以下、modが素数のときのみ mint inv() const{ mint res(*this); return res.pow(mod - 2); } mint& operator/=(const mint &a){ (x *= a.inv().x)%=mod; return *this; } mint operator/(const mint &a) const{ mint res(*this); return res /= a; } }; ostream& operator<<(ostream& os, const mint& a){ os << a.x; return os; } unsigned long long mint::mod = int(1e+9) + 7; unsigned long long mint::mod_plus = (LLONG_MAX / mint::mod) * mint::mod; typedef vector<mint> vm; vector<vm> mpo(vector<vm> &a, vector<vm> &b){ vector<vm> res(a.size(), vm(b[0].size(), 0)); if(a[0].size() != b.size()) return res; rep(i,a.size()){ rep(j, b[0].size()){ rep(k, b.size()){ res[i][j] += a[i][k] * b[k][j]; } } } return res; } int main() { cin>>S>>K; N = S.size(); vec num(11, 0); rep(i,N) S[i] == '?' ? ++num[10] : ++num[S[i] - '0']; if(K == 0){ cout<<(num[10] + num[0] == N)<<endl; return 0; } ll sum = 0; rep(i,10) sum += num[i] * i; K %= 9; sum %= 9; vector<vm> mv(9, vm(9, 1)), res(9, vm(9, 0)); rep(i,9){ mv[i][i] += 1; res[i][i] += 1; } int temp = num[10]; while(num[10]){ if(num[10]&1) res = mpo(res, mv); mv = mpo(mv, mv); num[10] /= 2; } /* rep(i,10){ rep(j,10) cout<<res[i][j]<<' '; cout<<endl; } */ cout<<res[K][sum] - ll(K == 0 && temp + num[0] == N)<<endl; }