LeetCode
https://leetcode.cn
1331. 数组序号转换
给你一个整数数组 arr ，请你将数组中的每个元素替换为它们排序后的序号。
序号代表了一个元素有多大。序号编号的规则如下：
- 序号从 1 开始编号。
- 一个元素越大，那么序号越大。如果两个元素相等，那么它们的序号相同。
- 每个数字的序号都应该尽可能地小。
示例 1：
1
2
3
输入：arr = [40,10,20,30]
输出：[4,1,2,3]
解释：40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。
示例 2：
1
2
3
输入：arr = [100,100,100]
输出：[1,1,1]
解释：所有元素有相同的序号。
示例 3：
1
2
输入：arr = [37,12,28,9,100,56,80,5,12]
输出：[5,3,4,2,8,6,7,1,3]
提示：
0 <= arr.length <= 10^5-10^9 <= arr[i] <= 10^9
答案：
C++
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
#include <vector> #include <numeric> #include <algorithm> #include <iostream> using namespace std; vector<int> arrayRankTransform(vector<int> &arr) { vector<int> idx(arr.size()); iota(idx.begin(), idx.end(), 0); sort(idx.begin(), idx.end(), [r = arr.data()](int a, int b){return r[a] < r[b];}); int rk = 0, rear = INT_MIN; for (auto x: idx) { if (arr[x] > rear) ++rk; rear = arr[x]; arr[x] = rk; } return arr; } int main() { vector<int> arr = {37,12,28,9,100,56,80,5,12}; arr = arrayRankTransform(arr); std::cout << '['; for (auto &x: arr) std::cout << x << ','; std::cout << '\b' << ']'; return 0; }
难点：
iota()函数构建连续的值，用于构建索引，位于numeric头文件；区别itoa()把一个整数转换为字符串sort()位于algorithmn头文件中lambda 函数基本格式
[capture](parameters) -> return_type { /* ... */ }- [capture]:
[]内为外部变量的传递方式，值、引用等，如下
1 2 3 4 5 6 7 8
```c++ [] //表示的是在lambda定义之前的域，对外部参数的调用； [=] //表示外部参数直接传值 [&] //表示外部参数传引用，可修改值。当默认捕获符是 & 时，后继的简单捕获符必须不以 & 开始。而当默认捕获符是 = 时，后继的简单捕获符必须以 & 开始。 [x, &y] //x is captured by value, y is captured by reference [&, x] //x is explicitly captured by value. Other variables will be captured by reference [=, &z] //z is explicitly captured by reference. Other variables will be captured by value ```
(parameters):
()内为形参，和普通函数的形参一样。-> return_type:
->后面为 lambda 函数的返回类型，如-> int、-> string等。一般情况下，编译器推出 lambda 函数的返回值，所以这部分可以省略不写。{ /* … */ }:
{}内为函数主体，和普通函数一样。定义函数指针一般用
auto或std::function<return_type(parameters_type)>
- [capture]:
用一个 vector 对另一个 vector 排序，把 comp 条件设置为另一个 vector 即可