C++知识点总结(47)：倍增法

一、意义

倍增法：一种用于解决一些计算问题的算法。它基于一个简单的思想：通过重复执行一个操作来逐步逼近最终结果。

二、例题

1. 链式反应

题目描述

最初，有 $x$ 个中子。每轮裂变后，每个中子会再分裂出一个中子。
请问，在进行几轮裂变后，至少有 $y$ 个中子。

输入描述

第一行两个正整数 $x$ 和 $y$，用一个空格分隔。

输出描述

输出一行一个整数表示答案。

样例1

输入

1 1

输出

0

样例2

输入

1 2

输出

1

提示

$1\le x,y\le 10^9$。

这种题目要是在 CSP-J 考到了，我只能说有点水。这道题目我们可以直接用 while 循环来套：

#include <iostream>
using namespace std;

int x, y;
int cnt;

int main()
{
    cin >> x >> y;
    
    while (x < y)
    {
        cnt++;
        x *= 2;
    }
    
    cout << cnt;
    return 0;
}

当然，请注意一个非常恐怖的点：$1$ 到 $10^9$ 只需要乘 $30$ 次！那么怎么办呢？那就看下面的题吧。

2. 迷途

题目描述

有 $n$ 扇传送门，从左到右编号为 $1$ 到 $n$。第 $i$ 座传送门会把你传送到第 $a[i]$ 扇门前。
有 $q$ 次询问，每次给出 $x$ 和 $k$。请回答：从第 $x$ 扇门出发，经过 $k$ 次传送后，会到达第几扇门前。

输入描述

第一行两个正整数 $n$ 和 $q$，用一个空格隔开。
第二行 $n$ 个正整数 $a[1]-a[n]$，用一个空格隔开。
随后 $q$ 行，每行两个正整数 $x$ 和 $k$，用一个空格隔开。

输出描述

对于每组数据，输出一行一个正整数表示答案。

样例1

输入

6 3
3 1 2 4 6 5
1 4
4 2
5 3

输出

3
4
6

提示

$1\le n,q\le 200000$
$1\le a[i]\le n$
$1\le x\le n，1\le k\le 10^9$

我们先来用遍历的方法写一下逝逝看（当然是不能通过的）。

#include <iostream>
using namespace std;

int n, q;
int x, k;
int a[200005];

int main()
{
    cin >> n >> q;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    while (q--)
    {
        cin >> x >> k;
        while (k--)
            x = a[x];
        cout << x << endl;
    }
    return 0;
}

分析一下，我们的时间复杂度是 $O(qk)$，这相当于最坏的情况下要执行 $20^{14}$ 次！

接下来我们就需要来寻找一种方法降低时间复杂度，可以尝试用记忆化搜索来试一试。

但是如果用一个 $n\times k$ 的表格，512MB 的内存肯定过不了。轮到倍增出场！

每个正整数都可以分解为若干个 $2$ 的幂数之和。 换句话说，每个十进制数都可以转换成二进制数，那么十进制数就可以转换为：
$$n_{(10)}=2^{p_1}+2^{p_2}+\cdots +2^{p_n}$$

因为 $2^{30}>10^9$，所以我们的行就可以变成 $30$ 的大小了！因此我们有一个 to[][] 数组，to[i][j] 表示从第 $j$ 扇门开始传送 $2^i$ 次的终点。

所以我们得出如下递归式：
$$\text{to}[i][j]=\text{to}[i-1][\text{to}[i-1][j]]$$

上代码！

#include <iostream>
using namespace std;

int n, q;
int x, k;
int tmp;
int a[200005];
int to[35][200005];

int main()
{
    cin >> n >> q;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        to[0][i] = a[i];
    }
    
    for (int i = 1; i <= 30; i++)
        for (int j = 1; j <= n; j++)
            to[i][j] = to[i-1][to[i-1][j]];
    
    while (q--)
    {
        cin >> x >> k;
        for (int i = 0; i <= 30; i++)
            if (k & (1 << i))
                x = to[i][x];
        cout << x << endl;
    }
    return 0;
}

三、应用

1. RMQ Ⅰ

题目描述

有一个长度为 $n$ 的数组，编号从 $1$ 到 $n$。
有 $q$ 次询问，每次给出 $l$ 和 $r$，请输出第 $l$ 到第 $r$ 个数中的最大值。

输入描述

第一行两个正整数 $n$ 和 $q$，用一个空格隔开。
第二行 $n$ 个正整数 $a[1]-a[n]$，用一个空格隔开。
随后 $q$ 行，每行两个正整数 $l$ 和 $r$，用一个空格隔开。

输出描述

对于每组数据，输出一行一个正整数表示答案。

样例1

输入

8 2
5 3 4 2 6 1 10 100
2 4
1 6

输出

4
6

提示

$1\le n,q\le 200,000$
$1\le a[i]\le 10^9$
$1\le l\le r\le n$

套上区间最值问题（$\text{RMQ}$，求任意连续区间的最大值或最小值）模板。我们有一个 st 表，st[i][j] 表示从第 $j$ 个数字开始，连续 $2^i$ 个数字中的最大值。我们可以推导出如下递归式：
$$\text{st}[i][j]=\text{max}(\text{st}[i-1][j],\text{st}[i-1][j+2^{i-1}])$$

#include <iostream>
#include <cmath>
using namespace std;

int n, q;
int l, r;
int a[200005];
int st[20][200005];

int main()
{
    cin >> n >> q;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        st[0][i] = a[i];
    }
    for (int i = 1; i <= 18; i++)
        for (int j = 1; j <= n; j++)
            st[i][j] = max(st[i-1][j], st[i-1][j+(1<<(i-1))]);
    while (q--)
    {
        cin >> l >> r;
        int k = log2(r-l+1);
        cout << max(st[k][l], st[k][r+1-(1<<k)]) << endl;
    }
    return 0;
}

2. RMQ Ⅱ

有一个长度为 $n$ 的数组，编号从 $1$ 到 $n$。
有 $q$ 次询问，每次给出 $l$ 和 $r$，请输出第 $l$ 到第 $r$ 个数中的最大值减去最小值的答案。

#include <iostream>
#include <cmath>
using namespace std;

int n, q;
int l, r;
int a[200005];
int st1[20][200005];
int st2[20][200005];

int main()
{
    cin >> n >> q;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        st1[0][i] = a[i];
        st2[0][i] = a[i];
    }
    for (int i = 1; i <= 18; i++)
        for (int j = 1; j <= n; j++)
        {
            st1[i][j] = max(st1[i-1][j], st1[i-1][min(j+(1<<(i-1)), n)]);
            st2[i][j] = min(st2[i-1][j], st2[i-1][min(j+(1<<(i-1)), n)]);
        }
    while (q--)
    {
        cin >> l >> r;
        int k = log2(r-l+1);
        int maxn = max(st1[k][l], st1[k][r-(1<<k)+1]);
        int minn = min(st2[k][l], st2[k][r-(1<<k)+1]);
        cout << maxn-minn << endl;
    }
    return 0;
}