华为OD机试真题 C++ 实现【积木最远距离】2022.11 Q4

2023-03-24 08:42:55 浏览数 (1)

题目描述

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

输入描述
第一行输入为N,表示小华排成一排的积木上数字

输出描述
相同数字的积木的位置最远距离;如果所有积木数字都不相同,请返回-1.

示例1 输入输出示例仅供调试,后台判题数据一般不包含示例
输入
5
1
2
3
1
4

输出
3

说明:
共有5个积木,第1个积木和第4个积木数字相同,其距离为3.

示例2 输入输出示例仅供调试,后台判题数据一般不包含示例
输入
2
1
2

输出
-1

说明
一共有2个积木,没有积木数字相同,返回-1.

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

思路

代码语言:javascript复制
本题是简单题,思路比较简单,首先用maxDistance记录数字两个相同数字的最远距离,初始化为-1。
接着遍历数组nums,使用一个unordered_map<long, long> cntMap;用来记录数组中第一次出现某个数字的位置,即key为数字本身,value为所在数组索引下标。
在遍历过程中,
1、在cntMap中查找是否有nums[i],如果没有出现过则将{nums[i], i}添加到cntMap中
2、如果cntMap中已经存在nums[i],判断当前索引下标i减去nums[i]第一次出现的下标是否大于maxDistance,如果大于则:
maxDistance = i - iter->second;
遍历完数组nums,返回maxDistance即可。
注意两点:
A、如果nums[i]中不存在两个相同的数,则返回-1,cntMap的长度和nums长度相同。
B、由于题目备注中的积木上的数字最大为10^9,积木长度最大为10^5,超过了int的最大范围,需要使用long表示数字。

C 代码实现

代码语言:javascript复制
#include <iostream>
#include <vector>
#include <string>
#include <unordered_map>

using namespace std;

int GetFurthestDistacneOfBlocks(vector<long>& nums)
{
    int maxDistance = -1;
    // key为数字的值,value为第一次出现该数字的数字下标索引
    unordered_map<long, long> cntMap;
    for (long i = 0; i < nums.size(); i  ) {
        auto iter = cntMap.find(nums[i]);
        if (iter == cntMap.end()) {
            cntMap.insert({ std::pair<int, int>(nums[i], i)});
        } else {
            // 之前已经有过该数字,则更新相同数字的最远积木距离
            if (i - iter->second > maxDistance) {
                maxDistance = i - iter->second;
            }
        }
    }

    return maxDistance;
}

int main()
{
    long N;
    while (cin >> N) {
        vector<long> nums;
        long num;
        for (long i = 0; i < N; i  ) {
            cin >> num;
            nums.push_back(num);
        }
        cout << GetFurthestDistacneOfBlocks(nums) << std::endl;
    }

    return 0;
}

0 人点赞