#include "lang.h"
#include "grader.h"
#include <bits/stdc++.h>
using namespace std;
#define debug(x) [&] { auto _x = x; cerr << __LINE__ << ": " << #x << " = " << _x << endl; return _x; }()
using ll = long long;
using pint = pair<int, int>;
double magic(double x) {
return abs(x);
}
template <typename T, class Hash, int MAX>
struct Data {
int tot = 0;
unordered_map<T, int, Hash> m;
set<pair<int, T>> inv;
double operator&(const Data<T, Hash, MAX> &o) const {
double ret = 0;
for (auto [k, v]: m) {
ret += magic((double) v / tot - (o.m.count(k) ? (double) o.m.at(k) / o.tot : 0));
}
for (auto [k, v]: o.m) {
if (not m.count(k)) ret += magic((double) v / o.tot);
}
return ret;
}
void operator+=(const Data<T, Hash, MAX> &o) {
for (auto &[k, v]: o.m) {
inv.erase({m[k], k});
inv.emplace(m[k] += v, k);
}
tot += o.tot;
while (inv.size() > MAX) {
tot -= inv.begin()->first;
m.erase(inv.begin()->second);
inv.erase(inv.begin());
}
}};
struct hash2 {
ll operator() (const pint &x) const { return (ll) x.first << 16 | x.second; }
};
struct hash3 {
ll operator()(const array<int, 3> &x) const { return ((ll) x[0] << 32) | ((ll) x[1] << 16) | x[2]; }
};
struct Profile {
Data<int, hash<int>, 50> sing;
Data<pint, hash2, 50> doub;
Data<array<int, 3>, hash3, 50> trip;
Profile() {}
Profile(int *arr) {
for (int i = 0; i < 100; ++i) {
sing.m[arr[i]]++;
if (i) doub.m[{arr[i - 1], arr[i]}]++;
if (i >=2) trip.m[{arr[i - 2], arr[i - 1], arr[i]}];
}
for (auto [k, v]: sing.m) sing.inv.emplace(v, k);
for (auto [k, v]: doub.m) doub.inv.emplace(v, k);
for (auto [k, v]: trip.m) trip.inv.emplace(v, k);
sing.tot = 100;
doub.tot = 99;
trip.tot = 98;
}
double operator&(const Profile &o) const {
return (sing & o.sing) + (doub & o.doub) + (trip & o.trip);
}
void operator+=(const Profile &o) {
sing += o.sing;
doub += o.doub;
trip += o.trip;
}
} profile[56];
void excerpt(int *E) {
Profile p(E);
pair<double, int> mn{1e18, 0};
for (int i = 0; i < 56; ++i) {
mn = min(mn, {p & profile[i], i});
}
int ans = language(mn.second);
profile[ans] += p;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4005 ms |
6880 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Partially correct |
3881 ms |
6916 KB |
Output is partially correct - 77.91% |