#include #include #include #include #include "HungryCowsEasy.cpp" using namespace std; int main(int argc, char **argv) { int i; string s; istringstream ss; class HungryCowsEasy TheClass; vector retval; vector cowPositions; vector barnPositions; if (argc != 2) { fprintf(stderr, "usage: a.out num (or - to enter values on stdin)\n"); exit(1); } if ((string) argv[1] == "-") { if (!getline(cin, s)) return 1; ss.clear(); ss.str(s); while (ss >> i) cowPositions.push_back(i); if (!getline(cin, s)) return 1; ss.clear(); ss.str(s); while (ss >> i) barnPositions.push_back(i); } else { if (atoi(argv[1]) == 0) { cowPositions.push_back(0); barnPositions.push_back(-5); barnPositions.push_back(5); } if (atoi(argv[1]) == 1) { cowPositions.push_back(7); cowPositions.push_back(7); barnPositions.push_back(7); barnPositions.push_back(10000); } if (atoi(argv[1]) == 2) { cowPositions.push_back(2); cowPositions.push_back(6); cowPositions.push_back(0); cowPositions.push_back(4); cowPositions.push_back(8); barnPositions.push_back(3); barnPositions.push_back(-1); barnPositions.push_back(5); barnPositions.push_back(1); barnPositions.push_back(7); } if (atoi(argv[1]) == 3) { cowPositions.push_back(2); barnPositions.push_back(0); barnPositions.push_back(1); barnPositions.push_back(2); barnPositions.push_back(3); barnPositions.push_back(4); } } retval = TheClass.findFood(cowPositions, barnPositions); for (i = 0; i < (int) retval.size(); i++) { if (i != 0) cout << " "; cout << retval[i]; } cout << endl; exit(0); }