제출 #148279

#제출 시각아이디문제언어결과실행 시간메모리
148279ksun48 (#201)백점을 받아랏! (FXCUP4_hundred)C++17
100 / 100
8 ms384 KiB
#include "hundred.h"
#include <bits/stdc++.h>

std::string GetHundredPoints(int A, int B, int C) {
	using namespace std;
	assert(A + B + C == 100);
	string str = string(A, 'A') + string(B, 'B') + string(C, 'C');
	if(A == 100 || B == 100 || C == 100) return str;
	
	string ans(A + B + C, 'A');
	int value = Mark(str);
	auto f = [&](int a, int b){
		string new_str = str;
		swap(new_str[a], new_str[b]);
		int diff = (Mark(new_str) - value) * (str[b] - str[a]) % 3;
		if(diff < 0) diff += 3;
		ans[b] = ((ans[a] - 'A' + diff) % 3) + 'A';
	};
	if(A && B){
		for(int i = A; i < A+B+C; i++) f(0, i);
		for(int i = 1; i < A; i++) f(A, i);
	} else {
		for(int i = A+B; i < A+B+C; i++) f(0, i);
		for(int i = 1; i < A+B; i++) f(A+B, i);
	}
	for(int i = 0; i < 3; i++){
		map<char	, int> cnt;
		for(char x : ans) cnt[x] += 1;
		if(cnt['A'] == A && cnt['B'] == B && cnt['C'] == C) return ans;
		for(char& x : ans) x = 'A' + ((x - 'A' + 1) % 3);
	}
	assert(false);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...