結果

問題 No.1043 直列大学
ユーザー tactac
提出日時 2020-05-02 17:19:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 84 ms / 2,000 ms
コード長 4,890 bytes
コンパイル時間 2,313 ms
コンパイル使用メモリ 203,380 KB
実行使用メモリ 161,460 KB
最終ジャッジ日時 2023-08-24 12:42:23
合計ジャッジ時間 5,622 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
161,184 KB
testcase_01 AC 45 ms
161,296 KB
testcase_02 AC 46 ms
161,132 KB
testcase_03 AC 44 ms
161,244 KB
testcase_04 AC 45 ms
161,144 KB
testcase_05 AC 45 ms
161,240 KB
testcase_06 AC 45 ms
161,136 KB
testcase_07 AC 46 ms
161,276 KB
testcase_08 AC 45 ms
161,140 KB
testcase_09 AC 65 ms
161,196 KB
testcase_10 AC 69 ms
161,196 KB
testcase_11 AC 80 ms
161,356 KB
testcase_12 AC 84 ms
161,192 KB
testcase_13 AC 77 ms
161,260 KB
testcase_14 AC 78 ms
161,328 KB
testcase_15 AC 82 ms
161,196 KB
testcase_16 AC 77 ms
161,196 KB
testcase_17 AC 68 ms
161,152 KB
testcase_18 AC 68 ms
161,424 KB
testcase_19 AC 76 ms
161,288 KB
testcase_20 AC 68 ms
161,156 KB
testcase_21 AC 74 ms
161,208 KB
testcase_22 AC 75 ms
161,200 KB
testcase_23 AC 83 ms
161,264 KB
testcase_24 AC 71 ms
161,256 KB
testcase_25 AC 76 ms
161,388 KB
testcase_26 AC 79 ms
161,200 KB
testcase_27 AC 61 ms
161,200 KB
testcase_28 AC 77 ms
161,460 KB
testcase_29 AC 53 ms
161,244 KB
testcase_30 AC 68 ms
161,212 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define F first
#define S second
#define pii pair<int, int>
#define eb emplace_back
#define all(v) v.begin(), v.end()
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep3(i, l, n) for (int i = l; i < (n); ++i)
#define sz(v) (int)v.size()
#define endl '\n'
const int inf = 1000000007;
const ll INF = 1e18;
// int mod = 998244353;
int mod = 1000000007;
#define abs(x) (x >= 0 ? x : -(x))
#define lb(v, x) (int)(lower_bound(all(v), x) - v.begin())
#define ub(v, x) (int)(upper_bound(all(v), x) - v.begin())
template<typename T1, typename T2> inline bool chmin(T1 &a, T2 b) { if (a > b) { a = b; return 1; } return 0; }
template<typename T1, typename T2> inline bool chmax(T1 &a, T2 b) { if (a < b) { a = b; return 1; } return 0; }
ll gcd(ll a, ll b) { if (b == 0) return a; return gcd(b, a % b); }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
ll pow(ll a, int b) { return b ? pow(a * a, b / 2) * (b % 2 ? a : 1) : 1; }
ll modpow(ll a, ll b, ll _mod) { return b ? modpow(a * a % _mod, b / 2, _mod) * (b % 2 ? a : 1) % _mod : 1; }
template<class T, class U> ostream& operator << (ostream& os, const pair<T, U>& p) { os << p.F << " " << p.S; return os; }
template<class T> ostream& operator << (ostream& os, const vector<T>& vec) { rep(i, sz(vec)) { if (i) os << " "; os << vec[i]; } return os; }
template<typename T> inline istream& operator >> (istream& is, vector<T>& v) { rep(j, sz(v)) is >> v[j]; return is; }
template<class T, class T2> inline void add(T &a, T2 b) { a += b; if (a >= mod) a -= mod; }


void solve();

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cout << fixed << setprecision(10);
    int T;
    T = 1;
    while (T--) solve();
}

struct mint {
    ll x;
    mint (ll x = 0) : x(x >= 0 ? x % mod : (mod + x % mod) % mod) {}
    mint operator -() const { return mint(-x); }
    mint& operator += (const mint a) { // コンストラクタの、modで割った余りがくる
        if ((x += a.x) >= mod) x -= mod;
        return *this;
    }
    mint& operator -= (const mint a) {
        if ((x += -a.x + mod) >= 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 { // mint(a).pow(t) tは余りとったりしたらあかん
        if (t == 0) return 1;
        mint a = pow(t >> 1); // 再帰
        a *= a;
        if (t & 1) a *= *this;
        return a;
    }
    
    // 逆元
    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;
    }
    bool operator == (const mint a) { return &a == this; }
    friend ostream &operator << (ostream &s, mint a) { return s << a.x; }
};

// https://yukicoder.me/submissions/474362
const ll N = 100005;
mint dp[101][N], dp2[101][N];
void solve() {
    ll n, m;
    cin >> n >> m;
    vector<ll> v(n);
    cin >> v;
    vector<ll> r(m);
    cin >> r;
    ll a, b;
    cin >> a >> b;
    
    dp[0][0] = 1;
    rep(i, n) rep(j, N) {
        dp[i + 1][j] += dp[i][j];
        if (j + v[i] < N) dp[i + 1][j + v[i]] += dp[i][j]; // debuged, if
    }
    dp[n][0] = 0;
    rep3(j, 1, N) dp[n][j] += dp[n][j - 1];
    // rep(j, N) cout << dp[n][j] << " "; cout << endl;
    
    dp2[0][0] = 1;
    rep(i, m) rep(j, N) {
        dp2[i + 1][j] += dp2[i][j];
        if (j + r[i] < N) dp2[i + 1][j + r[i]] += dp2[i][j];
    }
    dp2[m][0] = 0;
    rep3(j, 1, N) dp2[m][j] += dp2[m][j - 1]; // debuged, dp to dp2
    // rep(j, N) cout << dp2[n][j] << " "; cout << endl;
    
    mint ans = 0;
    /*
     rep3(V, 1, N) {
     int R2 = V / a - 2;
     int tmp = R2;
     rep3(i, -3, 4) {
     if (tmp + i <= 0) continue;
     if (V / (tmp + i) < a) chmin(R2, tmp + i);
     else chmax(R2, tmp + i);
     }
     int R = V / b + 2;
     tmp = R;
     rep3(i, -3, 4) {
     if (tmp + i <= 0) continue;
     if (V / (tmp + i) > b) chmax(R, tmp + i);
     else chmin(R, tmp + i);
     }
     cout << V << " " << R2 << " " << R << endl;
     R--;
     ans += (dp[n][V] - dp[n][V - 1]) * (dp2[m][R2] - dp2[m][R]);
     }
     */
    rep3(R, 1, N) {
        ll l = R * a;
        if (l >= N) break;
        ll r = min(R * b, N - 1); // debged, overflow
        // cerr << R << " " << (dp2[m][R] - dp2[m][R - 1]) << " " << (dp[n][r] -  dp[n][l - 1]) << endl;
        ans += (dp2[m][R] - dp2[m][R - 1]) * (dp[n][r] -  dp[n][l - 1]);
    }
    cout << ans << endl;
}

0