ADDED SRM/611-U/1A.cpp Index: SRM/611-U/1A.cpp ================================================================== --- SRM/611-U/1A.cpp +++ SRM/611-U/1A.cpp @@ -0,0 +1,143 @@ +#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; + +LL gcd(LL a, LL b) +{ + while(a) + swap(a, b%=a); + return b; +} + +LL lcm(LL a, LL b) +{ + return a/gcd(a,b)*b; +} + +class LCMSet { public: + string equal(vector A, vector B) + { + return is_equal(A,B) ? "Equal" : "Not equal"; + } + + bool is_equal(const vector& A, const vector& B) + { + return can_make(A, B) && can_make(B, A); + } + + bool can_make(const vector& A, const vector& B) + { + return count_if(A.begin(), A.end(), [&](int a){return can_make(a, B);}) == A.size(); + } + + bool can_make(int a, const vector& B) + { + LL cur = 1; + for(int b: B) { + if(lcm(lcm(cur, b), a) == a) + cur = lcm(cur, b); + } + return a == cur; + } +}; + +// 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 string& Expected, const string& 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(_, LCMSet().equal(A, B));} +int main(){ + +CASE(0) + int A_[] = {2,3,4,12}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {2,3,4,6}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Equal"; +END +CASE(1) + int A_[] = {4,9}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {6,36}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Not equal"; +END +CASE(2) + int A_[] = {2,3,5,7,14,105}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {2,3,5,6,7,30,35}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Equal"; +END +CASE(3) + int A_[] = {2,3,5,7,14,105}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {2,3,5,6,7,30,36}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Not equal"; +END +CASE(4) + int A_[] = {2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Equal"; +END +CASE(5) + int A_[] = {999999999,1953125,512,1000000000}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {999999999,1953125,512}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Equal"; +END +CASE(6) + int A_[] = {999999998,999999999,1000000000}; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = {999999999,1000000000}; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = "Not equal"; +END +/* +CASE(7) + int A_[] = ; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = ; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = ; +END +CASE(8) + int A_[] = ; + vector A(A_, A_+sizeof(A_)/sizeof(*A_)); + int B_[] = ; + vector B(B_, B_+sizeof(B_)/sizeof(*B_)); + string _ = ; +END +*/ +} +// END CUT HERE ADDED SRM/611-U/1B-U.cpp Index: SRM/611-U/1B-U.cpp ================================================================== --- SRM/611-U/1B-U.cpp +++ SRM/611-U/1B-U.cpp @@ -0,0 +1,178 @@ +#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; + +struct UnionFind +{ + vector uf, sz; + int nc; + + UnionFind(int N): uf(N), sz(N,1), nc(N) + { for(int i=0; i= sz[b] ) swap(a, b); + uf[a] = b; + sz[b] += sz[a]; + --nc; + } + return (a!=b); + } +}; + +class Egalitarianism2 { public: + int N; + vector> d2table; + + typedef tuple edge; + + double minStdev(vector x, vector y) + { + N = x.size(); + d2table.resize(N, vector(N)); + for(int v=0; v edges; + for(int v=0; v& edges, double sx, double sx2) + { + if(uf.size() == 1) + return sx2/(N-1) - sx*sx/(N-1)/(N-1); + if(i==edges.size()) + return 1e+300; + + vector cand; + int e=i; + for(; e(edges[i])==get<0>(edges[e]); ++e) + if(uf.Find(get<1>(edges[e])) != uf.Find(get<2>(edges[e]))) + cand.push_back(e); + + if(cand.empty()) + return rec(e, uf, edges, sx, sx2); + + double best = 1e+300; + for(int e: cand) { + UnionFind uf2 = uf; + LL d = get<0>(edges[e]); + int u = get<1>(edges[e]); + int v = get<2>(edges[e]); + uf2.Union(u, v); + best = min(best, rec(e+1, uf2, edges, sx+sqrt(d), sx2+d)); + } + 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 double& Expected, const double& Received) { + bool ok = (abs(Expected - Received) < 1e-9); + 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(_, Egalitarianism2().minStdev(x, y));} +int main(){ + +CASE(0) + int x_[] = {0,0,1,1}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {0,1,0,1}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 0.0; +END +CASE(1) + int x_[] = {0,0,0}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {0,9,10}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 0.5; +END +CASE(2) + int x_[] = {12,46,81,56}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {0,45,2,67}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 6.102799971320872; +END +CASE(3) + int x_[] = {0,0,0,0,0,0,0}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {0,2,3,9,10,15,16}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 0.9428090415820617; +END +CASE(4) + int x_[] = {167053, 536770, -590401, 507047, 350178, -274523, -584679, -766795, -664177, 267757, -291856, -765547, 604801, -682922, -404590, 468001, 607925, 503849, -499699, -798637}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {-12396, -66098, -56843, 20270, 81510, -23294, 10423, 24007, -24343, -21587, -6318, -7396, -68622, 56304, -85680, -14890, -38373, -25477, -38240, 11736}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 40056.95946451678; +END +CASE(5) + int x_[] = {-306880, 169480, -558404, -193925, 654444, -300247, -456420, -119436, -620920, -470018, -914272, -691256, -49418, -21054, 603373, -23656, 891691, 258986, -453793, -782940}; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = {-77318, -632629, -344942, -361706, 191982, 349424, 676536, 166124, 291342, -268968, 188262, -537953, -70432, 156803, 166174, 345128, 58614, -671747, 508265, 92324}; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = 36879.1512763429; +END +/* +CASE(6) + int x_[] = ; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = ; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = ; +END +CASE(7) + int x_[] = ; + vector x(x_, x_+sizeof(x_)/sizeof(*x_)); + int y_[] = ; + vector y(y_, y_+sizeof(y_)/sizeof(*y_)); + double _ = ; +END +*/ +} +// END CUT HERE