QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#825033#9778. Brotatoucup-team4474#TL 509ms8320kbC++204.3kb2024-12-21 17:04:032024-12-21 17:04:03

Judging History

This is the latest submission verdict.

  • [2024-12-21 17:04:03]
  • Judged
  • Verdict: TL
  • Time: 509ms
  • Memory: 8320kb
  • [2024-12-21 17:04:03]
  • Submitted

answer

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
using i128 = __int128_t;

bool Memory_begin;

template <int mod>
unsigned int down(unsigned int x)
{
    return x >= mod ? x - mod : x;
}
template <int mod>
struct Modint
{
    unsigned int x;
    Modint() = default;
    Modint(unsigned int x) : x(x) {}
    friend istream &operator>>(istream &in, Modint &a) { return in >> a.x; }
    friend ostream &operator<<(ostream &out, Modint a) { return out << a.x; }
    friend Modint operator+(Modint a, Modint b) { return down<mod>(a.x + b.x); }
    friend Modint operator-(Modint a, Modint b) { return down<mod>(a.x - b.x + mod); }
    friend Modint operator*(Modint a, Modint b) { return 1ULL * a.x * b.x % mod; }
    friend Modint operator/(Modint a, Modint b) { return a * ~b; }
    friend Modint operator^(Modint a, i64 b)
    {
        Modint ans = 1;
        for (; b; b >>= 1, a *= a)
            if (b & 1)
                ans *= a;
        return ans;
    }
    friend Modint operator~(Modint a) { return a ^ (mod - 2); }
    friend Modint operator-(Modint a) { return down<mod>(mod - a.x); }
    friend Modint &operator+=(Modint &a, Modint b) { return a = a + b; }
    friend Modint &operator-=(Modint &a, Modint b) { return a = a - b; }
    friend Modint &operator*=(Modint &a, Modint b) { return a = a * b; }
    friend Modint &operator/=(Modint &a, Modint b) { return a = a / b; }
    friend Modint &operator^=(Modint &a, i64 b) { return a = a ^ b; }
    friend Modint &operator++(Modint &a) { return a += 1; }
    friend Modint operator++(Modint &a, int)
    {
        Modint x = a;
        a += 1;
        return x;
    }
    friend Modint &operator--(Modint &a) { return a -= 1; }
    friend Modint operator--(Modint &a, int)
    {
        Modint x = a;
        a -= 1;
        return x;
    }
    friend bool operator==(Modint a, Modint b) { return a.x == b.x; }
    friend bool operator!=(Modint a, Modint b) { return !(a == b); }
};
const int mod = 998244353;
typedef Modint<mod> mint;

__float128 read()
{
    long double ret;
    cin >> ret;
    return (__float128)ret;
}
void print(__float128 res)
{
    long double ret = res;
    cout << fixed << setprecision(10) << ret << '\n';
}

bool Memory_end;

signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    cerr << (&Memory_end - &Memory_begin) / 1048576.0 << "MB" << '\n';

    int n, k;
    cin >> n >> k;
    const __float128 e = 1.0;
    __float128 p = read();

    clock_t cur = clock();
    vector<__float128> dp(n + 2), a = dp, b = a;
    for (int i = n; i >= 1; i--)
    {
        a[i] = p, b[i] = e;
        a[i] += (e - p) * a[i + 1];
        b[i] += (e - p) * b[i + 1];
    }
    dp[1] = b[1] / (e - a[1]);
    for (int i = n; i >= 2; i--)
        dp[i] = e + p * dp[1] + (e - p) * dp[i + 1];
    int idx = 0;
    for (int _ = 1; _ <= 300 and k; _++, k--)
    {
        for (int i = n; i >= 1; i--)
            dp[i] = e + p * dp[i] + (e - p) * dp[i + 1];
        auto cal = [&](int c) -> __float128
        {
            if (c == n)
                return dp[1];

            __float128 d = (e - powl(e - p, n - c)) / p;
            __float128 b = powl(e - p, n - c) * dp[n - c + 1] + d;
            __float128 a = d * p;
            return b / (e - a);
            // vector<long double> ret(n + 2);
            // for (int i = n; i >= n - c + 1; i--)
            // {
            //     ret[i] = e + p * dp[i] + (e - p) * ret[i + 1];
            //     a[i] = 0, b[i] = ret[i];
            // }
            // for (int i = n - c; i >= 1; i--)
            // {
            //     a[i] = p, b[i] = e;
            //     a[i] += (e - p) * a[i + 1];
            //     b[i] += (e - p) * b[i + 1];
            // }
            // ret[1] = b[1] / (e - a[1]);
            // for (int i = n - c; i >= 2; i--)
            //     ret[i] = e + p * ret[1] + (e - p) * ret[i + 1];
            // return ret;
        };
        while (idx < n)
        {
            auto dp1 = cal(idx), dp2 = cal(idx + 1);
            if (dp1 > dp2)
                idx++;
            else
                break;
        }
        dp[1] = cal(idx);
        for (int i = n - idx; i >= 2; i--)
            dp[i] = e + p * dp[1] + (e - p) * dp[i + 1];
    }
    print(dp[1]);
}
/*
 */

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3924kb

input:

5 0
0.5

output:

62.0000000000

result:

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

Test #2:

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

input:

5 1
0.5

output:

47.0000000000

result:

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

Test #3:

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

input:

10000 0
0.002

output:

247489700298.2537131310

result:

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

Test #4:

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

input:

100000 10
0.0002

output:

38767507133.2322458252

result:

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

Test #5:

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

input:

100000 0
0.0002

output:

2430683127170.3809044361

result:

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

Test #6:

score: 0
Accepted
time: 290ms
memory: 8312kb

input:

100000 20
0.0002

output:

801073272.2316816133

result:

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

Test #7:

score: 0
Accepted
time: 509ms
memory: 8184kb

input:

100000 40
0.0002

output:

478148.7184263072

result:

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

Test #8:

score: 0
Accepted
time: 85ms
memory: 8256kb

input:

99995 4
0.0001

output:

38976542.8681758503

result:

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

Test #9:

score: 0
Accepted
time: 184ms
memory: 8208kb

input:

99995 10
0.0001

output:

3549184.5977120187

result:

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

Test #10:

score: 0
Accepted
time: 273ms
memory: 8252kb

input:

99995 16
0.0001

output:

399507.4705567421

result:

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

Test #11:

score: 0
Accepted
time: 157ms
memory: 8304kb

input:

99990 8
0.0001

output:

7773463.9479307245

result:

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

Test #12:

score: 0
Accepted
time: 183ms
memory: 8320kb

input:

99990 10
0.0001

output:

3547428.9454722986

result:

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

Test #13:

score: 0
Accepted
time: 217ms
memory: 8312kb

input:

99990 12
0.0001

output:

1647102.0204343962

result:

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

Test #14:

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

input:

16664 1
0.0012

output:

257920044630.8605401069

result:

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

Test #15:

score: 0
Accepted
time: 51ms
memory: 4360kb

input:

16664 21
0.0012

output:

92190688.5444150171

result:

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

Test #16:

score: 0
Accepted
time: 87ms
memory: 4368kb

input:

16664 41
0.0012

output:

59865.0917859199

result:

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

Test #17:

score: 0
Accepted
time: 24ms
memory: 4340kb

input:

16659 5
0.0003

output:

63366.2644069548

result:

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

Test #18:

score: 0
Accepted
time: 41ms
memory: 4284kb

input:

16659 11
0.0003

output:

18120.9118635425

result:

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

Test #19:

score: 0
Accepted
time: 49ms
memory: 4328kb

input:

16659 17
0.0003

output:

16666.5554519674

result:

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

Test #20:

score: 0
Accepted
time: 34ms
memory: 4336kb

input:

16654 9
0.0001

output:

16656.0794165756

result:

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

Test #21:

score: 0
Accepted
time: 40ms
memory: 4368kb

input:

16654 11
0.0001

output:

16655.6741221724

result:

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

Test #22:

score: 0
Accepted
time: 36ms
memory: 4332kb

input:

16654 13
0.0001

output:

16655.6656953627

result:

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

Test #23:

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

input:

2774 2
0.0072

output:

28951389306.9875136279

result:

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

Test #24:

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

input:

2774 22
0.0072

output:

11312241.4506539419

result:

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

Test #25:

score: 0
Accepted
time: 15ms
memory: 4192kb

input:

2774 42
0.0072

output:

8222.4151085086

result:

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

Test #26:

score: 0
Accepted
time: 5ms
memory: 4132kb

input:

2769 6
0.0021

output:

13600.5823557316

result:

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

Test #27:

score: 0
Accepted
time: 7ms
memory: 4180kb

input:

2769 12
0.0021

output:

3239.5499782111

result:

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

Test #28:

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

input:

2769 18
0.0021

output:

2776.6281948752

result:

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

Test #29:

score: 0
Accepted
time: 7ms
memory: 4096kb

input:

2764 10
0.0007

output:

2765.9870739551

result:

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

Test #30:

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

input:

2764 12
0.0007

output:

2765.9373628500

result:

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

Test #31:

score: 0
Accepted
time: 8ms
memory: 4116kb

input:

2764 14
0.0007

output:

2765.9361767035

result:

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

Test #32:

score: -100
Time Limit Exceeded

input:

100000 1000000000
0.0001

output:


result: