【CCF CSP 201909-3 字符画 C++ 100分】

 

试题编号：	201909-3
试题名称：	字符画
时间限制：	5.0s
内存限制：	512.0MB

题意描述模糊，坑题。

思路：正常模拟，核心函数是calc_avg_rgb()，逻辑理顺即可。

 

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;

int m, n, p, q, num_pixel_blocks;
vector <string> vec_pixels;
vector <int> current_state(3, 0), zero_rgb(3, 0);
vector <vector <int> > vec_avg_block_rgb;
string res = "", reset_sequence = "\\x1B\\x5B\\x30\\x6D";

void get_input() {
	string tmp_str, ipt;
	cin >> m >> n;
	cin >> p >> q;
	num_pixel_blocks = (m * n) / (p * q);
	for (int i = 0; i < num_pixel_blocks; i++) {
		vec_avg_block_rgb.push_back(zero_rgb);
	}
	for (int j = 0; j < m * n; j++) {
		tmp_str = "";
		cin >> ipt;
		vec_pixels.push_back(ipt.substr(1, ipt.length()));
		if (vec_pixels.back().length() == 3) {
			for (auto ch : vec_pixels.back()) {
				tmp_str += ch;
				tmp_str += ch;
			}
			vec_pixels.back() = tmp_str;
		}
		if (vec_pixels.back().length() == 1) {
			for (int k = 0; k < 5; k++) {
				vec_pixels.back() += vec_pixels.back()[0];
			}
		}
	}
}

void calc_avg_rgb() {
	vector <vector <string> > matrix_pixels(n);
	for (int i = 0; i < n; i++) {
		for (int j = i * m; j < (i + 1) * m; j++) {
			matrix_pixels[i].push_back(vec_pixels[j]);
		}
	}
	for (int row = 0; row < n; row++) {
		for (int col = 0; col < m; col++) {
			vec_avg_block_rgb[(row / q) * (m / p) + col / p][0] += stoi(matrix_pixels[row][col].substr(0, 2), nullptr, 16);
			vec_avg_block_rgb[(row / q) * (m / p) + col / p][1] += stoi(matrix_pixels[row][col].substr(2, 2), nullptr, 16);
			vec_avg_block_rgb[(row / q) * (m / p) + col / p][2] += stoi(matrix_pixels[row][col].substr(4, 2), nullptr, 16);
		}
	}
	for (int i = 0; i < (int)vec_avg_block_rgb.size(); i++) {
		for (int j = 0; j < (int)vec_avg_block_rgb[i].size(); j++) {
			vec_avg_block_rgb[i][j] = (int)(vec_avg_block_rgb[i][j] / (p * q));
		}
	}
}

string generate_sequence(vector <int> rgb) {
	char hex_value[10];
	string sequence_str = "[48;2;" + to_string(rgb[0]) + ';' + to_string(rgb[1]) + ';' + to_string(rgb[2]) + "m ";
	string sequence_str_hex = "\\x1B";
	for (char ch : sequence_str) {
		sprintf(hex_value, "%X", (int)ch);
		sequence_str_hex += ("\\x" + string(hex_value));
	}
	return sequence_str_hex;
}

/* check if the current color of terminal is set to default when one line ends */
void additional_procedure(int idx) {
	if ((idx + 1) % (m / p) == 0) {
		if (vec_avg_block_rgb[idx] != zero_rgb) {
			res += reset_sequence;
			current_state = zero_rgb;
		}
		res += "\\x0A";
	}
}

int main() {
	get_input();
	calc_avg_rgb();
	for (int i = 0; i < (int)vec_avg_block_rgb.size(); i++) {
		if (vec_avg_block_rgb[i] == current_state) {
			res += "\\x20";
			additional_procedure(i);
			continue;
		}
		if (vec_avg_block_rgb[i] == zero_rgb) {
			res += (reset_sequence + "\\x20");
			current_state = zero_rgb;
		}
		else {
			res += generate_sequence(vec_avg_block_rgb[i]);
			current_state = vec_avg_block_rgb[i];
		}
		additional_procedure(i);
	}
	cout << res;
	return 0;
}