結果

問題 No.1043 直列大学
ユーザー rxoxixyxd
提出日時 2020-05-02 12:18:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,975 bytes
コンパイル時間 2,016 ms
コンパイル使用メモリ 175,304 KB
実行使用メモリ 813,712 KB
最終ジャッジ日時 2024-12-29 15:18:34
合計ジャッジ時間 8,352 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 9 WA * 11 RE * 7 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for (int i = 0; i < n; ++i)
#define ALL(c) (c).begin(), (c).end()
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v, x) (std::find(v.begin(), v.end(), x) != v.end())
using namespace std;

using ll = long long;

template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }
const int INF = 1e9;
const long long INFL = 1LL<<60;

const int mod = 1000000007;
struct mint {
  ll x; // typedef long long ll;
  mint(ll x=0):x((x%mod+mod)%mod){}
  mint& operator+=(const mint a) {
    if ((x += a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator-=(const mint a) {
    if ((x += mod-a.x) >= mod) x -= mod;
    return *this;
  }
  mint& operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const {
    mint res(*this);
    return res+=a;
  }
  mint operator-(const mint a) const {
    mint res(*this);
    return res-=a;
  }
  mint operator*(const mint a) const {
    mint res(*this);
    return res*=a;
  }
  mint pow(ll t) const {
    if (!t) return 1;
    mint a = pow(t>>1);
    a *= a;
    if (t&1) a *= *this;
    return a;
  }

  // for prime mod
  mint inv() const {
    return pow(mod-2);
  }
  mint& operator/=(const mint a) {
    return (*this) *= a.inv();
  }
  mint operator/(const mint a) const {
    mint res(*this);
    return res/=a;
  }
};

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);

  int n, m;
  cin >> n >> m;
  vector<int> v(n);
  vector<int> r(m);
  rep(i, n) cin >> v[i];
  rep(i, m) cin >> r[i];
  int a, b;
  cin >> a >> b;

  sort(ALL(v));
  sort(ALL(r));

  int v_sum_max = n * v[n-1] + 1;
  int r_sum_max = m * r[n-1] + 1;

  // i 番目までの要素を使用して合計が j になる組み合わせ
  vector<vector<int>> dpv(n+1, vector<int>(v_sum_max + v[n-1], 0));
  vector<vector<int>> dpr(m+1, vector<int>(r_sum_max + r[n-1], 0));

  dpv[0][0] = 1;
  for (int i = 1; i <= n; i++) {
    for (int j = 0; j < v_sum_max; j++) {
        dpv[i][j + v[i-1]] += dpv[i-1][j];
        dpv[i][j] += dpv[i-1][j];
    }
  }

  dpr[0][0] = 1;
  for (int i = 1; i <= m; i++) {
    for (int j = 0; j < r_sum_max; j++) {
      dpr[i][j + r[i-1]] += dpr[i-1][j];
      dpr[i][j] += dpr[i-1][j];
    }
  }

  vector<int> vs(v_sum_max, 0);  // 合計電圧が i 以下となる組み合わせの数
  for (int i = 1; i < v_sum_max; i++) {
    vs[i] += vs[i-1];
    vs[i] += dpv[n][i];
  }

  mint ans(0);
  for (int i = 1; i < r_sum_max; i++) {
    ll n_res = dpr[m][i];
    // n_vol = a * i <= v <= b * i を満たす v の個数
    if (a*i-1 >= v_sum_max) continue;
    ll n_vol = vs[min(b*i, v_sum_max-1)] - vs[a*i-1];
    ans += n_res * n_vol;
  }

  cout << ans.x << endl;

  return 0;
}
0