oj.uz

Submission #917222

# Submission time Handle Problem Language Result Execution time Memory
917222 2024-01-27T13:09:30 Z GrindMachine Super Dango Maker (JOI22_dango3) C++17
100 / 100
 2491 ms 1204 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

template<typename T> using Tree = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
typedef long long int ll;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

#define fastio ios_base::sync_with_stdio(false); cin.tie(NULL)
#define pb push_back
#define endl '\n'
#define sz(a) (int)a.size()
#define setbits(x) __builtin_popcountll(x)
#define ff first
#define ss second
#define conts continue
#define ceil2(x,y) ((x+y-1)/(y))
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define yes cout << "Yes" << endl
#define no cout << "No" << endl

#define rep(i,n) for(int i = 0; i < n; ++i)
#define rep1(i,n) for(int i = 1; i <= n; ++i)
#define rev(i,s,e) for(int i = s; i >= e; --i)
#define trav(i,a) for(auto &i : a)

template<typename T>
void amin(T &a, T b) {
    a = min(a,b);
}

template<typename T>
void amax(T &a, T b) {
    a = max(a,b);
}

#ifdef LOCAL
#include "debug.h"
#else
#define debug(x) 42
#endif

/*

read the solution a long time ago, remember the key idea of the solution

looking at the constraints and the allowed #of queries, O(nm*log(m)) looks like the intended #of queries
maintain m groups, all of which are initially empty
for each guy from 1 to n*m, insert him into the first group that doesnt contain the value
because we always insert a value into the first group that doesnt contain the value, all occurrences of a value at any given time lie on a prefix of groups
so we can b.s to find the first group that doesnt contain a[i]

how to check if a group contains a[i]?
put i into the group and query the complement set of the group
if there are 2 occurrences of a[i], then the result of the query would be m-2
otherwise, a[i] is not present in the group, so we get a result >= m-1

*/

const int MOD = 1e9 + 7;
const int N = 1e5 + 5;
const int inf1 = int(1e9) + 5;
const ll inf2 = ll(1e18) + 5;

#include "dango3.h"

void Solve(int n, int m) {
    vector<vector<int>> group(m+5);
    vector<bool> there(n*m+5);

    auto ok = [&](vector<int> v){
        fill(all(there),0);
        trav(x,v) there[x] = 1;

        vector<int> v_compl;
        rep1(i,n*m){
            if(!there[i]){
                v_compl.pb(i);
            }
        }

        int res = Query(v_compl);

        if(res == m-2) return false;
        else return true;
    };

    rep1(i,n*m){
        int l = 1, r = m;
        int first = -1;

        while(l <= r){
            int mid = (l+r) >> 1;
            group[mid].pb(i);
            if(ok(group[mid])){
                first = mid;
                r = mid-1;
            }
            else{
                l = mid+1;
            }

            group[mid].pop_back();    
        }

        assert(first != -1);
        group[first].pb(i);
    }

    rep1(i,m){
        Answer(group[i]);
    }
}

Subtask #1
2.0 / 2.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 500 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 348 KB Output is correct
2 Correct 18 ms 504 KB Output is correct
3 Correct 18 ms 344 KB Output is correct
4 Correct 19 ms 348 KB Output is correct
5 Correct 19 ms 512 KB Output is correct
6 Correct 18 ms 504 KB Output is correct

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 562 ms 564 KB Output is correct
2 Correct 564 ms 592 KB Output is correct
3 Correct 629 ms 812 KB Output is correct
4 Correct 664 ms 832 KB Output is correct
5 Correct 556 ms 580 KB Output is correct
6 Correct 558 ms 852 KB Output is correct

Subtask #4
78.0 / 78.0

# Verdict Execution time Memory Grader output
1 Correct 2268 ms 952 KB Output is correct
2 Correct 2237 ms 864 KB Output is correct
3 Correct 2486 ms 1144 KB Output is correct
4 Correct 2491 ms 888 KB Output is correct
5 Correct 2242 ms 1204 KB Output is correct
6 Correct 2224 ms 1168 KB Output is correct