|
| 1 | +#include <iostream> |
| 2 | +#include <sstream> |
| 3 | +#include <vector> |
| 4 | +#include <algorithm> |
| 5 | +#include <set> |
| 6 | +#include <map> |
| 7 | +#include <queue> |
| 8 | +#include <cstring> |
| 9 | +#include <climits> |
| 10 | +using namespace std; |
| 11 | +typedef pair<int,int> pi; |
| 12 | +typedef vector<int> vi; |
| 13 | +typedef vector<vi> vvi; |
| 14 | +typedef vector<string> vs; |
| 15 | +typedef vector<vs> vvs; |
| 16 | +int M[55][55]; |
| 17 | +int v[55]; |
| 18 | +int n; |
| 19 | + |
| 20 | +class Circuits { |
| 21 | + public: |
| 22 | + int dfs(int s, int c = 0){ |
| 23 | + v[s] = 1; |
| 24 | + int rc = 0; |
| 25 | + for(int i = 0; i < n; i++){ |
| 26 | + if((v[i] == 0 || v[i] == 2) && M[s][i]){ |
| 27 | + v[i] = 1; |
| 28 | + rc = max(rc, dfs(i)+M[s][i]); |
| 29 | + } |
| 30 | + } |
| 31 | + v[s] = 2; |
| 32 | + return rc+c; |
| 33 | + } |
| 34 | + |
| 35 | + int howLong(vector <string> connects, vector <string> costs) |
| 36 | + { |
| 37 | + n = costs.size(); |
| 38 | + for(int i = 0; i < n; i++) |
| 39 | + for(int j = 0; j < n; j++) |
| 40 | + M[i][j] = 0; |
| 41 | + |
| 42 | + for(int i = 0; i < n; i++){ |
| 43 | + string s = connects[i], t = costs[i]; |
| 44 | + int j, c; |
| 45 | + stringstream ss(s), st(t); |
| 46 | + while(ss>>j){ |
| 47 | + st>>c; |
| 48 | + M[i][j] = c; |
| 49 | + } |
| 50 | + } |
| 51 | + fill(v, v+55, 0); |
| 52 | + int c = 0; |
| 53 | + for(int i = 0; i < n; i++) |
| 54 | + if(v[i] == 0) c = max(c, dfs(i)); |
| 55 | + return c; |
| 56 | + } |
| 57 | + |
| 58 | +// BEGIN CUT HERE |
| 59 | + public: |
| 60 | + 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(); } |
| 61 | + private: |
| 62 | + 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(); } |
| 63 | + void verify_case(int Case, const int &Expected, const int &Received) { cerr << "Test Case #" << Case << "..."; if (Expected == Received) cerr << "PASSED" << endl; else { cerr << "FAILED" << endl; cerr << "\tExpected: \"" << Expected << '\"' << endl; cerr << "\tReceived: \"" << Received << '\"' << endl; } } |
| 64 | + void test_case_0() { string Arr0[] = {"1 2", |
| 65 | + "2", |
| 66 | + ""}; vector <string> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); string Arr1[] = {"5 3", |
| 67 | + "7", |
| 68 | + ""}; vector <string> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arg2 = 12; verify_case(0, Arg2, howLong(Arg0, Arg1)); } |
| 69 | + void test_case_1() { string Arr0[] = {"1 2 3 4 5","2 3 4 5","3 4 5","4 5","5",""}; vector <string> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); string Arr1[] = {"2 2 2 2 2","2 2 2 2","2 2 2","2 2","2",""}; vector <string> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arg2 = 10; verify_case(1, Arg2, howLong(Arg0, Arg1)); } |
| 70 | + void test_case_2() { string Arr0[] = {"1","2","3","","5","6","7",""}; vector <string> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); string Arr1[] = {"2","2","2","","3","3","3",""}; vector <string> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arg2 = 9; verify_case(2, Arg2, howLong(Arg0, Arg1)); } |
| 71 | + void test_case_3() { string Arr0[] = {"","2 3 5","4 5","5 6","7","7 8","8 9","10", |
| 72 | + "10 11 12","11","12","12",""}; vector <string> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); string Arr1[] = {"","3 2 9","2 4","6 9","3","1 2","1 2","5", |
| 73 | + "5 6 9","2","5","3",""}; vector <string> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arg2 = 22; verify_case(3, Arg2, howLong(Arg0, Arg1)); } |
| 74 | + void test_case_4() { string Arr0[] = {"","2 3","3 4 5","4 6","5 6","7","5 7",""}; vector <string> Arg0(Arr0, Arr0 + (sizeof(Arr0) / sizeof(Arr0[0]))); string Arr1[] = {"","30 50","19 6 40","12 10","35 23","8","11 20",""}; vector <string> Arg1(Arr1, Arr1 + (sizeof(Arr1) / sizeof(Arr1[0]))); int Arg2 = 105; verify_case(4, Arg2, howLong(Arg0, Arg1)); } |
| 75 | + |
| 76 | +// END CUT HERE |
| 77 | + |
| 78 | + }; |
| 79 | + |
| 80 | +// BEGIN CUT HERE |
| 81 | +int main(){ |
| 82 | + |
| 83 | + Circuits ___test; |
| 84 | + ___test.run_test(-1); |
| 85 | + |
| 86 | +} |
| 87 | +// END CUT HERE |
0 commit comments