Submission #1034971

# Submission time Handle Problem Language Result Execution time Memory
1034971 2024-07-26T01:10:31 Z TAMREF COVID tests (CEOI24_covid) C++17
66.3 / 100
1442 ms 596 KB
// C
#ifndef _GLIBCXX_NO_ASSERT
#include <cassert>
#endif
#include <cctype>
#include <cerrno>
#include <cfloat>
#include <ciso646>
#include <climits>
#include <clocale>
#include <cmath>
#include <csetjmp>
#include <csignal>
#include <cstdarg>
#include <cstddef>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>

#if __cplusplus >= 201103L
#include <ccomplex>
#include <cfenv>
#include <cinttypes>
#include <cstdbool>
#include <cstdint>
#include <ctgmath>
#include <cwchar>
#include <cwctype>
#endif

// C++
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <utility>
#include <valarray>
#include <vector>

#if __cplusplus >= 201103L
#include <array>
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <forward_list>
#include <future>
#include <initializer_list>
#include <mutex>
#include <random>
#include <ratio>
#include <regex>
#include <scoped_allocator>
#include <system_error>
#include <thread>
#include <tuple>
#include <typeindex>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#endif
#define va first
#define vb second
#define lb lower_bound
#define ub upper_bound
#define bs binary_search
#define pp push_back
#define ep emplace_back
#define all(v) (v).begin(), (v).end()
#define szz(v) ((int)(v).size())
#define bi_pc __builtin_popcount
#define bi_pcll __builtin_popcountll
#define bi_tz __builtin_ctz
#define bi_tzll __builtin_ctzll
#define fio                       \
    ios_base::sync_with_stdio(0); \
    cin.tie(0);
#ifdef TAMREF
#define debug(...) fprintf(stderr, __VA_ARGS__)
#else
#define debug(...) 42
#endif
using namespace std;
template <typename... Args>
void logger(string vars, Args &&...values)
{
    cerr << vars << " = ";
    string delim = "";
    (..., (cerr << delim << values, delim = ", "));
    cerr << '\n';
}
#ifdef TAMREF
#define deb(...) logger(#__VA_ARGS__, __VA_ARGS__)
#else
#define deb(...) 42
#endif
using ll = long long;
using lf = long double;
using pii = pair<int, int>;
using ppi = pair<int, pii>;
using pll = pair<ll, ll>;
using pff = pair<lf, lf>;
using ti = tuple<int, int, int>;
using base = complex<double>;
const lf PI = 3.14159265358979323846264338L;
template <typename T>
inline T umax(T &u, T v) { return u = max(u, v); }
template <typename T>
inline T umin(T &u, T v) { return u = min(u, v); }
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

string ans;

int n, t, qnt, lqnt, mqnt;
lf p;

int query(string F)
{
    ++qnt;
    ++lqnt;

    cout << "Q " << F << '\n';
    cout << flush;

    int q = 0;
#ifdef TAMREF
    for (int i = 0; i < n; i++)
        q |= (ans[i] == '1') && (F[i] == '1');
#else
    string S;
    cin >> S;
    q = (S == "P");
#endif

    return q;
}

int query_seg(int i, int j)
{
    if (i > j)
        return 0;

    string f(n, '0');
    for (int k = i; k <= j; k++)
        f[k] = '1';

    return query(f);
}

int query_seg(pii x)
{
    return query_seg(x.va, x.vb);
}

void answer(string F)
{
    mqnt = max(mqnt, lqnt);
    lqnt = 0;

    cout << "A " << F << '\n';
    cout << flush;

#ifdef TAMREF
    assert(ans == F);
#else
    string C;
    cin >> C;
    assert(C == "C");
#endif
}

vector<pii> initial_segs;
void init()
{
    if (t > 1 && initial_segs.empty())
    {
        uniform_real_distribution st(0.3, 1.0);
        int step = 1;
        for (int i = 0; i < n; i += step)
        {
            step = max(5, int(st(rng) / p));

            initial_segs.emplace_back(i, min(n - 1, i + step - 1));
        }
    }
#ifdef TAMREF
    uniform_real_distribution<lf> d(0, 1);

    ans = string(n, '0');

    for (int i = 0; i < n; i++)
        ans[i] = (d(rng) < p ? '1' : '0');
    debug("ANS: %s\n", ans.c_str());
#endif
}

void run_det()
{
    string ans(n, '0');
    for (int i = 0; i < n; i++)
    {
        if (query_seg(i, i))
            ans[i] = '1';
    }

    answer(ans);
}

void run()
{
    string submit(n, '0');

    deque<pii> segs;
    for (auto &v : initial_segs)
    {
        if (query_seg(v))
            segs.emplace_back(v);
    }

    while (szz(segs))
    {
        pii sg = segs.front();
        segs.pop_front();

        if (sg.va == sg.vb)
        {
            submit[sg.va] = '1';
            continue;
        }

        if (sg.va + 1 == sg.vb)
        {
            if (query_seg(sg.va, sg.va))
            {
                submit[sg.va] = '1';
                if (query_seg(sg.vb, sg.vb))
                {
                    submit[sg.vb] = '1';
                }
            }
            else
            {
                submit[sg.vb] = '1';
            }
            continue;
        }

        int m = (sg.vb - sg.va + 1) / 2;
        pii lsg(sg.va, sg.va + m - 1), rsg(sg.va + m, sg.vb);

        if (lsg.va <= lsg.vb && query_seg(lsg))
        {
            segs.push_back(lsg);

            if (rsg.va <= rsg.vb && query_seg(rsg))
            {
                segs.push_back(rsg);
            }
        }
        else
        {
            if (rsg.va <= rsg.vb)
            {
                segs.push_back(rsg);
            }
        }
    }

    answer(submit);
}

int main()
{
    cin >> n >> p >> t;

    if (t == 1)
    {
        init();
        run_det();
    }
    else
    {
        for (int j = 0; j < t; j++)
        {
            init();
            run();
        }
        debug("average query: %.3Lf\n", lf(qnt) / t);
        debug("max query: %d\n", mqnt);
    }
}

Compilation message

Main.cpp: In function 'int main()':
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:313:9: note: in expansion of macro 'debug'
  313 |         debug("average query: %.3Lf\n", lf(qnt) / t);
      |         ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:314:9: note: in expansion of macro 'debug'
  314 |         debug("max query: %d\n", mqnt);
      |         ^~~~~
# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 6 ms 344 KB Output is correct
2 Correct 5 ms 344 KB Output is correct
3 Correct 5 ms 344 KB Output is correct
4 Correct 5 ms 344 KB Output is correct
5 Correct 4 ms 344 KB Output is correct
6 Correct 7 ms 344 KB Output is correct
7 Correct 5 ms 344 KB Output is correct
8 Correct 5 ms 344 KB Output is correct
9 Correct 5 ms 344 KB Output is correct
10 Correct 5 ms 344 KB Output is correct
11 Correct 5 ms 344 KB Output is correct
12 Correct 6 ms 596 KB Output is correct
13 Correct 5 ms 344 KB Output is correct
14 Correct 5 ms 344 KB Output is correct
15 Correct 5 ms 344 KB Output is correct
16 Correct 4 ms 344 KB Output is correct
17 Correct 4 ms 344 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 25 ms 344 KB Output is correct (P=0.001, F=15.1, Q=14.2) -> 90.00 points
2 Correct 107 ms 344 KB Output is correct (P=0.005256, F=51.1, Q=57.3) -> 60.59 points
3 Correct 205 ms 344 KB Output is correct (P=0.011546, F=94.9, Q=109.1) -> 56.30 points
4 Correct 435 ms 344 KB Output is correct (P=0.028545, F=191.5, Q=218.9) -> 57.24 points
5 Correct 529 ms 344 KB Output is correct (P=0.039856, F=246.3, Q=280.3) -> 57.98 points
6 Correct 836 ms 428 KB Output is correct (P=0.068648, F=366.2, Q=405.8) -> 62.83 points
7 Correct 1048 ms 344 KB Output is correct (P=0.104571, F=490.3, Q=527.3) -> 69.13 points
8 Correct 1279 ms 344 KB Output is correct (P=0.158765, F=639.1, Q=665.6) -> 77.20 points
9 Correct 1442 ms 344 KB Output is correct (P=0.2, F=731.4, Q=770.0) -> 74.31 points