|
| 1 | +#include <iostream> |
| 2 | +#include <sstream> |
| 3 | +#include <vector> |
| 4 | +#include <algorithm> |
| 5 | +#include <set> |
| 6 | +#include <map> |
| 7 | +#include <cstring> |
| 8 | +using namespace std; |
| 9 | +struct node{ |
| 10 | + int h; |
| 11 | + int b; |
| 12 | + int w; |
| 13 | + node(int hh = 0, int bb = 0, int ww = 0){ h = hh; b = bb; w = ww;} |
| 14 | +}; |
| 15 | +bool operator<(const node &l, const node &r){ |
| 16 | + return l.h < r.h; |
| 17 | +} |
| 18 | +vector<node> A; |
| 19 | +int done[1005]; |
| 20 | +int B[1005][1005]; |
| 21 | +int n; |
| 22 | +void find_block(){ |
| 23 | + |
| 24 | + vector<int> r(n); |
| 25 | + fill(r.begin(), r.end(), 0); |
| 26 | + for(int i = 0; i < n; i++){ |
| 27 | + node p = A[i]; |
| 28 | + for(int j = 0; j < n; j++){ |
| 29 | + node q = A[j]; |
| 30 | + if(j != i && !done[j] && (q.h < p.h) && ((q.b <= p.b && p.b <= q.w) || (q.b <= p.w && p.w <= q.w) || (p.b <= q.b && q.b <= p.w) || (p.b <= q.w && q.w <= p.w))){ |
| 31 | + B[i][j] = 1; |
| 32 | + cout<<q.h<<" is blocked by "<<p.h<<endl; |
| 33 | + } |
| 34 | + } |
| 35 | + } |
| 36 | +} |
| 37 | + |
| 38 | +class FlowerGarden { |
| 39 | + public: |
| 40 | + vector <int> getOrdering(vector <int> height, vector <int> bloom, vector <int> wilt) |
| 41 | + { |
| 42 | + n = height.size(); |
| 43 | + vector<int> ret; |
| 44 | + A.resize(n); |
| 45 | + memset(done, 0, sizeof(done)); |
| 46 | + memset(B, 0, sizeof(B)); |
| 47 | + for(int i = 0; i < n; i++) A[i] = node(height[i], bloom[i], wilt[i]); |
| 48 | + sort(A.rbegin(), A.rend()); |
| 49 | + find_block(); |
| 50 | + for(int k = 0; k < n; k++){ |
| 51 | + int block[1005]; |
| 52 | + memset(block, 0, sizeof(block)); |
| 53 | + for(int i = 0; i < n; i++) |
| 54 | + for(int j = i+1; j < n; j++) |
| 55 | + if(B[i][j] && !done[j]){ |
| 56 | + block[i] = 1; |
| 57 | + j = n; |
| 58 | + } |
| 59 | + |
| 60 | + for(int i = 0; i < n; i++) |
| 61 | + if(!block[i] && !done[i]){ |
| 62 | + ret.push_back(A[i].h); |
| 63 | + done[i] = 1; |
| 64 | + i = n; |
| 65 | + } |
| 66 | + |
| 67 | + } |
| 68 | + return ret; |
| 69 | + } |
| 70 | +// BEGIN CUT HERE |
| 71 | + public: |
| 72 | + void run_test(int Case) { if ((Case == -1) || (Case == 0)) test_case_0(); if ((Case == -1) || (Case == 1)) test_case_1(); if ((Case == -1) || (Case == 2)) test_case_2(); if ((Case == -1) || (Case == 3)) test_case_3(); if ((Case == -1) || (Case == 4)) test_case_4(); if ((Case == -1) || (Case == 5)) test_case_5(); } |
| 73 | + private: |
| 74 | + template <typename T> string print_array(const vector<T> &V) { ostringstream os; os << "{ "; for (typename vector<T>::const_iterator iter = V.begin(); iter != V.end(); ++iter) os << '\"' << *iter << "\","; os << " }"; return os.str(); } |
| 75 | + void verify_case(int Case, const vector <int> &Expected, const vector <int> &Received) { cerr << "Test Case #" << Case << "..."; if (Expected == Received) cerr << "PASSED" << endl; else { cerr << "FAILED" << endl; cerr << "\tExpected: " << print_array(Expected) << endl; cerr << "\tReceived: " << print_array(Received) << endl; } } |
| 76 | + void test_case_0() { int Arr0[] = {5,4,3,2,1}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,1,1,1,1}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {365,365,365,365,365}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 1, 2, 3, 4, 5 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(0, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 77 | + void test_case_1() { int Arr0[] = {5,4,3,2,1}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,5,10,15,20}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {4,9,14,19,24}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 5, 4, 3, 2, 1 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(1, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 78 | + void test_case_2() { int Arr0[] = {5,4,3,2,1}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,5,10,15,20}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {5,10,15,20,25}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 1, 2, 3, 4, 5 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(2, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 79 | + void test_case_3() { int Arr0[] = {5,4,3,2,1}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,5,10,15,20}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {5,10,14,20,25}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 3, 4, 5, 1, 2 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(3, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 80 | + void test_case_4() { int Arr0[] = {1,2,3,4,5,6}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,3,1,3,1,3}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {2,4,2,4,2,4}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 2, 4, 6, 1, 3, 5 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(4, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 81 | + void test_case_5() { int Arr0[] = {3,2,5,4}; vector <int> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); int Arr1[] = {1,2,11,10}; vector <int> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arr2[] = {4,3,12,13}; vector <int> Arg2(Arr2, Arr2 + (sizeof(Arr2) / sizeof(Arr2[0]))); int Arr3[] = { 4, 5, 2, 3 }; vector <int> Arg3(Arr3, Arr3 + (sizeof(Arr3) / sizeof(Arr3[0]))); verify_case(5, Arg3, getOrdering(Arg0, Arg1, Arg2)); } |
| 82 | + |
| 83 | +// END CUT HERE |
| 84 | + |
| 85 | +}; |
| 86 | + |
| 87 | +// BEGIN CUT HERE |
| 88 | + int main() |
| 89 | + { |
| 90 | + FlowerGarden ___test; |
| 91 | + ___test.run_test(-1); |
| 92 | + } |
| 93 | +// END CUT HERE |
0 commit comments