牛客网 - 找工作神器|笔试题库|面试经验|实习招聘内推,求职就业一站解决_牛客网 (nowcoder.com)
9.小美的字符串变换
小美拿到了一个长度为n的字符串,她希望将字符串从左到右平铺成一个矩阵(先平铺第一行,然后是第二行,以此类推,矩阵有x行y列,必须保证x∗y=n,即每y个字符换行,共x行)。
该矩阵的权值定义为这个矩阵的连通块数量。小美希望最终矩阵的权值尽可能小,你能帮小美求出这个最小权值吗?
注:我们定义,上下左右四个方向相邻的相同字符是连通的。
输入描述:
输出描述:输出一个整数表示最小权值
思路:矩阵能有多种组合,求所有组合中让“权值”最小的值。“权值”=“连通块”,即整体字符中相同的字符要尽量“相邻”。或者用暴力遍历所有组合,并求每个组合的“权值”
代码实现:求每个组合的“权值”时可用DFS方法,创建v_temp、v_flag分别记录当前组合的矩阵、当前是否可入(还不是连通块)。connect函数实现DFS算法将当前元素及其上下左右相同元素递归处理。
#include <algorithm>
#include <climits>
#include <iostream>
#include <vector>
using namespace std;
void connect(vector<vector<char>>& v_temp, vector<vector<bool>>& v_flag,
int& i_temp, int& j_temp, int& i, int& j) {
int temp;
if (i_temp + 1 < i && v_flag[i_temp + 1][j_temp]&&
v_temp[i_temp][j_temp] == v_temp[i_temp + 1][j_temp]) {
v_flag[i_temp + 1][j_temp] = false;
int temp = i_temp + 1;
connect(v_temp, v_flag, temp, j_temp, i, j);
}
if (j_temp + 1 < j && v_flag[i_temp ][j_temp + 1]&&
v_temp[i_temp][j_temp] == v_temp[i_temp][j_temp + 1]) {
v_flag[i_temp ][j_temp + 1] = false;
int temp = j_temp + 1;
connect(v_temp, v_flag, i_temp, temp, i, j);
}
if (j_temp - 1 >= 0 && v_flag[i_temp ][j_temp - 1] &&
v_temp[i_temp][j_temp] == v_temp[i_temp][j_temp - 1]) {
v_flag[i_temp ][j_temp - 1] = false;
int temp = j_temp - 1;
connect(v_temp, v_flag, i_temp, temp, i, j);
}
if (i_temp - 1 >= 0 && v_flag[i_temp - 1][j_temp] &&
v_temp[i_temp][j_temp] == v_temp[i_temp - 1][j_temp]) {
v_flag[i_temp - 1][j_temp] = false;
int temp = i_temp - 1;
connect(v_temp, v_flag, temp, j_temp, i, j);
}
v_flag[i_temp][j_temp] = false;
return;
}
int main() {
int n, ret = INT_MAX;
vector<char> v;
cin >> n;
for (int i = 0; i < n; i++) {
char temp;
cin >> temp;
v.push_back(temp);
}
for (int i = 1; i <= n ; i++) {
if (n % i == 0) {
int j = n / i;
vector<vector<char>> v_temp(i, vector<char>(j));
//标记
vector<vector<bool>> v_flag(i, vector<bool>(j, true));
int index = 0;
for (int i_temp = 0; i_temp < i; i_temp++) {
for (int j_temp = 0; j_temp < j; j_temp++) {
v_temp[i_temp][j_temp] = v[index];
index++;
}
}
//记录当前矩阵的“权值”
int sum = 0;
for (int i_temp = 0; i_temp < i; i_temp++) {
for (int j_temp = 0; j_temp < j; j_temp++) {
if (v_flag[i_temp][j_temp]) {
connect(v_temp, v_flag, i_temp, j_temp, i, j);
sum++;
}
}
}
ret = min(sum, ret);
}
}
cout << ret << endl;
return 0;
}