#include <bits/stdc++.h>
#include "cielib.h"
using namespace std;
#define M 500
//#define test
#ifdef test
#define DD 5
#define RR 1000000000
int Krotka[DD]={0,0,0,0,0};
int ym=-1;
int podajD() {return DD;}
int podajR() {return RR;}
int K;
int czyCieplo(int p[]) {
K++;
int m=0;
for (int i=0;i<DD;i++)
m = max(abs(Krotka[i]-p[i]),m);
bool cieplo = m<ym;
ym=m;
return cieplo;
}
void znalazlem(int p[]) {
cout << K << endl;
for (int i=0;i<DD;i++)
cout << p[i] << " ";
}
#endif
priority_queue<pair<int,int> > Q;
int a[M],b[M],p[M];
int D,R;
void init1() {
D=podajD();
R=podajR();
for (int i=0;i<D;i++) {
a[i]=0;
b[i]=R;
p[i]=(a[i]+b[i])/2;
Q.push(make_pair(b[i]-a[i],i));
}
}
void go() {
while (Q.top().first>0) {
int d=Q.top().second; Q.pop();
int g = (b[d]-a[d]+1)/2;
int pair = (b[d]-a[d])%2 == 0;
//printf("d=%d a=%d b=%d g=%d pair=%d\n",d,a[d],b[d],g,pair);
p[d]=a[d];
czyCieplo(p);
p[d]=b[d];
if (czyCieplo(p)) {
a[d] += g+pair;
if (b[d]-a[d]==1) a[d]--;
//cout << "prawo\n";
} else {
p[d]=a[d];
if (czyCieplo(p)) {
b[d] -= g+pair;
if (b[d]-a[d]==1) b[d]++;
//cout << "lewo\n";
} else {
int c=a[d];
a[d]=b[d]-g;
b[d]=c+g;
if (b[d]-a[d]==1) b[d]++;
//cout << "srodek\n";
}
}
Q.push(make_pair(b[d]-a[d],d));
p[d]=(a[d]+b[d])/2;
//printf("d=%d a=%d b=%d \n",d,a[d],b[d]);
}
znalazlem(p);
}
int main() {
init1();
go();
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 | #include <bits/stdc++.h> #include "cielib.h" using namespace std; #define M 500 //#define test #ifdef test #define DD 5 #define RR 1000000000 int Krotka[DD]={0,0,0,0,0}; int ym=-1; int podajD() {return DD;} int podajR() {return RR;} int K; int czyCieplo(int p[]) { K++; int m=0; for (int i=0;i<DD;i++) m = max(abs(Krotka[i]-p[i]),m); bool cieplo = m<ym; ym=m; return cieplo; } void znalazlem(int p[]) { cout << K << endl; for (int i=0;i<DD;i++) cout << p[i] << " "; } #endif priority_queue<pair<int,int> > Q; int a[M],b[M],p[M]; int D,R; void init1() { D=podajD(); R=podajR(); for (int i=0;i<D;i++) { a[i]=0; b[i]=R; p[i]=(a[i]+b[i])/2; Q.push(make_pair(b[i]-a[i],i)); } } void go() { while (Q.top().first>0) { int d=Q.top().second; Q.pop(); int g = (b[d]-a[d]+1)/2; int pair = (b[d]-a[d])%2 == 0; //printf("d=%d a=%d b=%d g=%d pair=%d\n",d,a[d],b[d],g,pair); p[d]=a[d]; czyCieplo(p); p[d]=b[d]; if (czyCieplo(p)) { a[d] += g+pair; if (b[d]-a[d]==1) a[d]--; //cout << "prawo\n"; } else { p[d]=a[d]; if (czyCieplo(p)) { b[d] -= g+pair; if (b[d]-a[d]==1) b[d]++; //cout << "lewo\n"; } else { int c=a[d]; a[d]=b[d]-g; b[d]=c+g; if (b[d]-a[d]==1) b[d]++; //cout << "srodek\n"; } } Q.push(make_pair(b[d]-a[d],d)); p[d]=(a[d]+b[d])/2; //printf("d=%d a=%d b=%d \n",d,a[d],b[d]); } znalazlem(p); } int main() { init1(); go(); return 0; } |
English