Submission #1119905

# Submission time Handle Problem Language Result Execution time Memory
1119905 2024-11-27T15:01:29 Z Tenis0206 COVID tests (CEOI24_covid) C++11
10 / 100
4942 ms 504 KB
#include <bits/stdc++.h>

using namespace std;

/// You may use:

// The number of students
int N;

// The probability any given student is positive
double P;

// This function performs a test on a subset of samples.
// Its argument is a vector of Booleans of length N,
// where the i-th element is true if the i-th sample should be added to the mix.
// It returns true if (and only if) at least one of the samples in the mix is positive.
bool test_students(std::vector<bool> mask)
{
    assert(mask.size() == (size_t)N);

    std::string mask_str(N, ' ');
    for (int i = 0; i < N; i++)
        mask_str[i] = mask[i] ? '1' : '0';

    printf("Q %s\n", mask_str.c_str());
    fflush(stdout);

    char answer;
    scanf(" %c", &answer);
    return answer == 'P';
}

/// You should implement:

// This function will be called once for each test instance.
// It should use test_students to determine which samples are positive.
// It must return a vector of Booleans of length N,
// where the i-th element is true if and only if the i-th sample is positive.

std::vector<bool> answer;

bool my_test_students(int st, int dr)
{
    vector<bool> mask(N);
    for(int i=st; i<=dr; i++)
    {
        mask[i] = true;
    }
    return test_students(mask);
}

void divide(int st, int dr)
{
    if(st == dr)
    {
        answer[st] = true;
        return;
    }
    int mij = (st + dr) >> 1;
    bool ok_st = my_test_students(st, mij);
    bool ok_dr = my_test_students(mij + 1, dr);
    if(ok_st)
    {
        divide(st, mij);
    }
    if(ok_dr)
    {
        divide(mij + 1, dr);
    }
}

int T;

void find_positive()
{
    if(P < 0.01)
    {
        int poz = 0;
        while(poz < N)
        {
            if(my_test_students(poz, N) == false)
            {
                break;
            }
            int st = poz;
            int dr = N - 1;
            int rez = -1;
            while(st <= dr)
            {
                int mij = (st + dr) >> 1;
                bool ok = my_test_students(poz, mij);
                if(ok)
                {
                    rez = mij;
                    dr = mij - 1;
                }
                else
                {
                    st = mij + 1;
                }
            }
            if(rez == -1)
            {
                break;
            }
            answer[rez] = true;
            poz = rez + 1;
        }
    }
    else if(P < 0.2 || T == 1)
    {
        int poz = 0;
        while(poz < N)
        {
            int st = poz;
            int dr = min(N - 1, (int)(poz + 3 * (1 / P)));
            int rez = -1;
            while(st <= dr)
            {
                int mij = (st + dr) >> 1;
                bool ok = my_test_students(poz, mij);
                if(ok)
                {
                    rez = mij;
                    dr = mij - 1;
                }
                else
                {
                    st = mij + 1;
                }
            }
            if(rez == -1)
            {
                st = poz;
                dr = N - 1;
                rez = -1;
                while(st <= dr)
                {
                    int mij = (st + dr) >> 1;
                    bool ok = my_test_students(poz, mij);
                    if(ok)
                    {
                        rez = mij;
                        dr = mij - 1;
                    }
                    else
                    {
                        st = mij + 1;
                    }
                }
                if(rez == -1)
                {
                    break;
                }
            }
            answer[rez] = true;
            poz = rez + 1;
        }
    }
    else
    {
        int poz = 0;
        while(poz < N)
        {
            int st = poz;
            int dr = min(N - 1, poz + 12);
            int rez = -1;
            while(st <= dr)
            {
                int mij = (st + dr) >> 1;
                bool ok = my_test_students(poz, mij);
                if(ok)
                {
                    rez = mij;
                    dr = mij - 1;
                }
                else
                {
                    st = mij + 1;
                }
            }
            if(rez == -1)
            {
                st = poz;
                dr = N - 1;
                rez = -1;
                while(st <= dr)
                {
                    int mij = (st + dr) >> 1;
                    bool ok = my_test_students(poz, mij);
                    if(ok)
                    {
                        rez = mij;
                        dr = mij - 1;
                    }
                    else
                    {
                        st = mij + 1;
                    }
                }
                if(rez == -1)
                {
                    break;
                }
            }
            answer[rez] = true;
            poz = rez + 1;
        }
    }
}

int main()
{
    scanf("%d %lf %d", &N, &P, &T);

    // You may perform any extra initialization here.

    for (int i = 0; i < T; i++)
    {
        answer.resize(N);
        for(int j=0; j<N; j++)
        {
            answer[j] = false;
        }
        find_positive();

        std::string answer_str(N, ' ');
        for (int j = 0; j < N; j++)
            answer_str[j] = answer[j] ? '1' : '0';

        printf("A %s\n", answer_str.c_str());
        fflush(stdout);

        char verdict;
        scanf(" %c", &verdict);
        if (verdict == 'W')
            exit(0);
    }

    return 0;
}

Compilation message

Main.cpp: In function 'bool test_students(std::vector<bool>)':
Main.cpp:29:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   29 |     scanf(" %c", &answer);
      |     ~~~~~^~~~~~~~~~~~~~~~
Main.cpp: In function 'int main()':
Main.cpp:214:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  214 |     scanf("%d %lf %d", &N, &P, &T);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:235:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  235 |         scanf(" %c", &verdict);
      |         ~~~~~^~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 504 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 27 ms 336 KB Output is correct
2 Correct 25 ms 336 KB Output is correct
3 Correct 31 ms 336 KB Output is correct
4 Correct 28 ms 504 KB Output is correct
5 Correct 37 ms 336 KB Output is correct
6 Correct 31 ms 336 KB Output is correct
7 Correct 26 ms 336 KB Output is correct
8 Correct 21 ms 336 KB Output is correct
9 Correct 29 ms 336 KB Output is correct
10 Correct 14 ms 336 KB Output is correct
11 Correct 24 ms 336 KB Output is correct
12 Correct 21 ms 336 KB Output is correct
13 Correct 24 ms 336 KB Output is correct
14 Correct 19 ms 336 KB Output is correct
15 Correct 24 ms 336 KB Output is correct
16 Correct 40 ms 336 KB Output is correct
17 Correct 1 ms 336 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 67 ms 336 KB Output is correct (P=0.001, F=15.1, Q=10.7) -> 90.00 points
2 Correct 300 ms 336 KB Output is correct (P=0.005256, F=51.1, Q=52.9) -> 78.89 points
3 Correct 595 ms 336 KB Output is correct (P=0.011546, F=94.9, Q=106.2) -> 60.96 points
4 Correct 1189 ms 336 KB Output is correct (P=0.028545, F=191.5, Q=210.5) -> 64.43 points
5 Correct 1633 ms 336 KB Output is correct (P=0.039856, F=246.3, Q=275.6) -> 60.98 points
6 Correct 2388 ms 336 KB Output is correct (P=0.068648, F=366.2, Q=410.2) -> 60.79 points
7 Correct 3175 ms 336 KB Output is correct (P=0.104571, F=490.3, Q=552.5) -> 59.70 points
8 Correct 4245 ms 336 KB Output is correct (P=0.158765, F=639.1, Q=735.7) -> 56.09 points
9 Execution timed out 4942 ms 336 KB Time limit exceeded (wall clock)
10 Halted 0 ms 0 KB -