Submission #501315

#	Time	Username	Problem	Language	Result	Execution time	Memory
501315		inksamurai	Zapina (COCI20_zapina)	C++17	110 / 110	126 ms	1612 KiB

zapina

#include <bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) a.begin(),a.end()
#define rep(i,n) for(int i=0;i<n;i++)
#define crep(i,x,n) for(int i=x;i<n;i++)
#define drep(i,n) for(int i=n-1;i>=0;i--)
#define vec(...) vector<__VA_ARGS__>
#define _3qplfh5 ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
typedef long long ll;
typedef long double ld;
using pii=pair<int,int>;
using vi=vector<int>;
using vll=vector<long long>;

//snuke's mod int
template <ll mod>
struct modint{
  ll x; // typedef long long ll;
  modint(ll x=0):x((x%mod+mod)%mod){}
  modint operator-()const{ return modint(-x);}
  modint& operator+=(const modint a){ if((x+=a.x)>=mod) x-=mod; return *this; }
  modint& operator-=(const modint a){ if((x+=mod-a.x)>=mod) x-=mod; return *this; }
  modint& operator*=(const modint a){ (x*=a.x)%=mod; return *this; }
  modint operator+(const modint a)const{ modint res(*this); return res+=a; }
  modint operator-(const modint a)const{ modint res(*this); return res-=a; }
  modint operator*(const modint a)const{ modint res(*this); return res*=a; }
  modint pow(ll n) const {
    modint res=1,x(*this);
    while(n){
        if(n&1)res*=x;
        x*=x;
        n>>=1;
    }
    return res;
  }
  modint inv() const { return pow(mod-2); }
};
using mint=modint<1000000007>;

const int mxn=400;

mint cnktable[mxn][mxn];

void precomb(){
	rep(i,mxn-1){
		rep(j,mxn-1){
			if(!i){
				cnktable[i][j]=!j;
			}else{
				cnktable[i][j]=(i==j or j==0?1:cnktable[i-1][j-1]+cnktable[i-1][j]);
			}
		}
	}
}

int main(){
_3qplfh5;
precomb();
	int n;
	cin>>n;
	vec(vec(mint)) dp(n+1,vec(mint)(2)),nedp;
	dp[0][0]=1;
	rep(i,n){
		int need=i+1;
		nedp=vec(vec(mint))(n+1,vec(mint)(2));
		rep(j,n+1){
			rep(k,(n-j)+1){
				rep(t,2){
					int net=t;
					if(k==need) net=1;
					nedp[j+k][net]+=dp[j][t]*cnktable[n-j][k];
				}
			}
		}
		swap(dp,nedp);
		// cout<<dp[1][1].x<<"\n";
	}
	cout<<dp[n][1].x<<"\n";
//	
	return 0;
}

• Subtask 1: 22 / 22
• Subtask 2: 33 / 33
• Subtask 3: 55 / 55