|
| 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 | +#include <cstdio> |
| 11 | +#include <numeric> |
| 12 | +using namespace std; |
| 13 | +typedef pair<int,int> pi; |
| 14 | +typedef set<int> si; |
| 15 | +typedef vector<int> vi; |
| 16 | +typedef vector<vi> vvi; |
| 17 | +typedef vector<string> vs; |
| 18 | +typedef long long ll; |
| 19 | +#define sz size() |
| 20 | +#define mp make_pair |
| 21 | +#define pb push_back |
| 22 | +#define inf 1000000000 |
| 23 | +#define ri(a, b) for(int i=((int)(a)); i < ((int)(b)); i++) // i -> [a, b) |
| 24 | +#define rie(a, b) for(int i=((int)(a)); i <= ((int)(b)); i++) // i -> [a, b] |
| 25 | +#define rj(a, b) for(int j=((int)(a)); j < ((int)(b)); j++) // j -> [a, b) |
| 26 | +#define rje(a, b) for(int j=((int)(a)); j <= ((int)(b)); j++) // j -> [a, b] |
| 27 | +#define rk(a, b) for(int k=((int)(a)); k < ((int)(b)); k++) // k -> [a, b) |
| 28 | +#define rke(a, b) for(int k=((int)(a)); k <= ((int)(b)); k++) // k -> [a, b] |
| 29 | +#define fi(b) for(int i=0; i < ((int)(b)); i++) // i -> [0, b) |
| 30 | +#define fie(b) for(int i=0; i <= ((int)(b)); i++) // i -> [0, b] |
| 31 | +#define fj(b) for(int j=0; j < ((int)(b)); j++) // j -> [0, b) |
| 32 | +#define fje(b) for(int j=0; j <= ((int)(b)); j++) // j -> [0, b] |
| 33 | +#define fk(b) for(int k=0; k < ((int)(b)); k++) // k -> [0, b) |
| 34 | +#define fke(b) for(int k=0; k < ((int)(b)); k++) // k -> [0, b] |
| 35 | +#define fle(b) for(int l=0; l <= ((int)(b)); l++) // l -> [0, b] |
| 36 | +#define vaild(x, y) ( 0 <= (x) && (x) < n && 0 <= (y) && (y) < m ) |
| 37 | +ll n, m; |
| 38 | + |
| 39 | +class HyperKnight { |
| 40 | + public: |
| 41 | + ll countCells(int aa, int bb, int numRows, int numColumns, int k) { |
| 42 | + ll a = aa, b = bb; |
| 43 | + n = numRows; m = numColumns; |
| 44 | + if(a < b) swap(a, b); |
| 45 | + ll ret = 0; |
| 46 | + switch(k){ |
| 47 | + case 2 : |
| 48 | + ret = 4*b*b; |
| 49 | + break; |
| 50 | + case 3 : |
| 51 | + ret = 8*b*(a-b); |
| 52 | + break; |
| 53 | + case 4 : |
| 54 | + ret = 2*b*(m+n-4*a) + 4*(a-b)*(a-b); |
| 55 | + break; |
| 56 | + case 6 : |
| 57 | + ret = 2*(a-b)*(m+n-4*a); |
| 58 | + break; |
| 59 | + case 8 : |
| 60 | + ret = (m-2*a)*(n-2*a); |
| 61 | + break; |
| 62 | + } |
| 63 | + return ret; |
| 64 | + } |
| 65 | + |
| 66 | +// BEGIN CUT HERE |
| 67 | + public: |
| 68 | + 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(); } |
| 69 | + private: |
| 70 | + 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(); } |
| 71 | + void verify_case(int Case, const long long &Expected, const long long &Received) { cerr << "Test Case #" << Case << "..."; if (Expected == Received) cerr << "PASSED" << endl; else { cerr << "FAILED" << endl; cerr << "\tExpected: \"" << Expected << '\"' << endl; cerr << "\tReceived: \"" << Received << '\"' << endl; } } |
| 72 | + void test_case_0() { int Arg0 = 2; int Arg1 = 1; int Arg2 = 8; int Arg3 = 8; int Arg4 = 4; long long Arg5 = 20LL; verify_case(0, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 73 | + void test_case_1() { int Arg0 = 3; int Arg1 = 2; int Arg2 = 8; int Arg3 = 8; int Arg4 = 2; long long Arg5 = 16LL; verify_case(1, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 74 | + void test_case_2() { int Arg0 = 1; int Arg1 = 3; int Arg2 = 7; int Arg3 = 11; int Arg4 = 0; long long Arg5 = 0LL; verify_case(2, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 75 | + void test_case_3() { int Arg0 = 3; int Arg1 = 2; int Arg2 = 10; int Arg3 = 20; int Arg4 = 8; long long Arg5 = 56LL; verify_case(3, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 76 | + void test_case_4() { int Arg0 = 1; int Arg1 = 4; int Arg2 = 100; int Arg3 = 10; int Arg4 = 6; long long Arg5 = 564LL; verify_case(4, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 77 | + void test_case_5() { int Arg0 = 2; int Arg1 = 3; int Arg2 = 1000000000; int Arg3 = 1000000000; int Arg4 = 8; long long Arg5 = 999999988000000036LL; verify_case(5, Arg5, countCells(Arg0, Arg1, Arg2, Arg3, Arg4)); } |
| 78 | + |
| 79 | +// END CUT HERE |
| 80 | + |
| 81 | +}; |
| 82 | + |
| 83 | + // BEGIN CUT HERE |
| 84 | + int main() |
| 85 | + { |
| 86 | + HyperKnight ___test; |
| 87 | + ___test.run_test(-1); |
| 88 | + } |
| 89 | + // END CUT HERE |
| 90 | + |
0 commit comments