【东华大学oj】无人机方阵

无人机方阵

作者: 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;
}