Submission #3106201


Source Code Expand

#include <bits/stdc++.h>
#define N 1000000007
using namespace std;

int n, m, q, k, x;
int a[2005] = {0};
// dp[i][j] = combi i:type j:num
int dp[2005][2005] = {0};
// ans[i][j] = combi i:not use j:num
int ans[2005][2005] = {0};

void solve();

int main() {
  int i;
  cin >> n >> m >> q;
  for(i = 0; i < n; ++i) cin >> a[i];
  solve();
  for(i = 0; i < q; ++i) {
    cin >> k >> x;
    cout << ans[k][m - x] << endl;
  }
  return 0;
}

void solve() {
  int i, j;
  for(i = 0; i <= n; ++i) dp[i][0] = 1;
  for(i = 1; i <= n; ++i)
    for(j = 1; j <= m; ++j) {
      dp[i][j] = dp[i][j - 1] + dp[i - 1][j];
      dp[i][j] %= N;
      if(j - a[i - 1] - 1 >= 0)
        dp[i][j] -= dp[i - 1][j - a[i - 1] - 1];
      while(dp[i][j] < 0) dp[i][j] += N;
      dp[i][j] %= N;
    }
  for(i = 1; i <= n; ++i) {
    for(j = 0; j <= m; ++j) {
      ans[i][j] = dp[n][j] % N;
      if(j != 0) ans[i][j] -= dp[n][j - 1];
      ans[i][j] %= N;
      if(j - a[i - 1] - 1 >= 0)
        ans[i][j] += ans[i][j - a[i - 1] - 1];
      while(dp[i][j] < 0) dp[i][j] += N;
      ans[i][j] %= N;
    }
  }
}

Submission Info

Submission Time
Task A - 小石を取るゲーム
User m_tsubasa
Language C++14 (GCC 5.4.1)
Score 0
Code Size 1134 Byte
Status WA
Exec Time 23 ms
Memory 18688 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 100
Status
WA × 2
WA × 15
Set Name Test Cases
Sample sample_01.txt, sample_02.txt
All sample_01.txt, sample_02.txt, subtask1_01.txt, subtask1_02.txt, subtask1_03.txt, subtask1_04.txt, subtask1_05.txt, subtask1_06.txt, subtask1_07.txt, subtask1_08.txt, subtask1_09.txt, subtask1_10.txt, subtask1_11.txt, subtask1_12.txt, subtask1_13.txt
Case Name Status Exec Time Memory
sample_01.txt WA 2 ms 2304 KB
sample_02.txt WA 1 ms 2304 KB
subtask1_01.txt WA 1 ms 2304 KB
subtask1_02.txt WA 23 ms 18688 KB
subtask1_03.txt WA 5 ms 17664 KB
subtask1_04.txt WA 6 ms 17664 KB
subtask1_05.txt WA 21 ms 18688 KB
subtask1_06.txt WA 21 ms 18688 KB
subtask1_07.txt WA 10 ms 10496 KB
subtask1_08.txt WA 9 ms 17920 KB
subtask1_09.txt WA 4 ms 5248 KB
subtask1_10.txt WA 8 ms 9984 KB
subtask1_11.txt WA 12 ms 12416 KB
subtask1_12.txt WA 5 ms 7808 KB
subtask1_13.txt WA 9 ms 10368 KB