QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#825007#9778. Brotatoucup-team4474#WA 16ms8232kbC++204.6kb2024-12-21 16:56:062024-12-21 16:56:06

Judging History

This is the latest submission verdict.

  • [2024-12-21 16:56:06]
  • Judged
  • Verdict: WA
  • Time: 16ms
  • Memory: 8232kb
  • [2024-12-21 16:56:06]
  • 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 mypow(__float128 x, int y)
{
    __float128 ret = 1;
    while (y)
    {
        if (y & 1)
            ret *= x;
        x *= x;
        y >>= 1;
    }
    return ret;
}
__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;
    while (k-- and clock() - cur <= 1488)
    {
        auto tmp = dp;
        for (int i = n; i >= 1; i--)
            tmp[i] = e + p * dp[i] + (e - p) * tmp[i + 1];
        auto cal = [&](int c) -> __float128
        {
            if (c == n)
                return tmp[1];

            __float128 d = (e - mypow(e - p, n - c)) / p;
            __float128 b = mypow(e - p, n - c) * tmp[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;
        }
        tmp[1] = cal(idx);
        for (int i = n - idx; i >= 2; i--)
            tmp[i] = e + p * tmp[1] + (e - p) * tmp[i + 1];
        dp = move(tmp);
    }
    print(dp[1]);
}
/*
 */

詳細信息

Test #1:

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

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: 1ms
memory: 4092kb

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: 4132kb

input:

10000 0
0.002

output:

247489700298.2537131310

result:

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

Test #4:

score: -100
Wrong Answer
time: 16ms
memory: 8232kb

input:

100000 10
0.0002

output:

2430683127170.3809044361

result:

wrong answer 1st numbers differ - expected: '38767507133.2322159', found: '2430683127170.3808594', error = '61.6989793'