結果
問題 | No.1263 ご注文は数学ですか? |
ユーザー | chocorusk |
提出日時 | 2020-10-17 17:22:21 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 6,901 bytes |
コンパイル時間 | 1,801 ms |
コンパイル使用メモリ | 149,808 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-21 02:53:44 |
合計ジャッジ時間 | 2,394 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | WA | - |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | AC | 2 ms
5,376 KB |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #define popcount __builtin_popcount using namespace std; typedef long long ll; typedef pair<int, int> P; template<int MOD> struct ModInt{ int x; ModInt(): x(0){} ModInt(ll y): x(y>=0 ? y%MOD : (MOD-(-y)%MOD)%MOD){} ModInt &operator+=(const ModInt &p){ if((x+=p.x)>=MOD) x-=MOD; return *this; } ModInt &operator-=(const ModInt &p){ if((x+=MOD-p.x)>=MOD) x-=MOD; return *this; } ModInt &operator*=(const ModInt &p){ x=(int)(1ll*x*p.x%MOD); return *this; } ModInt &operator/=(const ModInt &p){ *this*=p.inv(); return *this; } ModInt operator-() const{ return ModInt(-x);} ModInt operator+(const ModInt &p) const{ return ModInt(*this)+=p;} ModInt operator-(const ModInt &p) const{ return ModInt(*this)-=p;} ModInt operator*(const ModInt &p) const{ return ModInt(*this)*=p;} ModInt operator/(const ModInt &p) const{ return ModInt(*this)/=p;} bool operator==(const ModInt &p) const{ return x==p.x;} bool operator!=(const ModInt &p) const{ return x!=p.x;} ModInt pow(ll n) const{ ModInt ret(1), p(x); while(n){ if(n&1) ret*=p; p*=p; n>>=1; } return ret; } ModInt inv() const{ return pow(MOD-2); } }; const int MOD=1e9+7; using mint=ModInt<MOD>; template<typename T> struct Matrix{ vector<vector<T>> a; Matrix(){} Matrix(size_t n, size_t m):a(n, vector<T>(m, 0)){} Matrix(size_t n):Matrix(n, n){} Matrix(vector<vector<T>> a):a(a){} size_t height() const{ return a.size(); } size_t width() const{ return a[0].size(); } inline const vector<T> &operator[](size_t k) const{ return a[k]; } inline vector<T> &operator[](size_t k){ return a[k]; } static Matrix I(size_t n){ Matrix mat(n); for(int i=0; i<n; i++) mat[i][i]=1; return mat; } Matrix &operator+=(const Matrix &b){ size_t n=height(), m=width(); for(int i=0; i<n; i++){ for(int j=0; j<m; j++){ (*this)[i][j]+=b[i][j]; } } return (*this); } Matrix &operator-=(const Matrix &b){ size_t n=height(), m=width(); for(int i=0; i<n; i++){ for(int j=0; j<m; j++){ (*this)[i][j]-=b[i][j]; } } return (*this); } Matrix &operator*=(const Matrix &b){ size_t n=height(), m=width(), l=b.width(); vector<vector<T>> c(n, vector<T>(l, 0)); for(int i=0; i<n; i++){ for(int j=0; j<l; j++){ for(int k=0; k<m; k++){ c[i][j]+=(*this)[i][k]*b[k][j]; } } } a.swap(c); return (*this); } Matrix operator+(const Matrix &b) const{ return (Matrix(*this)+=b); } Matrix operator-(const Matrix &b) const{ return (Matrix(*this)-=b); } Matrix operator*(const Matrix &b) const{ return (Matrix(*this)*=b); } Matrix pow(ll k) const{ Matrix ap(a), ret=I(height()); while(k){ if(k&1) ret*=ap; ap*=ap; k>>=1; } return ret; } static pair<Matrix, Matrix> Gauss_Jordan(const Matrix &a, const Matrix &b){ size_t n=a.height(), m=a.width(), l=b.width(); Matrix c(n, m+l); for(int i=0; i<n; i++) for(int j=0; j<m; j++) c[i][j]=a[i][j]; for(int i=0; i<n; i++) for(int j=0; j<l; j++) c[i][j+m]=b[i][j]; int d=0; for(int i=0; i<m; i++){ int p=-1; for(int j=d; j<n; j++){ if(c[j][i]!=0){ p=j; break; } } if(p==-1) continue; swap(c[p], c[d]); T invc=T(1)/c[d][i]; for(int j=i; j<m+l; j++) c[d][j]*=invc; for(int j=0; j<n; j++){ if(j==d) continue; T c0=c[j][i]; for(int k=i; k<m+l; k++){ c[j][k]-=c0*c[d][k]; } } d++; } Matrix reta(n, m), retb(n, l); for(int i=0; i<n; i++) for(int j=0; j<m; j++) reta[i][j]=c[i][j]; for(int i=0; i<n; i++) for(int j=0; j<l; j++) retb[i][j]=c[i][j+m]; return make_pair(reta, retb); } static pair<vector<T>, vector<vector<T>>> linear_equations(const Matrix &a, const vector<T> &b){ int n=a.height(), m=a.width(); Matrix B(n, 1); for(int i=0; i<n; i++) B[i][0]=b[i]; auto p=Gauss_Jordan(a, B); vector<int> myon(n,-1); vector<int> nuo(m, -1); for(int i=0; i<n; i++){ bool allzero=1; for(int j=0; j<m; j++){ if(p.first[i][j]!=0){ allzero=0; myon[i]=j; nuo[j]=i; break; } } if(allzero && p.second[i][0]!=0){ vector<T> retc; vector<vector<T>> retd; return make_pair(retc, retd); } } vector<T> c(m); vector<vector<T>> d; for(int i=0; i<m; i++){ if(nuo[i]==-1){ vector<T> v(m); v[i]=1; for(int j=0; j<n; j++){ if(myon[j]!=-1) v[myon[j]]=-p.first[j][i]; } d.push_back(v); }else{ c[i]=p.second[nuo[i]][0]; } } return make_pair(c, d); } Matrix inv() const{ int n=height(); Matrix b=I(n); auto p=Gauss_Jordan(*this, b); if(p.first[n-1][n-1]==0){ Matrix ret(0); return ret; } return p.second; } int rank() const{ int n=height(), m=width(); Matrix b(n, 0); auto p=Gauss_Jordan(*this, b); for(int i=0; i<n; i++){ bool allzero=1; for(int j=0; j<m; j++){ if(p.first[i][j]!=0){ allzero=0; break; } } if(allzero) return i; } return n; } T det() const{ size_t n=height(); Matrix A(a); T ret(1); for(int i=0; i<n; i++){ int p=-1; for(int j=i; j<n; j++){ if(A[j][i]!=0){ p=j; break; } } if(p==-1){ return 0; } if(p!=i) ret*=(-1); swap(A[p], A[i]); ret*=A[i][i]; T inva=T(1)/A[i][i]; for(int j=i+1; j<n; j++){ T a0=A[j][i]; for(int k=i; k<n; k++){ A[j][k]-=inva*a0*A[i][k]; } } } return ret; } }; using mat=Matrix<mint>; vector<vector<int>> w; void gen_bunkatu(vector<int> v, int t, int s){ if(t==s){ vector<int> vs=v; reverse(vs.begin(), vs.end()); w.push_back(vs); return; } int m=s; if(t) m=v.back(); for(int i=1; i<=min(m, s-t); i++){ v.push_back(i); gen_bunkatu(v, t+i, s); v.pop_back(); } } int main() { int x; cin>>x; vector<int> v0; gen_bunkatu(v0, 0, x); int n=w.size(); mat a(n); vector<mint> b(n); for(int i=0; i<n; i++){ vector<int> v; int t=0; for(int j=0; j<w[i].size(); j++){ for(int k=0; k<w[i][j]; k++) v.push_back(t); t++; } if(w[i].size()==x) b[i]=mint(1); for(int j=0; j<n; j++){ bool ok=0; int s=0; do{ bool dame=0; for(auto p:w[j]){ for(int k=s; k<s+p-1; k++){ if(v[k]!=v[k+1]){ dame=1; } } s+=p; if(dame) break; } if(!dame){ ok=1; break; } }while(next_permutation(w[j].begin(), w[j].end())); if(ok) a[i][j]=mint(1); } } vector<mint> c=mat::linear_equations(a, b).first; mint ans(1); for(auto z:c) ans*=z; cout<<ans.x<<endl; return 0; }