QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#825467#9778. Brotatothinking (Fedor Romashov, Alexey Mikhnenko, Gimran Abdullin)#AC ✓1397ms8160kbC++233.2kb2024-12-21 19:33:542024-12-21 19:33:56

Judging History

This is the latest submission verdict.

  • [2024-12-21 19:33:56]
  • Judged
  • Verdict: AC
  • Time: 1397ms
  • Memory: 8160kb
  • [2024-12-21 19:33:54]
  • Submitted

answer

#include "bits/stdc++.h"


#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep1(i, n) for (int i = 1; i < (n); ++i)
#define rep1n(i, n) for (int i = 1; i <= (n); ++i)
#define repr(i, n) for (int i = (n) - 1; i >= 0; --i)
#define pb push_back
#define eb emplace_back
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()
#define each(x, a) for (auto &x : a)
#define ar array
#define vec vector
#define range(i, n) rep(i, n)

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using str = string;
using pi = pair<int, int>;
using pl = pair<ll, ll>;

using vi = vector<int>;
using vl = vector<ll>;
using vpi = vector<pair<int, int>>;
using vvi = vector<vi>;

int Bit(int mask, int b) { return (mask >> b) & 1; }

template<class T>
bool ckmin(T &a, const T &b) {
    if (b < a) {
        a = b;
        return true;
    }
    return false;
}

template<class T>
bool ckmax(T &a, const T &b) {
    if (b > a) {
        a = b;
        return true;
    }
    return false;
}

// [l, r)
template<typename T, typename F>
T FindFirstTrue(T l, T r, const F &predicat) {
    --l;
    while (r - l > 1) {
        T mid = l + (r - l) / 2;
        if (predicat(mid)) {
            r = mid;
        } else {
            l = mid;
        }
    }
    return r;
}


template<typename T, typename F>
T FindLastFalse(T l, T r, const F &predicat) {
    return FindFirstTrue(l, r, predicat) - 1;
}

const int INFi = 2e9;
const ll INF = 2e18;

double GetTime() {
    return (double)clock() / CLOCKS_PER_SEC;
}

void solve() {
    double st = GetTime();
    int n, k; cin >> n >> k;
    vector<ld> dp(n + 1);
    ld p; cin >> p;
    rep(i, n) dp[i] = INF;
    vector<ld> pwp(n + 1);
    pwp[0] = 1;
    for(int i = 1; i <= n; ++i) {
        pwp[i] = (pwp[i - 1]) * (1 - p);
    }
    const ld EPS = 1e-9;
    ld mn = (ld)n + (ld)n * p;
    for(int t = 0; t <= k; ++t) {
        if ((GetTime() - st) > 1.4) break;
        vector<ld> ndp(n + 1);
        ndp[n] = 0;
        for(int i = n; i >= 0; --i) {
            if (i < n) {
                ndp[i] = 1 + (1 - p) * ndp[i + 1] + p * dp[i];
            }
            // ndp[0] + 1 = 1 + ndp[i] * (1 - p)^{i} + p * (ndp[0] + 1) * (1 + (1 - p) + .. (1 - p)^{i-1})
            // ndp[0] + 1 = 1 + ndp[i] * (1 - p)^{i} + (ndp[0] + 1) * (1 - (1-p)^{i})
            // (ndp[0] + 1) * (1-p)^{i};
            // (ndp[0] + 1) * c = ndp[i] * pwp[i] + 1
            if (!i) break;
            ld e0 = (ndp[i] + (1 - pwp[i]) / (p * pwp[i]));
            if (e0 <= dp[i - 1] || (t == 0 && i == n)) {
                for(int j = i - 1; j >= 0; --j) {
                    ndp[j] = 1 + (1 - p) * ndp[j + 1] + p * e0;
                }
                break;
            }
        }
        swap(dp, ndp);
        if (abs(dp[0] - mn) <= EPS) {
            cout << mn << '\n';
            return;
        }
//        cerr << dp[0] << endl;
    }
    cout << dp[0] << '\n';
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout << setprecision(12) << fixed;
    int t = 1;
//    cin >> t;
    rep(i, t) {
        solve();
    }
    return 0;
}

/*
 * 100000 10000000
0.0002
 */

这程序好像有点Bug,我给组数据试试?

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3888kb

input:

5 0
0.5

output:

62.000000000000

result:

ok found '62.000000000', expected '62.000000000', error '0.000000000'

Test #2:

score: 0
Accepted
time: 0ms
memory: 3804kb

input:

5 1
0.5

output:

47.000000000000

result:

ok found '47.000000000', expected '47.000000000', error '0.000000000'

Test #3:

score: 0
Accepted
time: 1ms
memory: 3936kb

input:

10000 0
0.002

output:

247489700298.253686755896

result:

ok found '247489700298.253692627', expected '247489700298.253692627', error '0.000000000'

Test #4:

score: 0
Accepted
time: 3ms
memory: 8048kb

input:

100000 10
0.0002

output:

38767507133.232227981091

result:

ok found '38767507133.232231140', expected '38767507133.232215881', error '0.000000000'

Test #5:

score: 0
Accepted
time: 3ms
memory: 8088kb

input:

100000 0
0.0002

output:

2430683127170.376990795135

result:

ok found '2430683127170.376953125', expected '2430683127170.376953125', error '0.000000000'

Test #6:

score: 0
Accepted
time: 14ms
memory: 7980kb

input:

100000 20
0.0002

output:

801073272.231681398058

result:

ok found '801073272.231681347', expected '801073272.231680870', error '0.000000000'

Test #7:

score: 0
Accepted
time: 16ms
memory: 8048kb

input:

100000 40
0.0002

output:

478148.718426307409

result:

ok found '478148.718426307', expected '478148.718426307', error '0.000000000'

Test #8:

score: 0
Accepted
time: 0ms
memory: 8096kb

input:

99995 4
0.0001

output:

38976542.868175859934

result:

ok found '38976542.868175857', expected '38976542.868175834', error '0.000000000'

Test #9:

score: 0
Accepted
time: 6ms
memory: 8084kb

input:

99995 10
0.0001

output:

3549184.597712019549

result:

ok found '3549184.597712019', expected '3549184.597712017', error '0.000000000'

Test #10:

score: 0
Accepted
time: 12ms
memory: 8096kb

input:

99995 16
0.0001

output:

399507.470556742168

result:

ok found '399507.470556742', expected '399507.470556742', error '0.000000000'

Test #11:

score: 0
Accepted
time: 4ms
memory: 7984kb

input:

99990 8
0.0001

output:

7773463.947930723188

result:

ok found '7773463.947930723', expected '7773463.947930722', error '0.000000000'

Test #12:

score: 0
Accepted
time: 6ms
memory: 8096kb

input:

99990 10
0.0001

output:

3547428.945472297845

result:

ok found '3547428.945472298', expected '3547428.945472297', error '0.000000000'

Test #13:

score: 0
Accepted
time: 10ms
memory: 8160kb

input:

99990 12
0.0001

output:

1647102.020434396225

result:

ok found '1647102.020434396', expected '1647102.020434395', error '0.000000000'

Test #14:

score: 0
Accepted
time: 0ms
memory: 4140kb

input:

16664 1
0.0012

output:

257920044630.860549092293

result:

ok found '257920044630.860534668', expected '257920044630.860534668', error '0.000000000'

Test #15:

score: 0
Accepted
time: 2ms
memory: 4408kb

input:

16664 21
0.0012

output:

92190688.544415018492

result:

ok found '92190688.544415012', expected '92190688.544415027', error '0.000000000'

Test #16:

score: 0
Accepted
time: 4ms
memory: 4324kb

input:

16664 41
0.0012

output:

59865.091785919911

result:

ok found '59865.091785920', expected '59865.091785920', error '0.000000000'

Test #17:

score: 0
Accepted
time: 1ms
memory: 4320kb

input:

16659 5
0.0003

output:

63366.264406954761

result:

ok found '63366.264406955', expected '63366.264406955', error '0.000000000'

Test #18:

score: 0
Accepted
time: 2ms
memory: 4248kb

input:

16659 11
0.0003

output:

18120.911863542531

result:

ok found '18120.911863543', expected '18120.911863543', error '0.000000000'

Test #19:

score: 0
Accepted
time: 0ms
memory: 4404kb

input:

16659 17
0.0003

output:

16666.555451967406

result:

ok found '16666.555451967', expected '16666.555451967', error '0.000000000'

Test #20:

score: 0
Accepted
time: 0ms
memory: 4244kb

input:

16654 9
0.0001

output:

16656.079416575626

result:

ok found '16656.079416576', expected '16656.079416576', error '0.000000000'

Test #21:

score: 0
Accepted
time: 2ms
memory: 4408kb

input:

16654 11
0.0001

output:

16655.674122172401

result:

ok found '16655.674122172', expected '16655.674122172', error '0.000000000'

Test #22:

score: 0
Accepted
time: 0ms
memory: 4348kb

input:

16654 13
0.0001

output:

16655.665695362667

result:

ok found '16655.665695363', expected '16655.665695363', error '0.000000000'

Test #23:

score: 0
Accepted
time: 1ms
memory: 4140kb

input:

2774 2
0.0072

output:

28951389306.987514117733

result:

ok found '28951389306.987514496', expected '28951389306.987514496', error '0.000000000'

Test #24:

score: 0
Accepted
time: 1ms
memory: 4196kb

input:

2774 22
0.0072

output:

11312241.450653942030

result:

ok found '11312241.450653942', expected '11312241.450653942', error '0.000000000'

Test #25:

score: 0
Accepted
time: 1ms
memory: 4064kb

input:

2774 42
0.0072

output:

8222.415108508618

result:

ok found '8222.415108509', expected '8222.415108509', error '0.000000000'

Test #26:

score: 0
Accepted
time: 0ms
memory: 4072kb

input:

2769 6
0.0021

output:

13600.582355731611

result:

ok found '13600.582355732', expected '13600.582355732', error '0.000000000'

Test #27:

score: 0
Accepted
time: 1ms
memory: 4200kb

input:

2769 12
0.0021

output:

3239.549978211126

result:

ok found '3239.549978211', expected '3239.549978211', error '0.000000000'

Test #28:

score: 0
Accepted
time: 1ms
memory: 4108kb

input:

2769 18
0.0021

output:

2776.628194875240

result:

ok found '2776.628194875', expected '2776.628194875', error '0.000000000'

Test #29:

score: 0
Accepted
time: 1ms
memory: 4104kb

input:

2764 10
0.0007

output:

2765.987073955086

result:

ok found '2765.987073955', expected '2765.987073955', error '0.000000000'

Test #30:

score: 0
Accepted
time: 1ms
memory: 4068kb

input:

2764 12
0.0007

output:

2765.937362849972

result:

ok found '2765.937362850', expected '2765.937362850', error '0.000000000'

Test #31:

score: 0
Accepted
time: 1ms
memory: 4100kb

input:

2764 14
0.0007

output:

2765.936176703473

result:

ok found '2765.936176703', expected '2765.936176704', error '0.000000000'

Test #32:

score: 0
Accepted
time: 1397ms
memory: 8016kb

input:

100000 1000000000
0.0001

output:

100010.001000100054

result:

ok found '100010.001000100', expected '100010.001000100', error '0.000000000'

Test #33:

score: 0
Accepted
time: 639ms
memory: 3940kb

input:

1 1000000000
0.5

output:

2.000000000000

result:

ok found '2.000000000', expected '2.000000000', error '0.000000000'

Test #34:

score: 0
Accepted
time: 1ms
memory: 3988kb

input:

50 50
0.25

output:

66.734187333705

result:

ok found '66.734187334', expected '66.734187334', error '0.000000000'

Test #35:

score: 0
Accepted
time: 1ms
memory: 3932kb

input:

40 100
0.5

output:

788.633123966272

result:

ok found '788.633123966', expected '788.633123966', error '0.000000000'

Test #36:

score: 0
Accepted
time: 1ms
memory: 3820kb

input:

40 160
0.5

output:

80.000000255248

result:

ok found '80.000000255', expected '80.000000255', error '0.000000000'

Test #37:

score: 0
Accepted
time: 1ms
memory: 4088kb

input:

40 162
0.5

output:

80.000000098234

result:

ok found '80.000000098', expected '80.000000098', error '0.000000000'

Test #38:

score: 0
Accepted
time: 1121ms
memory: 3928kb

input:

40 1000000000
0.5

output:

80.000000000000

result:

ok found '80.000000000', expected '80.000000000', error '0.000000000'

Test #39:

score: 0
Accepted
time: 1ms
memory: 3932kb

input:

40 0
0.5000

output:

2199023255550.000000000000

result:

ok found '2199023255550.000000000', expected '2199023255550.000000000', error '0.000000000'

Test #40:

score: 0
Accepted
time: 0ms
memory: 3940kb

input:

40 1
0.5000

output:

1649267441663.000000000000

result:

ok found '1649267441663.000000000', expected '1649267441663.000000000', error '0.000000000'

Test #41:

score: 0
Accepted
time: 0ms
memory: 3988kb

input:

40 2
0.5000

output:

1236950581248.250000000000

result:

ok found '1236950581248.250000000', expected '1236950581248.250000000', error '0.000000000'

Test #42:

score: 0
Accepted
time: 0ms
memory: 3876kb

input:

40 3
0.5000

output:

962072674305.000000000000

result:

ok found '962072674305.000000000', expected '962072674305.000000000', error '0.000000000'

Test #43:

score: 0
Accepted
time: 0ms
memory: 4012kb

input:

40 4
0.5000

output:

738734374914.218750000000

result:

ok found '738734374914.218750000', expected '738734374914.218750000', error '0.000000000'

Test #44:

score: 0
Accepted
time: 1ms
memory: 4084kb

input:

40 5
0.5000

output:

575525617666.953125000000

result:

ok found '575525617666.953125000', expected '575525617666.953125000', error '0.000000000'

Test #45:

score: 0
Accepted
time: 0ms
memory: 4000kb

input:

40 10
0.5000

output:

173409304583.132568359375

result:

ok found '173409304583.132568359', expected '173409304583.132568359', error '0.000000000'

Test #46:

score: 0
Accepted
time: 0ms
memory: 3896kb

input:

40 15
0.5000

output:

55390502923.073508262634

result:

ok found '55390502923.073509216', expected '55390502923.073509216', error '0.000000000'

Test #47:

score: 0
Accepted
time: 0ms
memory: 3800kb

input:

40 20
0.5000

output:

18180087821.747404500842

result:

ok found '18180087821.747406006', expected '18180087821.747406006', error '0.000000000'

Test #48:

score: 0
Accepted
time: 0ms
memory: 3940kb

input:

40 25
0.5000

output:

5963269457.318783601746

result:

ok found '5963269457.318783760', expected '5963269457.318783760', error '0.000000000'

Test #49:

score: 0
Accepted
time: 0ms
memory: 3940kb

input:

40 30
0.5000

output:

2008652291.209798980970

result:

ok found '2008652291.209799051', expected '2008652291.209799051', error '0.000000000'

Test #50:

score: 0
Accepted
time: 0ms
memory: 4068kb

input:

40 35
0.5000

output:

676323760.766804063285

result:

ok found '676323760.766804099', expected '676323760.766804099', error '0.000000000'

Test #51:

score: 0
Accepted
time: 0ms
memory: 3936kb

input:

40 40
0.5000

output:

231036317.117801572982

result:

ok found '231036317.117801577', expected '231036317.117801577', error '0.000000000'

Test #52:

score: 0
Accepted
time: 0ms
memory: 3880kb

input:

40 45
0.5000

output:

79224360.245689896939

result:

ok found '79224360.245689899', expected '79224360.245689899', error '0.000000000'

Test #53:

score: 0
Accepted
time: 1ms
memory: 3932kb

input:

40 50
0.5000

output:

27134304.850833610879

result:

ok found '27134304.850833610', expected '27134304.850833610', error '0.000000000'

Test #54:

score: 0
Accepted
time: 0ms
memory: 3932kb

input:

2 1
0.0001

output:

2.000200030004

result:

ok found '2.000200030', expected '2.000200030', error '0.000000000'

Test #55:

score: 0
Accepted
time: 0ms
memory: 3928kb

input:

2 0
0.0001

output:

2.000300040005

result:

ok found '2.000300040', expected '2.000300040', error '0.000000000'

Test #56:

score: 0
Accepted
time: 0ms
memory: 3920kb

input:

2 2
0.0001

output:

2.000200020003

result:

ok found '2.000200020', expected '2.000200020', error '0.000000000'

Test #57:

score: 0
Accepted
time: 0ms
memory: 3872kb

input:

2 3
0.0001

output:

2.000200020002

result:

ok found '2.000200020', expected '2.000200020', error '0.000000000'

Test #58:

score: 0
Accepted
time: 32ms
memory: 8124kb

input:

100000 60
0.0002

output:

100020.274838305050

result:

ok found '100020.274838305', expected '100020.274838305', error '0.000000000'

Test #59:

score: 0
Accepted
time: 42ms
memory: 8008kb

input:

100000 80
0.0002

output:

100020.004000800251

result:

ok found '100020.004000800', expected '100020.004000800', error '0.000000000'

Test #60:

score: 0
Accepted
time: 54ms
memory: 8100kb

input:

100000 100
0.0002

output:

100020.004000800248

result:

ok found '100020.004000800', expected '100020.004000800', error '0.000000000'

Test #61:

score: 0
Accepted
time: 74ms
memory: 8084kb

input:

100000 120
0.0002

output:

100020.004000800248

result:

ok found '100020.004000800', expected '100020.004000800', error '0.000000000'

Test #62:

score: 0
Accepted
time: 82ms
memory: 8092kb

input:

100000 140
0.0002

output:

100020.004000800248

result:

ok found '100020.004000800', expected '100020.004000800', error '0.000000000'

Test #63:

score: 0
Accepted
time: 108ms
memory: 8012kb

input:

100000 180
0.0002

output:

100020.004000800248

result:

ok found '100020.004000800', expected '100020.004000800', error '0.000000000'

Test #64:

score: 0
Accepted
time: 30ms
memory: 8096kb

input:

100000 55
0.0002

output:

100070.850328203300

result:

ok found '100070.850328203', expected '100070.850328203', error '0.000000000'

Test #65:

score: 0
Accepted
time: 23ms
memory: 8080kb

input:

100000 50
0.0002

output:

103207.684193091292

result:

ok found '103207.684193091', expected '103207.684193091', error '0.000000000'

Extra Test:

score: 0
Extra Test Passed