ADDED SRM/TCO18-2A-U/1A.cpp Index: SRM/TCO18-2A-U/1A.cpp ================================================================== --- SRM/TCO18-2A-U/1A.cpp +++ SRM/TCO18-2A-U/1A.cpp @@ -0,0 +1,103 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +using namespace std; +typedef long long LL; +typedef complex CMP; + +class ArithmeticSequenceDiv1 { public: + int findMinCost(vector x) + { + int best = 0x7fffffff; + for (int i = 0; i < x.size(); ++i) { + // x[i]: fixed + for (int z = 0; z < 1000; ++z) { + int ss[] = { -1,+1 }; + for (int s : ss) { + int cur_cost = 0; + for (int k = 0; k < x.size(); ++k) { + int v = x[i]+s*z*(k-i); + cur_cost += abs(v - x[k]); + } + best = min(best, cur_cost); + } + } + } + return best; + } +}; + +// BEGIN CUT HERE +#include +double start_time; string timer() + { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); } +template ostream& operator<<(ostream& os, const vector& v) + { os << "{ "; + for(typename vector::const_iterator it=v.begin(); it!=v.end(); ++it) + os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; } +void verify_case(const int& Expected, const int& Received) { + bool ok = (Expected == Received); + if(ok) cerr << "PASSED" << timer() << endl; else { cerr << "FAILED" << timer() << endl; + cerr << "\to: \"" << Expected << '\"' << endl << "\tx: \"" << Received << '\"' << endl; } } +#define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock(); +#define END verify_case(_, ArithmeticSequenceDiv1().findMinCost(x));} +int main(){ + +CASE(0) + int x_[] = {1,3,2}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 2; +END +CASE(1) + int x_[] = {1,1,1,2,3,4,5}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 3; +END +CASE(2) + int x_[] = {1,2,3,4}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 0; +END +CASE(3) + int x_[] = {1,5,2,5}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 5; +END +CASE(4) + int x_[] = {11,33,22}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 17; +END +CASE(5) + int x_[] = {1, 3, 5, 7, 2, 4, 6}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = 12; +END +CASE(6) + int x_[] = {17,68,99,48,66,66,20,88,91,4,20,7,82,23,76,55,50,7,46,88,73,40,41,48,32,18,71,58,56,68,2,99,7,13,19,11,55,95,23,40,17,89,41,12,5,82,93,93,19,20,41,8,15,24,53,87,54,10,12,41,86,95,92,13,56,6,67,72,95,78,82,34,75,12,11,100,99,46,32,45,59,1,63,60,13,94,33,13,85,35,65,40,37,22,37,87,35,2,92,64}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = -1; +END +/* +CASE(7) + int x_[] = ; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int _ = ; +END +*/ +} +// END CUT HERE ADDED SRM/TCO18-2A-U/1B.cpp Index: SRM/TCO18-2A-U/1B.cpp ================================================================== --- SRM/TCO18-2A-U/1B.cpp +++ SRM/TCO18-2A-U/1B.cpp @@ -0,0 +1,228 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +using namespace std; +typedef long long LL; +typedef complex CMP; + +class MakingRegularGraph { public: + vector addEdges(int n, vector x, vector y) + { + vector deg(n); + for (int i = 0; i < x.size(); ++i) + deg[x[i]]++, deg[y[i]]++; + + set d0, d1; + for (int v = 0; v < n; ++v) + if (deg[v] == 0) d0.insert(v); + else if (deg[v] == 1) d1.insert(v); + map d1e; + for (int i = 0; i < x.size(); ++i) + if (d1.count(x[i]) && d1.count(y[i])) { + d1e[x[i]] = y[i]; + d1e[y[i]] = x[i]; + d1.erase(x[i]); + d1.erase(y[i]); + } + + #define BAD (((d0.size()==1||d0.size()==2) && d1.empty() && d1e.empty()) || (d0.empty() && d1.empty() && d1e.size()==2 && d1e.begin()->first==d1e.rbegin()->second)) + if(BAD) + return vector(1, -1); + + vector used(n*n); + for (int i = 0; i < x.size(); ++i) + used[x[i] * n + y[i]] = true; + + vector ans; + for(int v=0; v +double start_time; string timer() + { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); } +template ostream& operator<<(ostream& os, const vector& v) + { os << "{ "; + for(typename vector::const_iterator it=v.begin(); it!=v.end(); ++it) + os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; } +void verify_case(const vector & Expected, const vector & Received) { + bool ok = (Expected == Received); + if(ok) cerr << "PASSED" << timer() << endl; else { cerr << "FAILED" << timer() << endl; + cerr << "\to: " << Expected << endl << "\tx: " << Received << endl; } } +#define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock(); +#define END verify_case(_, MakingRegularGraph().addEdges(n, x, y));} +int main(){ + +CASE(0) + int n = 6; + int x_[] = {0,2}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {2,3}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = {0, 1, 1, 4, 3, 5, 4, 5 }; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +CASE(1) + int n = 4; + int x_[] = {1,2,1}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {2,3,3}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = {-1 }; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +CASE(2) + int n = 3; + vector x; + vector y; + int __[] = {0, 1, 0, 2, 1, 2 }; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +CASE(3) + int n = 5; + int x_[] = {0}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {4}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = {0, 1, 1, 2, 2, 3, 3, 4 }; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +CASE(4) + int n = 5; + int x_[] = {2}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {3}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = {0, 1, 0, 2, 1, 4, 3, 4 }; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +/* +CASE(5) + int n = ; + int x_[] = ; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = ; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = ; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +CASE(6) + int n = ; + int x_[] = ; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = ; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + int __[] = ; + vector _(__, __+sizeof(__)/sizeof(*__)); +END +*/ +} +// END CUT HERE