結果

問題 No.1043 直列大学
コンテスト
ユーザー siman
提出日時 2022-11-01 03:14:12
言語 C++17(clang)
(clang++ 21.1.8 + boost 1.89.0)
コンパイル:
clang++ -O2 -lm -std=c++1z -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 1,494 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,564 ms
コンパイル使用メモリ 147,328 KB
実行使用メモリ 6,272 KB
最終ジャッジ日時 2026-03-30 16:06:47
合計ジャッジ時間 7,307 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:22:9: warning: variable length arrays in C++ are a Clang extension [-Wvla-cxx-extension]
   22 |   int V[N];
      |         ^
main.cpp:22:9: note: read of non-const variable 'N' is not allowed in a constant expression
main.cpp:19:7: note: declared here
   19 |   int N, M;
      |       ^
main.cpp:27:9: warning: variable length arrays in C++ are a Clang extension [-Wvla-cxx-extension]
   27 |   int R[M];
      |         ^
main.cpp:27:9: note: read of non-const variable 'M' is not allowed in a constant expression
main.cpp:19:10: note: declared here
   19 |   int N, M;
      |          ^
2 warnings generated.

ソースコード

diff #
raw source code 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

const ll MOD = 1000000007;

int main() {
  int N, M;
  cin >> N >> M;

  int V[N];
  for (int i = 0; i < N; ++i) {
    cin >> V[i];
  }

  int R[M];
  for (int i = 0; i < M; ++i) {
    cin >> R[i];
  }

  int A, B;
  cin >> A >> B;

  ll dp1[100010];
  memset(dp1, 0, sizeof(dp1));
  dp1[0] = 1;
  for (int i = 0; i < N; ++i) {
    int v = V[i];

    for (int i = 100000 - v; i >= 0; --i) {
      dp1[i + v] += dp1[i];
      dp1[i + v] %= MOD;
    }
  }

  ll dp2[100010];
  memset(dp2, 0, sizeof(dp2));
  dp2[0] = 1;
  for (int i = 0; i < M; ++i) {
    int r = R[i];

    for (int i = 100000 - r; i >= 0; --i) {
      dp2[i + r] += dp2[i];
      dp2[i + r] %= MOD;
    }
  }

  for (int r = 1; r <= 100000; ++r) {
    dp2[r] += dp2[r - 1];
    dp2[r] %= MOD;
  }

  ll ans = 0;

  for (int v = 1; v <= 100000; ++v) {
    if (dp1[v] == 0) continue;

    // v / B <= r && r <= v / A;
    int left = max(1, min(100000, (v + B - 1) / B));
    int right = min(100000, v / A);

    ll v_cnt = dp1[v];
    ll r_cnt = (dp2[right] - dp2[left - 1] + MOD) % MOD;
    // fprintf(stderr, "v: %d, left: %d, right: %d, v_cnt: %lld, r_cnt: %lld\n", v, left, right, v_cnt, r_cnt);
    ans += v_cnt * r_cnt;
    ans %= MOD;
  }

  cout << ans << endl;

  return 0;
}
0