逐层探秘：揭秘华为OD机考统一考试机试C卷「找朋友」难题

后端

2023-03-30 23:22:34

找朋友：华为机考难题的算法剖析

简介

华为OD机考统一考试机试C卷中的「找朋友」难题是一个经典算法问题，考察考生对数据结构、算法和编程语言的掌握程度。本文将深入剖析此难题，为你提供分步指南，让你自信应对类似问题。

题目概述

给定N个小朋友的身高，每个小朋友都可以看到身高比自己高的第一个小朋友，那么这个小朋友就是他的好朋友。你需要找到每个小朋友的好朋友。

算法设计

解决此难题的关键在于高效地找到每个小朋友的好朋友。我们将采用「双指针」算法：

初始化两个指针 ：left指向当前小朋友，right指向其后的第一个小朋友。
遍历小朋友队列 ：
- 如果right指向的小朋友比left指向的小朋友高，则将left指向的小朋友的好朋友设为right，并移动left指针到right位置。
- 否则，将right指针后移一位。
处理剩余小朋友 ：对于没有找到好朋友的小朋友，将他们的好朋友设为-1。

代码示例

#include <iostream>
#include <vector>

using namespace std;

int main() {
    // 获取小朋友数量和身高
    int N;
    cin >> N;
    vector<int> height(N);
    for (int i = 0; i < N; i++) {
        cin >> height[i];
    }

    // 初始化双指针
    int left = 0, right = 1;

    // 遍历小朋友队列
    while (right < N) {
        if (height[right] > height[left]) {
            friends[left] = right;
            left = right;
        }
        right++;
    }

    // 处理剩余小朋友
    for (int i = left + 1; i < N; i++) {
        friends[i] = -1;
    }

    // 输出结果
    for (int i = 0; i < N; i++) {
        cout << friends[i] << " ";
    }

    return 0;
}