AcWing算法题常用代码模板

一、基础算法

1、常用代码模板

快速排序算法模板

—— 模板题 AcWing 785. 快速排序

算法步骤：

​ ①确定分界点

​ ②调整区间

​ ③递归处理左右两段

Tips：

​ ①如果输入数据量比较大的话，用scanf输入更快，不要使用cin。

​ ②只要背诵一种万能模板即可避开所有边界问题。

​ ③考试时一般不会用到快排，面试时喜欢考察。

​ ④ #include<bits/stdc++.h>包含了目前c++所包含的所有头文件 。

​ ⑤分治一般用不到，只要掌握快速排序和归并排序就行。

void quick_sort(int q[], int l, int r)
{
   
    if (l >= r) return;

    int i = l - 1, j = r + 1, x = q[l + r >> 1];
    while (i < j)
    {
   
        do i ++ ; while (q[i] < x);
        do j -- ; while (q[j] > x);
        if (i < j) swap(q[i], q[j]);
    }
    quick_sort(q, l, j), quick_sort(q, j + 1, r);
}

归并排序算法模板

—— 模板题 AcWing 787. 归并排序

算法步骤：

​ ①确定分界点

​ ②递归排序

​ ③归并——合二为一

void merge_sort(int q[], int l, int r)
{
   
    if (l >= r) return;

    int mid = l + r >> 1;
    merge_sort(q, l, mid);
    merge_sort(q, mid + 1, r);
	
    // 归并的过程
    int k = 0, i = l, j = mid + 1;
    while (i <= mid && j <= r)
        if (q[i] <= q[j]) tmp[k ++ ] = q[i ++ ];
        else tmp[k ++ ] = q[j ++ ];

    while (i <= mid) tmp[k ++ ] = q[i ++ ];
    while (j <= r) tmp[k ++ ] = q[j ++ ];
        
    for (i = l, j = 0; i <= r; i ++, j ++ ) q[i] = tmp[j];
}

整数二分算法模板

—— 模板题 AcWing 789. 数的范围

Tips：

​ ①二分区间一点包含答案。

​ ②二分的时候一定是有解的，如果无解就是和题目有关，二分的模板一定保证有解。

bool check(int x) {
   /* ... */} // 检查x是否满足某种性质

// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用：
// 求左边界点
int bsearch_1(int l, int r)
{
   
    while (l < r)
    {
   
        int mid = l + r >> 1;
        if (check(mid)) r = mid;    // check()判断mid是否满足性质
        else l = mid + 1;
    }
    return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用：
// 求右边界点
int bsearch_2(int l, int r)
{
   
    while (l < r)
    {
   
        int mid = l + r + 1 >> 1;
        if (check(mid)) l = mid;
        else r = mid - 1;
    }
    return l;
}

浮点数二分算法模板

—— 模板题 AcWing 790. 数的三次方根

bool check(double x) {
   /* ... */} // 检查x是否满足某种性质

double bsearch_3(double l, double r)
{
   
    const double eps = 1e-6;   // eps 表示精度，取决于题目对精度的要求
    while (r - l > eps)
    {
   
        double mid = (l + r) / 2;
        if (check(mid)) r = mid;
        else l = mid;
    }
    return l;
}

高精度加法

—— 模板题 AcWing 791. 高精度加法

Tips：

​ ①高精度只对C++有要求

​ ②在笔试时偶尔会出现

// C = A + B, A >= 0, B >= 0
// 用引用传参会快很多
vector<int> add(vector<int> &A, vector<int> &B)
{
   
    if (A.size() < B.size()) return add(B, A);

    vector<int> C;
    int t = 0;
    for (int i = 0; i < A.size(); i ++ )
    {
   
        t += A[i];
        if (i < B.size()) t += B[i];
        C.push_back(t % 10);
        t /= 10;
    }

    if (t) C.push_back(t);
    return C;
}

高精度减法

—— 模板题 AcWing 792. 高精度减法

// C = A - B, 满足A >= B, A >= 0, B >= 0
vector<int> sub(vector<int> &A, vector<int> &B)
{
   
    vector<int> C;
    for (int i = 0, t = 0; i < A.size(); i ++ )
    {
   
        t = A[i] - t;
        if (i < B.size()) t -= B[i];
        C.push_back((t + 10) % 10);
        if (t < 0) t = 1;
        else t = 0;
    }
	
    // 去掉前导0
    while (C.size() > 1 && C.back() == 0) C.pop_back();
    return C;
}

高精度乘低精度

—— 模板题 AcWing 793. 高精度乘法

// C = A * b, A >= 0, b >= 0
vector<int> mul(vector<int> &A, int b)
{
   
    vector<int> C;

    int t = 0; // 进位
    for (int i = 0; i < A.size() || t; i ++ )
    {
   
        if (i < A.size()) t += A[i] * b;
        C.push_back(t % 10);
        t /= 10;
    }

    while (C.size() > 1 && C.back() == 0) C.pop_back();

    return C;
}

高精度除以低精度

—— 模板题 AcWing 794. 高精度除法

// A / b = C ... r, A >= 0, b > 0
vector<int> div(vector<int> &A, int b, int &r)
{
   
    vector<int> C;
    r = 0;
    for (int i = A.size() - 1; i >= 0; i -- )
    {
   
        r = r * 10 + A[i];
        C.push_back(r / b);
        r %= b;
    }r= 
    reverse(C.begin(), C.end());
    while (C.size() > 1 && C.back() == 0) C.pop_back();
    return C;
}

一维前缀和

—— 模板题 AcWing 795. 前缀和

前缀和的作用：快速计算某一段数的和

tips：

​ ①ios::sync_with_stdio(false); 提高cin的读取速度、不能再使用scanf了，但依旧没有scanf快。数据输入>=100w，建议使用scanf，否则建议使用cin。

​ ②只要定义了i - 1，则下标一般从1开始。

S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]

二维前缀和

—— 模板题 AcWing 796. 子矩阵的和

S[i, j] = 第i行j列格子左上部分所有元素的和
以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

一维差分

—— 模板题 AcWing 797. 差分

给区间[l, r]中的每个数加上c：B[l] += c, B[r + 1] -= c

二维差分

—— 模板题 AcWing 798. 差分矩阵

给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上c：
S[x1, y1] += c, S[x2 + 1, y1] -= c, S[x1, y2 + 1] -= c, S[x2 + 1, y2 + 1] += c

位运算

—— 模板题 AcWing 801. 二进制中1的个数

tips：

​ ①把第k位移到最后一位 n>>k

​ ②看个位是几 x&1

​ ③lowbit(x)：x的二进制表达式中最低位的1所对应的值

​ ④x&-x = x&(~x+1) ~x：x取反 简单应用：统计x中有多少个1

​ ⑤x = 1010 原码：1010 反码：0101 补码：0110

求n的第k位数字: n >> k & 1
返回n的最后一位1：lowbit(n) = n & -n

双指针算法

—— 模板题 AcWIng 799. 最长连续不重复子序列

—— 模板题 AcWing 800. 数组元素的目标和

tips：

​ ①算法复杂度都是O(n）

​ ②先想一个朴素算法，观察单调关系，再想双指针算法

for (int i = 0, j = 0; i < n; i ++ )
{
   
    while (j < i && check(i, j)) j ++ ;

    // 具体问题的逻辑
}
常见问题分类：
    (1) 对于一个序列，用两个指针维护一段区间
    (2) 对于两个序列，维护某种次序，比如归并排序中合并两个有序序列的操作

整数离散化

—— 模板题 AcWing 802. 区间和

数据离散化处理

定义：
离散化，把无限空间中有限的个体映射到有限的空间中去，以此提高算法的时空效率。

通俗的说，离散化是在不改变数据相对大小的条件下，对数据进行相应的缩小。例如：

原数据：1,999,100000,15；处理后：1,3,4,2；

原数据：{100,200}，{20,50000}，{1,400}；

处理后：{3,4}，{2,6}，{1,5}；

例如有时需要将数据直接作为数组下标，但因为数据太大而无法开数组时，若数据的总量相对较少时，就可以对其进行离散化处理

tips：

​ ①pair将一对值(T1和T2)组合成一个值，这一对值可以具有不同的数据类型（T1和T2），两个值可以分别用pair的两个公有函数first和second访问。

​ ②pair排序默认先按左端点排序，后按右端点排序。

​ ③c++用vector做离散化

vector<int> alls; // 存储所有待离散化的值
sort(alls.begin(), alls.end()); // 将所有值排序
alls.erase(unique(alls.begin(), alls.end()), alls.end());   // 去掉重复元素

// 二分求出x对应的离散化的值
int find(int x) // 找到第一个大于等于x的位置
{
   
    int l = 0, r = alls.size() - 1;
    while (l < r)
    {
   
        int mid = l + r >> 1;
        if (alls[mid] >= x) r = mid;
        else l = mid + 1;
    }
    return r + 1; // 映射到1, 2, ...n
}

区间合并

—— 模板题 AcWing 803. 区间合并

// 将所有存在交集的区间合并
void merge(vector<PII> &segs)
{
   
    vector<PII> res;

    sort(segs.begin(), segs.end());

    int st = -2e9, ed = -2e9;
    for (auto seg : segs)
        if (ed < seg.first)
        {
   
            if (st != -2e9) res.push_back({
   st, ed});
            st = seg.first, ed = seg.second;
        }
        else ed = max(ed, seg.second);

    if (st != -2e9) res.push_back({
   st, ed});

    segs = res;
}

上课：明白主要思想

课下：代码整个流程弄懂、大致的背过、题目默写相关模板、写出来并调试通过

如何提高熟练度：重复写代码（删、写）、一个题目写3-5遍

二、数据结构

1、常用代码模板

单链表

—— 模板题 AcWing 826. 单链表

Tips:

​ ①scanf每次%c之前都要加上空格,防止读取空格

struct Node
{
   
	int val;
	Node *next;
};//在面试题中用的多，在笔试中用一般不会用，因为new Node()非常慢，用数组模拟链表很快

// head存储链表头，e[]存储节点的值，ne[]存储节点的next指针，idx表示当前用到了哪个节点
int head, e[N], ne[N], idx;

// 初始化
void init()
{
   
    head = -1;
    idx = 0;
}

// 在链表头插入一个数a，最常用的插入操作
void insert(int a)
{
   
    e[idx] = a, ne[idx] = head, head = idx ++ ;
}

// 将头结点删除，需要保证头结点存在
void remove()
{
   
    head = ne[head];
}

双链表

—— 模板题 AcWing 827. 双链表

// e[]表示节点的值，l[]表示节点的左指针，r[]表示节点的右指针，idx表示当前用到了哪个节点
int e[N], l[N], r[N], idx;

// 初始化
void init(