Submission #566432

#	Time	Username	Problem	Language	Result	Execution time	Memory
566432		drdilyor	Lottery (CEOI18_lot)	C++17	0 / 100	3096 ms	520 KiB

lot

#ifdef ONPC
    //#define _GLIBCXX_DEBUG
    #define debug(...) cerr << "[" << #__VA_ARGS__ << "]: ", debug_out(__VA_ARGS__)
#else
    #define debug(...) 42
#endif
#define allit(a) (a).begin(), (a).end()
#define memclr(a) memset(a, 0, sizeof(0))
#define sz(a) ((int) (a).size())

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ii = pair<int, int>;
using vi = vector<int>;
using vii = vector<pair<int, int>>;
using vvi = vector<vector<int>>;
template<typename T>
using uset = unordered_set<T>;
template<typename K, typename V>
using umap = unordered_map<K, V>;

//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/detail/standard_policies.hpp>
//namespace pd = __gnu_pbds;
//template<typename K>
//using ordered_set = pd::tree<
//    K,
//    pd::null_type,
//    less<int>,
//    pd::rb_tree_tag,
//    pd::tree_order_statistics_node_update>;
//using pd::gp_hash_table;

void debug_out() {
    cerr << endl;
}
template<typename... Tail>
void debug_out(vector<int> h, Tail... t) {
    cerr << "[";
    for (int i = 0; i < sz(h); i++) {
        if (i != 0) cerr << ", ";
        cerr << h[i];
    }
    cerr << "]";
    debug_out(t...);
}
template<typename P1, typename P2, typename... Tail>
void debug_out(pair<P1, P2> h, Tail... t) {
    cerr << "[" << h.first << ", " << h.second << "]";
    debug_out(t...);
}
template<typename Head, typename... Tail>
void debug_out(Head h, Tail... t) {
    cerr << "[" << h << "]";
    debug_out(t...);
}

const int RANDOM = chrono::high_resolution_clock::now().time_since_epoch().count();
struct chash {
    int operator() (int x) const { return x ^ RANDOM; }
};

const int INF = 1e9+1;
const ll INFL = 1e18+1;
const int N = 1e5;

int solve();

signed main() {
    cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    //cin >> t;
    while (t-- && cin) {
        if (solve()) break;
        #ifdef ONPC
            cout << "____________________" << endl;
        #endif
    }

    return 0;
}

struct UnionFind {
    int n;
    vi parent;
    UnionFind(int n_) : n(n_), parent(n_) {
        for (int i = 0; i < n; i++) parent[i] = i;
    }
    int get(int i) {
        if (parent[i] == i) return i;
        return parent[i] = get(parent[i]);
    }
    void merge(int a, int b) {
        parent[b] = a;
    }
};

int solve() {
    int n, l, q;
    cin >> n >> l;
    vi a(n);
    for (auto& i : a) cin >> i;

    vi res(n-l+1);
    vi done(n-l+1);
    for (int i1 = 0; i1 <= n - l; i1++) {
        if (done[i1]) continue;
        uset<int> cur = {i1};
        for (int i2 = i1 + 1; i2 <= n-l; i2++) {
            if (done[i2]) continue;
            int bad = 0;
            for (int j = 0; j < l; j++) {
                if (a[i1+j] != a[i2+j]) {
                    bad = 1;
                    break;
                }
            }
            if (!bad) {
                cur.insert(i2);
            }
        }
        for (int i : cur) {
            done[i] = 1;
            res[i] = sz(cur)-1;
        }
    }
    for (int i : res) cout << i << ' ';
    return 0;
}

Compilation message (stderr)

lot.cpp: In function 'int solve()':
lot.cpp:100:15: warning: unused variable 'q' [-Wunused-variable]
  100 |     int n, l, q;
      |               ^

• Subtask 1: 0 / 25
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20