제출 #711539

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
711539		Pety	Calvinball championship (CEOI15_teams)	C++14	100 / 100	563 ms	548 KiB

teams

#include <bits/stdc++.h>
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,abm,avx,mmx,popcnt,tune=native")

using namespace std;

const int mod = 1e6 + 7;

int n,  dp[2][10002][2], p[10002];

void add (int &a, int b) {
  a += b;
  if (a >= mod)
    a-= mod;
}

int main () 
{
  ios_base::sync_with_stdio(false);
  cin.tie(0); cout.tie(0);
  cin >> n;
  for (int i = 1; i <= n; i++)
    cin >> p[i];
  dp[0][0][1] = 1;
  for (int i = 0; i < n; i++) {
    int k = (i & 1);
    for (int j = 0; j <= n; j++)
      dp[k ^ 1][j][0] = dp[k ^ 1][j][1] = 0;
    for (int j = 0; j <= i; j++) {
      ///1 lmao
      if (j > 0)
        add(dp[k ^ 1][j][1], dp[k][j][1] * (p[i + 1] <= j));
      if (j < n)
        add(dp[k ^ 1][j + 1][1], dp[k][j][1] * (j + 1 == p[i + 1]));
      ///0 lmao
      if (j > 0) {
        add(dp[k ^ 1][j][0], 1ll * dp[k][j][1] * min(j, p[i + 1] - 1) % mod);
        add(dp[k ^ 1][j][0], 1ll * dp[k][j][0] * j % mod);
      }
      if (j < n) {
        add(dp[k ^ 1][j + 1][0], dp[k][j][1] * (j + 1 < p[i + 1]));
        add(dp[k ^ 1][j + 1][0], dp[k][j][0]);
      }
    }
  }
  int ans = 0;
  for (int i = 1; i <= n; i++) {
    add(ans, dp[n & 1][i][0]);
    add(ans, dp[n & 1][i][1]);
  }
  cout << ans << "\n";
  return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10