AfanasevGad4

solution.cpp (2552B)

---

 #include <iostream>
 #include <vector>
 #include <algorithm>
 #include <random>
 #include <string>
 #include <ctime>
 
 class Guesser {
 private:
 	std::vector<std::string> possible_numbers;
 
 	bool isValidNumber(const std::string& num) {
 		for (int i = 0; i < 4; i++) {
 			for (int j = i + 1; j < 4; j++) {
 				if (num[i] == num[j]) return false;
 			}
 		}
 		return true;
 	}
 
 	void calculateBullsCows(const std::string& num1, const std::string& num2, int& bulls, int& cows) {
 		bulls = 0;
 		cows = 0;
 
 		for (int i = 0; i < 4; i++) {
 			if (num1[i] == num2[i]) {
 				bulls++;
 			}
 			else if (num2.find(num1[i]) != std::string::npos) {
 				cows++;
 			}
 		}
 	}
 
 public:
 	Guesser() {
 
 		for (int i = 0; i <= 9999; i++) {
 			std::string num = std::to_string(i);
 
 			while (num.length() < 4) {
 				num = "0" + num;
 			}
 			if (isValidNumber(num)) {
 				possible_numbers.push_back(num);
 			}
 		}
 
 		std::shuffle(possible_numbers.begin(), possible_numbers.end(),
 			std::default_random_engine(time(0)));
 	}
 
 	std::string makeGuess(int attempt) {
 		if (possible_numbers.empty()) return "";
 
 		if (attempt == 1) {
 			return "0123";
 		}
 
 		if (attempt == 2) {
 			return "6789";
 		}
 
 		return possible_numbers[0];
 	}
 
 	void processResult(const std::string& guess, int bulls, int cows) {
 		std::vector<std::string> new_list;
 
 		for (const auto& num : possible_numbers) {
 			int b, c;
 			calculateBullsCows(num, guess, b, c);
 			if (b == bulls && c == cows) {
 				new_list.push_back(num);
 			}
 		}
 
 		possible_numbers = new_list;
 
 		if (!possible_numbers.empty()) {
 			std::shuffle(possible_numbers.begin(), possible_numbers.end(),
 				std::default_random_engine(time(0)));
 		}
 	}
 
 	bool hasGuesses() const {
 		return !possible_numbers.empty();
 	}
 };
 
 int main() {
 
 	Guesser g;
 	int attempt = 0;
 
 	while (g.hasGuesses()) {
 		attempt++;
 		std::string guess = g.makeGuess(attempt);
 
 		#ifdef TESTER
 		std::cout << guess << std::endl;
 		#else
 		std::cout << "\nTry " << attempt << ": " << guess << std::endl;
 		#endif
 
 		#ifndef TESTER
 		std::cout << "Bulls: ";
 		#endif
 		int bulls;
 		std::cin >> bulls;
 
 		if (bulls == 4) {
 			#ifdef TESTER
 			std::cout << "Cows: 0" << std::endl << std::endl;
 			std::cout << "Steps amount: " << attempt << std::endl;
 			#else
 			std::cout << "0" << std::endl << std::endl;
 			break;
 		}
 
 		#ifndef TESTER
 		std::cout << "Cows: ";
 		#endif
 		int cows;
 		std::cin >> cows;
 
 		g.processResult(guess, bulls, cows);
 
 		if (!g.hasGuesses()) {
 			std::cout << std::endl <<  "Unreal number. And Afanasev is gad." << std::endl;
 		}
 	}
 
 	return 0;
 }