#include "bits/stdc++.h"
#include "cielib.h"
using namespace std;
const int D = 500;
const int d = podajD();
const int r = podajR();
bool flip[D];
// (2 * d) x czyCieplo()
void reduceToNonZeroCoordinates(){
vector<int> t(d, 1);
for(int i = 0; i < d; ++i) {
t[i] = r;
czyCieplo(t.data());
t[i] = r - 1;
flip[i] = (czyCieplo(t.data()) == 1);
t[i] = 1;
}
}
vector<int> flipped(int* t){
vector<int> tf(t, t + d);
for(int i = 0; i < d; ++i){
if(flip[i]) tf[i] = r - tf[i];
}
return tf;
}
int reducedCzyCieplo(int* t) {
return czyCieplo(flipped(t).data());
}
void reducedZnalazlem(int* t) {
znalazlem(flipped(t).data());
}
int main() {
reduceToNonZeroCoordinates();
vector<int> t(d, 0), tt(d);
unordered_set<int> poll;
for(int i = 0; i < d; ++i) {
poll.insert(i);
}
for(int i = 0; i < d; ++i) {
{
int lo = 0;
int hi = poll.size();
while(hi - lo > 1){
int mi = (lo + hi) / 2;
tt = t;
for(int j = 0; j < d; ++j) {
if(poll.count(j) == 0) ++tt[j];
}
auto it = poll.begin();
for(int k = 0; k < mi; k++) {
++tt[*it];
++it;
}
reducedCzyCieplo(t.data());
if(reducedCzyCieplo(tt.data()) == 1) {
hi = mi;
} else {
lo = mi;
}
}
auto jt = poll.begin();
advance(jt, lo);
poll.erase(jt);
}
{
tt = t;
for(int j = 0; j < d; ++j) {
if(poll.count(j) == 0) ++tt[j];
}
reducedCzyCieplo(t.data());
if(reducedCzyCieplo(tt.data()) == 0) {
continue;
}
}
{
int lo = 0;
int hi = r - *max_element(t.begin(), t.end());
while(hi - lo > 1) {
int mi = (lo + hi) / 2;
tt = t;
for(int j = 0; j < d; ++j) {
if(poll.count(j) == 0) tt[j] += mi;
}
reducedCzyCieplo(tt.data());
for(int j = 0; j < d; ++j) {
if(poll.count(j) == 0) ++tt[j];
}
if(reducedCzyCieplo(tt.data()) == 1) {
lo = mi;
} else {
hi = mi;
}
}
for(int j = 0; j < d; ++j) {
if(poll.count(j) == 0) t[j] += hi;
}
}
}
reducedZnalazlem(t.data());
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 | #include "bits/stdc++.h" #include "cielib.h" using namespace std; const int D = 500; const int d = podajD(); const int r = podajR(); bool flip[D]; // (2 * d) x czyCieplo() void reduceToNonZeroCoordinates(){ vector<int> t(d, 1); for(int i = 0; i < d; ++i) { t[i] = r; czyCieplo(t.data()); t[i] = r - 1; flip[i] = (czyCieplo(t.data()) == 1); t[i] = 1; } } vector<int> flipped(int* t){ vector<int> tf(t, t + d); for(int i = 0; i < d; ++i){ if(flip[i]) tf[i] = r - tf[i]; } return tf; } int reducedCzyCieplo(int* t) { return czyCieplo(flipped(t).data()); } void reducedZnalazlem(int* t) { znalazlem(flipped(t).data()); } int main() { reduceToNonZeroCoordinates(); vector<int> t(d, 0), tt(d); unordered_set<int> poll; for(int i = 0; i < d; ++i) { poll.insert(i); } for(int i = 0; i < d; ++i) { { int lo = 0; int hi = poll.size(); while(hi - lo > 1){ int mi = (lo + hi) / 2; tt = t; for(int j = 0; j < d; ++j) { if(poll.count(j) == 0) ++tt[j]; } auto it = poll.begin(); for(int k = 0; k < mi; k++) { ++tt[*it]; ++it; } reducedCzyCieplo(t.data()); if(reducedCzyCieplo(tt.data()) == 1) { hi = mi; } else { lo = mi; } } auto jt = poll.begin(); advance(jt, lo); poll.erase(jt); } { tt = t; for(int j = 0; j < d; ++j) { if(poll.count(j) == 0) ++tt[j]; } reducedCzyCieplo(t.data()); if(reducedCzyCieplo(tt.data()) == 0) { continue; } } { int lo = 0; int hi = r - *max_element(t.begin(), t.end()); while(hi - lo > 1) { int mi = (lo + hi) / 2; tt = t; for(int j = 0; j < d; ++j) { if(poll.count(j) == 0) tt[j] += mi; } reducedCzyCieplo(tt.data()); for(int j = 0; j < d; ++j) { if(poll.count(j) == 0) ++tt[j]; } if(reducedCzyCieplo(tt.data()) == 1) { lo = mi; } else { hi = mi; } } for(int j = 0; j < d; ++j) { if(poll.count(j) == 0) t[j] += hi; } } } reducedZnalazlem(t.data()); return 0; } |
English