无人机方阵
作者: Turbo
时间限制: 1s
章节: 课程设计
问题描述
在 「挑战赛」 开幕式的压轴节目 「无人机方阵」中,每一架无人机展示一种灯光颜色。 无人机方阵通过两种操作进行颜色图案变换:
(1) 调整无人机的位置布局
(2) 切换无人机展示的灯光颜色
给定两个大小均为 N*M 的二维数组 source 和 target 表示无人机方阵表演的两种颜色图案,由于无人机切换灯光颜色的耗能很大,请返回从 source 到 target 最少需要多少架无人机切换灯光颜色。
注意: 调整无人机的位置布局时无人机的位置可以随意变动。
示例 1:
输入:
2 2
1 3
5 4
3 1
6 5
输出:1
解释:
第一行输入2 2表示方阵为2行2列,后面跟着的两行为source,再两行为target。
最佳方案为
将 [0,1] 处的无人机移动至 [0,0] 处;
将 [0,0] 处的无人机移动至 [0,1] 处;
将 [1,0] 处的无人机移动至 [1,1] 处;
将 [1,1] 处的无人机移动至 [1,0] 处,其灯光颜色切换为颜色编号为 6 的灯光;
因此从source 到 target 所需要的最少灯光切换次数为 1。
示例 2:
输入:
2 3
1 2 3
3 4 5
1 3 5
2 3 4
输出:0
解释:
仅需调整无人机的位置布局,便可完成图案切换。因此不需要无人机切换颜色
输入说明
输入若干行:
第一行为两个整数n和m,n代表二维数组的行数,m代表二维数组的列数。
而后n行,每行输入m个整数,代表source数组的元素。
再n行,每行输入m个整数,代表target数组的元素。
提示:
1 <= n, m <=100
1 <= source[i][j], target[i][j] <=10^4
输出说明
输出一个整数表示结果。
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
int min_color_switches(int n, int m, const vector<vector<int>>& source, const vector<vector<int>>& target)
{
unordered_map<int, int> source_count;
unordered_map<int, int> target_count;
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < m; ++j)
{
source_count[source[i][j]]++;
}
}
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < m; ++j)
{
target_count[target[i][j]]++;
}
}
int switches = 0;
for (const auto& [color, count] : target_count)
{
if (source_count.find(color) != source_count.end())
{
if (count > source_count[color])
{
switches += count - source_count[color];
}
}
else
{
switches += count;
}
}
return switches;
}
int main()
{
int n, m;
cin >> n >> m;
vector<vector<int>> source(n, vector<int>(m));
vector<vector<int>> target(n, vector<int>(m));
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < m; ++j)
{
cin >> source[i][j];
}
}
for (int i = 0; i < n; ++i)
{
for (int j = 0; j < m; ++j)
{
cin >> target[i][j];
}
}
cout << min_color_switches(n, m, source, target) << endl;
return 0;
}