Submission #442913

#	Time	Username	Problem	Language	Result	Execution time	Memory
442913		Soumya1	Zapina (COCI20_zapina)	C++14	110 / 110	141 ms	1728 KiB

zapina

#include <bits/stdc++.h>
using namespace std;
int dp[351][351][2];
int c[351][351];
const int mod = 1e9 + 7;
int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int n;
  cin >> n;
  c[0][0] = 1;
  for (int i = 1; i <= n; i++) {
    c[i][0] = c[i][i] = 1;
    for (int j = 1; j < i; j++) {
      c[i][j] = c[i - 1][j - 1] + c[i - 1][j];
      if (c[i][j] >= mod) c[i][j] -= mod;
    }
  }
  dp[0][0][0] = 1;
  for (int i = 1; i <= n; i++) {
    for (int j = 0; j <= n; j++) {
      for (int k = 0; k <= j; k++) {
        if (i == k) {
          dp[i][j][1] += (1LL * c[j][k] * (dp[i - 1][j - k][0] + dp[i - 1][j - k][1])) % mod;
          if (dp[i][j][1] >= mod) dp[i][j][1] -= mod;
          continue;
        }
        dp[i][j][1] += (1LL * c[j][k] * dp[i - 1][j - k][1]) % mod;
        dp[i][j][0] += (1LL * c[j][k] * dp[i - 1][j - k][0]) % mod;
        if (dp[i][j][1] >= mod) dp[i][j][1] -= mod;
        if (dp[i][j][0] >= mod) dp[i][j][0] -= mod;
      }
    }
  }
  int ans = 0;
  cout << dp[n][n][1] << "\n";
  return 0;
}

Compilation message (stderr)

zapina.cpp: In function 'int main()':
zapina.cpp:35:7: warning: unused variable 'ans' [-Wunused-variable]
   35 |   int ans = 0;
      |       ^~~

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