#include "cielib.h" #include <bits/stdc++.h> using namespace std; int main() { int d = podajD(); int r = podajR(); int wynik[500]; for (int i = 0; i < d; i++) wynik[i] = 0; unordered_set<int> done; int plus = 0; while (done.size() < d) { czyCieplo(wynik); for (int i = 0; i < d; i++) { if (done.find(i) == done.end()) { for (int j = 0; j < d; j++) { if (j != i) { wynik[j]++; } } if (!czyCieplo(wynik)) { done.insert(i); } for (int j = 0; j < d; j++) { if (j != i) { wynik[j]--; } } czyCieplo(wynik); } } int left = 0, right = r - plus; while (left != right) { int middle = (left + right) / 2; for (int idx : done) { wynik[idx] += middle; } czyCieplo(wynik); for (int idx : done) { wynik[idx]++; } if (czyCieplo(wynik)) { left = middle + 1; } else { right = middle; } if (left != right) { for (int idx : done) { wynik[idx] -= middle + 1; } } else { for (int idx : done) { wynik[idx] -= middle + 1; wynik[idx] += left; } plus += left; } } } znalazlem(wynik); 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 | #include "cielib.h" #include <bits/stdc++.h> using namespace std; int main() { int d = podajD(); int r = podajR(); int wynik[500]; for (int i = 0; i < d; i++) wynik[i] = 0; unordered_set<int> done; int plus = 0; while (done.size() < d) { czyCieplo(wynik); for (int i = 0; i < d; i++) { if (done.find(i) == done.end()) { for (int j = 0; j < d; j++) { if (j != i) { wynik[j]++; } } if (!czyCieplo(wynik)) { done.insert(i); } for (int j = 0; j < d; j++) { if (j != i) { wynik[j]--; } } czyCieplo(wynik); } } int left = 0, right = r - plus; while (left != right) { int middle = (left + right) / 2; for (int idx : done) { wynik[idx] += middle; } czyCieplo(wynik); for (int idx : done) { wynik[idx]++; } if (czyCieplo(wynik)) { left = middle + 1; } else { right = middle; } if (left != right) { for (int idx : done) { wynik[idx] -= middle + 1; } } else { for (int idx : done) { wynik[idx] -= middle + 1; wynik[idx] += left; } plus += left; } } } znalazlem(wynik); return 0; } |