Submission #1036605

#	Time	Username	Problem	Language	Result	Execution time	Memory
1036605		TAMREF	COVID tests (CEOI24_covid)	C++17	97.70 / 100	1407 ms	344 KiB

Main

// 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());
 
map<lf, lf> score = {
    {0.001, 15.1},
    {0.005256, 51.1},
    {0.011546, 94.9},
    {0.028545, 191.5},
    {0.039856, 246.3},
    {0.068648,
     366.2},
    {0.104571,
     490.3},
    {0.158765, 639.1},
    {0.2, 731.4}};
string ans;
 
int n, t, qnt, lqnt, mqnt;
int cnt[10005];
lf p;
 
int query(string F)
{
    ++qnt;
    ++lqnt;
 
#ifndef TAMREF
    cout << "Q " << F << '\n';
    cout << flush;
#endif
 
    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)
{
    // debug("qseg %d %d\n", i, j);
    if (i > j)
        return 0;
    
    cnt[j - i + 1]++;
    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;
 
#ifndef TAMREF
    cout << "A " << F << '\n';
    cout << flush;
#endif
 
#ifdef TAMREF
    assert(ans == F);
#else
    string C;
    cin >> C;
    assert(C == "C");
#endif
}
 

lf dp[2][10005]; // 0 : (1-q^i)Ki, 1 : Ui
int opt[2][10005];
vector<int> fib({1, 2});
lf pows[10005] = {1.0};
vector<ti> initial_segs;
 
void init()
{
    if (t > 1 && initial_segs.empty())
    {
        while(fib.back() < n) fib.push_back(fib[szz(fib) - 2] + fib[szz(fib) - 1]);

        for (int i = 1; i <= n; i++)
            pows[i] = pow(1.0L - p, i);
 
        dp[1][1] = 1.L;

        for (int i = 2; i <= n; i++)
        {
            dp[0][i] = 100.L * i;

            for(int j = 1; j < i; j++) {
                // if (!bs(all(fib), j)) continue;
                lf pdp = 1.L - pows[i] + (1.L - pows[j]) * dp[1][i - j] + pows[j] * dp[0][i - j] + dp[0][j];
                if(pdp < dp[0][i]) {
                    dp[0][i] = pdp;
                    opt[0][i] = j;
                }
                if (pdp < dp[0][i] + 0.5) {
                    debug("dp[0][%d] = %.3Lf <- pdp[%d] = %.3Lf\n", i, dp[0][i], j, pdp);
                }
            }

            dp[1][i] = 100.L * i;

            // don't know -> dp
            for(int j = 1; j <= i; j++) {
                // if (!bs(all(fib), j) && j < i) continue;
                // dp[i] * q^i
                lf pdp = 1.L + dp[0][j] + dp[1][i - j];
                if (pdp < dp[1][i]) {
                    dp[1][i] = pdp;
                    opt[1][i] = j;
                }
                if (pdp < dp[1][i] + 0.5) {
                    debug("dp[1][%d] = %.3Lf <- pdp[%d] = %.3Lf\n", i, dp[1][i], j, pdp);
                }
            }
        }
    

        for(int i = 1; i <= n; i++) {
            debug("opt[0][%d] = %d, opt[1][%d] = %d\n", i, opt[0][i], i, opt[1][i]);
        }
        initial_segs.emplace_back(0, n-1, 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');
#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()
{
    // override
    if (p > 0.15)
    opt[1][4] = 3;

    debug("================\n");
    string submit(n, '0');
 
    deque<ti> segs;
    for (auto &[x, y, i] : initial_segs)
    {
        segs.emplace_back(x, y, i);
    }
 
    while (szz(segs))
    {
        auto [x, y, i] = segs.back();
        segs.pop_back();

        if (x > y) continue;

        // merge unknown segments
        if (i == 1) {
            while (szz(segs) && get<2>(segs.back()) == 1) {
                assert(y + 1 == get<0>(segs.back()));
                y = get<1>(segs.back());
                segs.pop_back();
            }
        }

        if (x == y)
        {
            if(i == 0) {
            submit[x] = '1';
            } else {
                if (query_seg(x, x)) {
                    submit[x] = '1';
                }
            }
            continue;
        }
 
        int len = y - x + 1;
        int m = opt[i][len];

        pii lsg(x, x + m - 1), rsg(x + m, y);
        
        if (query_seg(lsg)) {
            segs.emplace_back(rsg.va, rsg.vb, 1);
            segs.emplace_back(lsg.va, lsg.vb, 0);
        } else {
            segs.emplace_back(rsg.va, rsg.vb, i);
        }
    }
 
    debug("lqnt = %d\n", lqnt);
    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);
        debug("thres: %.3Lf", score.lower_bound(p - 1e-7)->vb);
        for(int i = 1; i <= n; i++) {
            if (cnt[i]) {
                debug("query(%d): %.3Lf\n", i, lf(cnt[i]) / qnt);
            }
        }
    }
}

Compilation message (stderr)

Main.cpp: In function 'void init()':
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:246:21: note: in expansion of macro 'debug'
  246 |                     debug("dp[0][%d] = %.3Lf <- pdp[%d] = %.3Lf\n", i, dp[0][i], j, pdp);
      |                     ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:262:21: note: in expansion of macro 'debug'
  262 |                     debug("dp[1][%d] = %.3Lf <- pdp[%d] = %.3Lf\n", i, dp[1][i], j, pdp);
      |                     ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:269:13: note: in expansion of macro 'debug'
  269 |             debug("opt[0][%d] = %d, opt[1][%d] = %d\n", i, opt[0][i], i, opt[1][i]);
      |             ^~~~~
Main.cpp: In function 'void run()':
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:301:5: note: in expansion of macro 'debug'
  301 |     debug("================\n");
      |     ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:351:5: note: in expansion of macro 'debug'
  351 |     debug("lqnt = %d\n", lqnt);
      |     ^~~~~
Main.cpp: In function 'int main()':
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:371:9: note: in expansion of macro 'debug'
  371 |         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:372:9: note: in expansion of macro 'debug'
  372 |         debug("max query: %d\n", mqnt);
      |         ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:373:9: note: in expansion of macro 'debug'
  373 |         debug("thres: %.3Lf", score.lower_bound(p - 1e-7)->vb);
      |         ^~~~~
Main.cpp:106:20: warning: statement has no effect [-Wunused-value]
  106 | #define debug(...) 42
      |                    ^~
Main.cpp:376:17: note: in expansion of macro 'debug'
  376 |                 debug("query(%d): %.3Lf\n", i, lf(cnt[i]) / qnt);
      |                 ^~~~~

• Subtask 0: 0 / 0
• Subtask 1: 10 / 10
• Subtask 2: 87.699999996 / 90