C++刷题技巧

零、前言

由于C++强大的STL，本人即便将Golang作为主力开发语言，仍对用C++刷题情有独钟。本文主要用于记录使用C++刷题时，比较关键但有点难记的技巧。

一、语法技巧

0、创建优先队列

• C++中的优先队列默认是最大堆，用如下方式创建最小堆：

  #include <queue>
  priority_queue<int, vector<int>, greater<int>> minimum_heap; // 最小堆
  priority_queue<int> maximum_heap; // 默认最大堆
  

  相关题目：数组中的第K个最大元素。
• 自定义优先队列的比较函数：

  auto cmp = [](const pair<int, int> &a, const pair<int, int> &b) { return a.first < b.first; };
  priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> window_nums(cmp);
  

  相关题目：滑动窗口最大值.

1、生成服从均匀分布的随机数

#include <random>
default_random_engine e;
uniform_int_distribution<int> u(begin, end);
int r = u(e); // r服从均匀分布，r∈[begin, end]

2、使用Lambda表达式

[](int a, int b) -> bool { return a < b; } // 显式指定返回类型
[](int a, int b) { return a < b; } // 隐式指定返回类型
[nums](int a, int b) { return nums[a] < nums[b]; } // 显式值捕获外部变量
[&nums](int a, int b) { return nums[a] < nums[b]; } // 显式引用捕获外部变量
[=](int a, int b) { return nums[a] < nums[b]; } // 隐式值捕获外部变量
[&](int a, int b) { return nums[a] < nums[b]; } // 隐式引用捕获外部变量

3、自定义排序

0）准备数据

struct sort_node {
    int val;
};
vector<sort_node> sort_arr(len);
for (int i = 0; i < len; i++) {
    sort_arr[i] = { u(e) }; // 使用前述生成服从均匀分布的随机数的方法
}

1）使用谓词函数

#include <algorithm>
bool sort_func(const sort_node &node1, const sort_node &node2) {
    return node1.val < node2.val;
}
sort(sort_arr.begin(), sort_arr.end(), sort_func);

2）使用Lambda表达式

#include <algorithm>
sort(sort_arr.begin(), sort_arr.end(), [](const sort_node &node1, const sort_node &node2) {
    return node1.val < node2.val;
});

4、交换数据

#include <algorithm>
swap(a, b); // 交换两个变量中的值
swap(arr[a], arr[b]); // 交换容器中两个下标的值
iter_swap(arr.begin()+i, arr.begin()+j); // 交换两个迭代器指向的值

5、数字和字符串互相转换

#include <string>
to_string(i); // 数字转换成字符串
// 字符串转换成数字
stoi(i); // int
stol(l); // long
stoll(ll); // long long
stoul(ul); // unsigned long
stoull(ull); // unsigned long long
stof(f); // float
stod(d); // double
stold(ld); // long double

6、处理string对象中的字符

// cctype头文件中的函数
isalnum(c); // 当c是字母或数字时为真
isalpha(c); // 当c是字母时为真
iscntrl(c); // 当c是控制字符时为真
isdigit(c); // 当c是数字时为真
isgraph(c); // 当c不是空格但可打印时为真
islower(c); // 当c是小写字母时为真
isprint(c); // 当c是可打印字符时为真
ispunct(c); // 当c是标点符号时为真
isspace(c); // 当c是空白时为真
isupper(c); // 当c是大写字母时为真
isxdigit(c); // 当c是十六进制数字时为真
tolower(c); // 输出对应的小写字母
toupper(c); // 输出对应的大写字母

二、算法技巧

0、逆序打印链表

void print_values_in_reverse(ListNode *head) {
    if (head != nullptr) {
        print_values_in_reverse(head->next);
        cout << head->val << endl;
    }
}

相关题目：回文链表（此题的递归解法会用到上述算法技巧，但此题还有空间复杂度更优的快慢指针解法）。

1、迭代遍历二叉树

0）二叉树节点定义

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

1）前序遍历

vector<int> preorderTraversal(TreeNode* root) {
    vector<int> values;
    stack<TreeNode*> nodes;
    TreeNode *curNode = root;
    while (curNode != nullptr || !nodes.empty()) {
        while (curNode != nullptr) {
            nodes.push(curNode);
            values.push_back(curNode->val);
            curNode = curNode->left;
        }
        curNode = nodes.top();
        nodes.pop();
        curNode = curNode->right;
    }
    return values;
}

2）中序遍历

vector<int> inorderTraversal(TreeNode* root) {
    vector<int> values;
    stack<TreeNode*> nodes;
    TreeNode *curNode = root;
    while (curNode != nullptr || !nodes.empty()) {
        while (curNode != nullptr) {
            nodes.push(curNode);
            curNode = curNode->left;
        }
        curNode = nodes.top();
        nodes.pop();
        values.push_back(curNode->val);
        curNode = curNode->right;
    }
    return values;
}

相关题目：二叉搜索树的第k大节点（此题采用了中序遍历和倒序遍历结合的方法）。

3）后序遍历

vector<int> postorderTraversal(TreeNode* root) {
    vector<int> values;
    stack<TreeNode*> nodes;
    TreeNode *prevNode = nullptr, *curNode = root;
    while (curNode != nullptr || !nodes.empty()) {
        while (curNode != nullptr) {
            nodes.push(curNode);
            curNode = curNode->left;
        }
        curNode = nodes.top();
        if (curNode->right == nullptr || curNode->right == prevNode) {
            nodes.pop();
            values.push_back(curNode->val);
            prevNode = curNode;
            curNode = nullptr;
        } else curNode = curNode->right;
    }
    return values;
}

三、结语

下图是某API参考文档网站总结的C++的STL中的容器的成员函数：