2024华为OD试题及答案-A021-积木最远距离

题目描述

小华和小薇一起通过玩积木游戏学习数学。
他们有很多积木，每个积木块上都有一个数字，积木块上的数字可能相同。
小华随机拿一些积木挨着排成一排，请小薇找到这排积木中数字相同且所处位置最远的2块积木块，计算他们的距离，小薇请你帮忙替她解决这个问题。

输入描述

第一行输入为N，表示小华排成一排的积木总数。
接下来N行每行一个数字，表示小华排成一排的积木上数字。

输出描述

相同数字的积木的位置最远距离；如果所有积木数字都不相同，请返回-1。

备注

• 0<=积木上的数字<10^9
• 1<=积木长度<=10^5

用例

输入	5 1 2 3 1 4
输出	3
说明	共有5个积木，第1个积木和第4个积木数字相同，其距离为3。

输入	2 1 2
输出	-1
说明	一共有两个积木，没有积木数字相同，返回-1。

题目解析

问题描述

小红和小蓝通过玩积木游戏学习数学。他们有很多积木，每个积木块上都有一个数字，积木块上的数字可能相同。小红想知道这些积木中相同数字的积木块之间相距最远的两个积木块的距离。如果所有积木块上的数字都不相同，返回-1。

输入描述

1. 第一行输入为一个整数N，表示小朋友排成一排的积木总数。
2. 接下来的N行，每行一个整数，表示积木上的数字。

输出描述

输出相同数字的积木之间最远的距离；如果所有积木数字都不相同，返回-1。

示例

输入：

5
1
2
3
1
4

输出：

3

解释：相同数字为1的积木块之间的最大距离为3。

解题思路

1. 记录位置：使用字典记录每个数字第一次和最后一次出现的位置。
2. 计算距离：遍历字典，计算相同数字的积木块之间的最大距离。
3. 返回结果：如果存在相同数字的积木块，返回最大距离；否则返回-1。

C++代码实现

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>

using namespace std;

int main() {
    int N;
    cin >> N;
    vector<int> blocks(N);
    
    for (int i = 0; i < N; ++i) {
        cin >> blocks[i];
    }
    
    unordered_map<int, pair<int, int>> positions;
    
    for (int i = 0; i < N; ++i) {
        if (positions.find(blocks[i]) == positions.end()) {
            positions[blocks[i]] = {i, i};
        } else {
            positions[blocks[i]].second = i;
        }
    }
    
    int max_distance = -1;
    
    for (const auto& entry : positions) {
        if (entry.second.first != entry.second.second) {
            max_distance = max(max_distance, entry.second.second - entry.second.first);
        }
    }
    
    cout << max_distance << endl;
    
    return 0;
}

代码说明

1. 读取输入：读取积木总数和每个积木上的数字。
2. 记录位置：使用unordered_map记录每个数字第一次和最后一次出现的位置。
3. 计算距离：遍历unordered_map，计算相同数字的积木块之间的最大距离。
4. 返回结果：输出最大距离，如果没有相同数字的积木块，则返回-1。

这个程序能够计算出积木中相同数字之间的最大距离，帮助小红和小蓝解决他们的问题。

python代码实现

def find_max_distance(blocks):
    positions = {}

    for i, num in enumerate(blocks):
        if num not in positions:
            positions[num] = (i, i)
        else:
            positions[num] = (positions[num][0], i)
    
    max_distance = -1

    for start, end in positions.values():
        if start != end:
            max_distance = max(max_distance, end - start)
    
    return max_distance

# 读取输入
N = int(input())
blocks = [int(input()) for _ in range(N)]

# 计算并输出最大距离
print(find_max_distance(blocks))

代码说明

1. find_max_distance函数：实现了记录位置和计算距离的逻辑。使用字典记录每个数字第一次和最后一次出现的位置。
2. 读取输入：读取积木总数和每个积木上的数字。
3. 计算并输出最大距离：调用find_max_distance函数计算最大距离，并输出结果。

这个Python程序可以计算出积木中相同数字之间的最大距离，解决题目中的问题。你可以在Python环境中运行这个程序来实现该功能。