結果
問題 | No.997 Jumping Kangaroo |
ユーザー | yukudo |
提出日時 | 2020-02-21 22:41:13 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,699 bytes |
コンパイル時間 | 2,141 ms |
コンパイル使用メモリ | 177,520 KB |
実行使用メモリ | 6,824 KB |
最終ジャッジ日時 | 2024-10-08 22:15:54 |
合計ジャッジ時間 | 3,753 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 58 ms
5,248 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | AC | 8 ms
5,248 KB |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | AC | 2 ms
6,816 KB |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | WA | - |
testcase_23 | WA | - |
testcase_24 | WA | - |
testcase_25 | WA | - |
testcase_26 | AC | 2 ms
6,820 KB |
testcase_27 | AC | 2 ms
6,820 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; #define REP(i,n) for(int i=0,_n=(int)(n);i<_n;++i) #define ALL(v) (v).begin(),(v).end() #define CLR(t,v) memset(t,(v),sizeof(t)) template<class T1,class T2>ostream& operator<<(ostream& os,const pair<T1,T2>&a){return os<<"("<<a.first<<","<<a.second<< ")";} template<class T>void pv(T a,T b){for(T i=a;i!=b;++i)cout<<(*i)<<" ";cout<<endl;} template<class T>void chmin(T&a,const T&b){if(a>b)a=b;} template<class T>void chmax(T&a,const T&b){if(a<b)a=b;} int nextInt() { int x; scanf("%d", &x); return x;} ll nextLong() { ll x; scanf("%lld", &x); return x;} const ll MOD = 1e9 + 7; struct mint { ll x; mint(ll x=0):x((x%MOD+MOD)%MOD){} mint& operator+=(const mint a) {if ((x += a.x) >= MOD) x -= MOD;return *this;} mint& operator-=(const mint a) {if ((x += MOD-a.x) >= MOD) x -= MOD;return *this;} mint& operator*=(const mint a) {(x *= a.x) %= MOD;return *this;} mint operator+(const mint a) const {mint res(*this);return res+=a;} mint operator-(const mint a) const {mint res(*this);return res-=a;} mint operator*(const mint a) const {mint res(*this);return res*=a;} mint pow(ll b) const { mint res(1), a(*this); while (b) { if (b & 1) res *= a; a *= a; b >>= 1; } return res; } // for prime MOD mint inv() const {return pow(MOD-2);} mint& operator/=(const mint a) {return (*this) *= a.inv();} mint operator/(const mint a) const {mint res(*this);return res/=a;} }; #define SZ(v) ((int)(v).size()) using Array = vector<ll>; using Matrix = vector<Array>; Matrix zero(int N){ return Matrix(N, Array(N)); } Matrix identity(int N) { Matrix A = zero(N); REP(i, N) A[i][i] = 1; return A; } Matrix add(const Matrix &A, const Matrix& B){ const int N = SZ(A); const int M = SZ(A[0]); Matrix C(N, Array(M)); REP(i, N) REP(j, M) { C[i][j] += A[i][j] + B[i][j]; if (C[i][j] >= MOD) C[i][j] %= MOD; } return C; } Array mul(const Matrix &A, const Array &x){ const int N = SZ(A); const int M = SZ(A[0]); Array y(N); REP(i, N) REP(j, M) y[i] += A[i][j] * x[j]; return y; } // A:[N,P] * B:[P,M] = C:[N,M] Matrix mul(const Matrix &A, const Matrix& B) { const int N = SZ(A); const int P = SZ(A[0]); const int M = SZ(B[0]); Matrix C(N, Array(M)); REP(i, N) REP(j, M) REP(k, P) { C[i][j] += A[i][k] * B[k][j]; if (C[i][j] >= MOD) C[i][j] %= MOD; } return C; } Matrix pow(Matrix A, ll b) { Matrix C = identity(SZ(A)); while (b > 0) { if ((b & 1) == 1) C = mul(C, A); A = mul(A, A); b >>= 1; } return C; } int main2() { int N = nextInt(); int W = nextInt(); ll K = nextLong(); vector<int> A(N); REP(i, N) A[i] = nextInt(); Matrix X = identity(2*W); mint ways[2*W][2]; REP(w0, W) REP(j0, 2) { REP(w, 2*W) REP(j, 2) ways[w][j] = 0; ways[w0][j0] = 1; REP(w, W) REP(j, 2) { REP(i, N) { int nw = w + A[i]; if (nw < W) { ways[nw][j] += ways[w][j]; } else if (nw <= 2*W) { int nj = (nw == W) ? 0 : 1; if (j == 1 && nj == 1) continue; ways[nw][nj] += ways[w][j]; } } } for (int w = W; w < 2*W; w++) { for (int j = 0; j < 2; j++) { int a = (w - W) + j *W; int b = (w0 ) + j0*W; X[b][a] = ways[w][j].x; } } } // REP(i, 2*W) { // REP(j, 2*W) { // cout << X[i][j] << " "; // } // cout << endl; // } X = pow(X, K); Array b = Array(2*W); b[0] = 1; auto v = mul(X, b); ll ans = v[0]; cout << ans << endl; return 0; } int main() { #ifdef LOCAL for (;!cin.eof();cin>>ws) #endif main2(); return 0; }