基础算法模板#

雄关漫道真如铁，而今迈步从头越

1 基础算法#

1.1 归并排序#

稳定的排序，很好敲（比起快排我更喜欢归并）

应用：求逆序对

Note

模板速记

核心思想：分治后合并两个有序区间
时间复杂度： $O(n\log n)$
常见扩展：合并时统计逆序对贡献

1
void msort(int a[],int l,int r) {
2
    if(l>=r) return;
3
    int mid=(l+r)>>1;
4
    msort(a,l,mid);
5
    msort(a,mid+1,r);
6
    int i=l,j=mid+1,k=l;
7
    while(i<=mid&&j<=r) b[k++]=(a[i]>a[j])?a[j++]:a[i++];
8
    while(i<=mid) b[k++]=a[i++];
9
    while(j<=r) b[k++]=a[j++];
10
    for(int i=l;i<=r;i++) a[i]=b[i];
11
    return;
12
}

1.2 快速排序#

~~应用？有什么应用呢？~~

1
void qsort(int a[],int l,int r) {
2
    if(l>=r) return;
3
    int i=l-1,j=r+1,x=a[l+r>>1];
4
    while(i<j) {
5
        do i++; while(a[i]<x);
6
        do j--; while(a[j]>x);
7
        if(i<j) swap(a[i],a[j]);
8
    }
9
    qsort(a,l,j);
10
    qsort(a,j+1,r);
11
}

1.3 二分#

吕优学长的稳健的二分模板

应用：二分答案，适用于单调的问题，通常是二分答案后进行贪心可行性验证。

Tip

二分最容易错的不是模板，而是 check(mid) 的方向。写模板前先确定答案区间中“可行/不可行”的分界线。

整数

1
int bsch(int L,int R) {
2
    int ans=L;
3
  while(L<=R) {
4
        int mid=(L+R)>>1;
5
        if(check(mid)) ans=mid,L=mid+1;
6
        else R=mid-1; //可根据题意与上句交换
7
    }
8
    return ans;
9
}

浮点数

1
const double eps=1e-6;
2
double bfsch(int L,int R) {
3
  while(R-L>eps) {
4
        double mid=(L+R)/2;
5
        if(check(mid)) L=mid;
6
        else R=mid; //可根据题意与上句交换
7
    }
8
    return L;
9
}

1.4 前缀和 & 差分#

前缀和：离线问题，单点修改 -> 区间查询

一维前缀和：
- 记录：s[i]=s[i-1]+a[i]
- 查询[l,r]：s[r]-s[l-1]
二维前缀和：
- 记录：s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]
- 查询[(x1,y1),(x2,y2)]：s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]

差分：离线问题，区间修改 -> 单点查询

一维差分：
- 区间[l,r]修改：d[l]+=k, d[r+1]-=k
- 计算前缀和：d[i]+=d[i-1]
二维差分：
- 区间[(x1,y1),(x2,y2)]修改：d[x1][y1]+=k, d[x2+1][y1]-=k, d[x1][y2+1]-=k, d[x2+1][y2+1]+=k
- 计算前缀和：d[i][j]+=d[i-1][j]+d[i][j-1]-d[i-1][j-1]

1.5 有序数组定位#

1.5.1 lower_bound & upper_bound#

函数原型：

1
typedef ForwardIterator FI;
2
// 在[first,last)区域内查找不小于 val 的元素
3
FI lower_bound(FI first, FI last, const T& val);
4
// 在[first,last)区域内查找第一个不符合 cmp 规则的元素
5
FI lower_bound(FI first, FI last, const T& val, Compare cmp);
6
// 在[first,last)区域内查找第一个大于 val 的元素。
7
FI upper_bound(FI first, FI last, const T& val);
8
// 在[first,last)区域内查找第一个不符合 cmp 规则的元素
9
FI upper_bound(FI first, FI last, const T& val, Compare cmp);
10
// 在[first,last)区域内查找所有等于 val 的元素
11
pair<FI,FI> equal_range (FI first, FI last, const T& val[, Compare cmp]);

应用举例：

1
//vector<int>
2
sort(v.begin(),v.end());
3
lower_bound(v.begin(),v.end(),x)-v.begin();   // 第一个大于等于x的元素位置
4
upper_bound(v.begin(),v.end(),x)-v.begin();   // 第一个大于x的元素位置
5
lower_bound(v.begin(),v.end(),x)-v.begin()-1;   // 最后一个小于x的元素位置
6
upper_bound(v.begin(),v.end(),x)-v.begin()-1;   // 最后一个小于等于x的元素位置
7

8
sort(v.begin(),v.end(),greater<int>());
9
lower_bound(v.begin(),v.end(),x,greater<int>())-v.begin();     // 第一个小于等于x的元素位置
10
upper_bound(v.begin(),v.end(),x,greater<int>())-v.begin();     // 第一个小于x的元素位置
11
lower_bound(v.begin(),v.end(),x,greater<int>())-v.begin()-1;   // 最后一个大于x的元素位置
12
upper_bound(v.begin(),v.end(),x,greater<int>())-v.begin()-1;   // 最后一个大于等于x的元素位置
13

14
//数组
15
sort(a,a+n);
16
lower_bound(a,a+n,x)-a;   // 第一个大于等于x的元素位置
17
upper_bound(a,a+n,x)-a;   // 第一个大于x的元素位置
18
lower_bound(a,a+n,x)-a-1;   // 最后一个小于x的元素位置
19
upper_bound(a,a+n,x)-a-1;   // 最后一个小于等于x的元素位置
20

21
bool cmp(int a,int b) {return a>b;}
22
sort(a,a+n,cmp);
23
lower_bound(a,a+n,x,cmp)-a;   // 第一个小于等于x的元素位置
24
upper_bound(a,a+n,x,cmp)-a;   // 第一个小于x的元素位置
25
lower_bound(a,a+n,x,cmp)-a-1;   // 最后一个大于x的元素位置
26
upper_bound(a,a+n,x,cmp)-a-1;   // 最后一个大于等于x的元素位置

【例题】

给定一个按照升序排列的长度为 $n$ 的整数数组，以及 $m$ 个查询。

对于每个查询，返回一个元素 $x$ 的起始位置和终止位置（位置从 0 开始计数）。

如果数组中不存在该元素，则返回 -1 -1。

法一：二分

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,m,a[N];
5
int SL(int x) { //左界
6
    int l=0,r=n-1,ret=1e9+7;
7
    while(l<=r) {
8
        int mid=l+r>>1;
9
        if(a[mid]>=x) ret=min(ret,mid),r=mid-1;
10
        else l=mid+1;
11
    }
12
    return ret;
13
}
14
int SR(int x) { //右界
15
    int l=0,r=n-1,ret=-1;
16
    while(l<=r) {
17
        int mid=l+r>>1;
18
        if(a[mid]<=x) ret=max(ret,mid),l=mid+1;
19
        else r=mid-1;
20
    }
21
    return r;
22
}
23
int main() {
24
    scanf("%d%d",&n,&m);
25
    for(int k=0;k<n;k++) scanf("%d",a+k);
26
    while(m--) {
27
        int x;
28
        scanf("%d",&x);
29
        int L=SL(x),R=SR(x);
30
        if(a[L]==x) printf("%d %d\n",L,R);
31
        else printf("-1 -1\n");
32
    }
33
    return 0;
34
}

法二：lowerbound & upperbound 定位

1
#include<bits/stdc++.h>
2
using namespace std;
3
int n,m,a[100010];
4
int main() {
5
    scanf("%d%d",&n,&m);
6
    for(int i=1;i<=n;i++) scanf("%d",a+i);
7
    while(m--) {
8
        int l,r,x;
9
        scanf("%d",&x);
10
        l=lower_bound(a+1,a+n+1,x)-a;
11
        r=upper_bound(a+1,a+n+1,x)-a-1;
12
        if(l>r) l=r=0;
13
        printf("%d %d\n",l-1,r-1);
14
    }
15
    return 0;
16
}

1.5.2 离散化#

不改变数据相对大小的前提下将稀疏的数据映射为连续的数据后存储。

【例题】

$n$ 个修改：x c：a[x]+=c
$m$ 个查询：l r：查询 $\sum_{[l,r]}a_i$
$1≤n,m≤10^5,\\ −10^9≤x≤10^9,\\ −10^9≤l≤r≤10^9,$

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1e5+10;
4
int n,m,s[N];
5
vector<pair<int,int> > a;
6
vector<int> q;
7
int main() {
8
    scanf("%d%d",&n,&m);
9
    while(n--) {
10
        int x,c;
11
        scanf("%d%d",&x,&c);
12
        a.push_back({x,c});
13
        q.push_back(x);
14
    }
15
    q.push_back(-0x3f3f3f); q.push_back(0x3f3f3f);    // 前后界防越界【养成好习惯】
16
    sort(q.begin(),q.end());              // 排序
17
    q.erase(unique(q.begin(),q.end()),q.end());      // 去重【记住这个用法！】
18
    for(auto p:a) {
19
        int i=lower_bound(q.begin(),q.end(),p.first)-q.begin();
20
        s[i]+=p.second;
21
    }
22
    for(int i=1;i<q.size();i++) s[i]+=s[i-1];
23

24
    while(m--) {
25
        int l,r;
26
        scanf("%d%d",&l,&r);
27
        l=lower_bound(q.begin(),q.end(),l)-q.begin();    // l:第一个大于等于l的位置
28
        r=upper_bound(q.begin(),q.end(),r)-q.begin()-1;    // r:最后一个小于等于r的位置
29
        printf("%d\n",s[r]-s[l-1]);
30
    }
31
}

1.6 位运算#

1.6.1 数字二进制位数#

1
int len(int x) {  //十进制同理
2
    int ret=0;
3
    do ret++,x>>=1; while(x);
4
    return ret;
5
}

1.6.2 数字二进制中’1’的个数#

1
int lowbit(int x) {return x&-x;}
2
int count1(int x) {
3
    int ret=0;
4
    while(x) ret++,x-=lowbit(x);
5
    return ret;
6
}

1.6.3 二进制枚举#

1
for(int i=0;i<1<<n;i++) {     // 枚举n位01串
2
    for(int j=0;j<n;j++) {
3
        if(check(i>>j&1)) {    // 检验i的第j位数字
4
            /* … */
5
        }
6
    }
7
}

1.7 高精度#

高精度运算合集

1.8 KMP#

nxt数组的含义：nxt[i] 表示 b[0~i-1] 最长公共前后缀长度（不包括本身）

1
char a[N],b[N]; //b[]为模式串
2
int lena,lenb,nxt[N];
3
void pre() {
4
    for(int i=1,j=0;i<lenb;i++) {
5
        while(j&&b[i]!=b[j]) j=nxt[j];
6
        if(b[i]==b[j]) j++;
7
        nxt[i+1]=j;  // nxt[i+1]对应b[i]
8
    }
9
}
10
void kmp() {
11
    for(int i=0,j=0;i<lena;i++) {
12
        while(j&&a[i]!=b[j]) j=nxt[j];
13
        if(a[i]==b[j]) j++;
14
        if(j==lenb) printf("%d\n",i+1-lenb+1); // 匹配位置
15
    }
16
}

1.9 Hash#

1.9.1 拉链法#

1
const int N=1e6+3;
2
int h[N],e[N],nxt[N],tot;
3
void insert(int k) {
4
    int x=(k%N+N)%N;
5
    e[++tot]=k;
6
    nxt[tot]=h[x];
7
    h[x]=tot;
8
}
9
bool find(int k) {
10
    int x=(k%N+N)%N;
11
    for(int i=h[x];i;i=nxt[i])
12
        if(e[i]==k) return 1;
13
    return 0;
14
}

1.9.2 开放寻址法（线性探查法）#

1
const int N=1e6+3;
2
int h[N]={-1};
3

4
int find(int k) { // 若k在哈希表中，返回k的下标；否则返回k应该插入的位置
5
  int x=(k%N+N)%N;
6
    while(h[x]!=-1&&h[x]!=k)  (x+=1)%=N;
7
  return x;
8
}

1.9.3 字符串Hash#

ELFHash

1
unsigned ELFHash(const char *str) {
2
  unsigned h=0,g;
3
  while(*str) {
4
    h=(h<<4)+*str++;
5
    if(g=h&0xf0000000) h^=g>>24;
6
    h&=~g;
7
  }
8
  return h;
9
}

1.10 快速乘快速幂#

1
ll ksc(ll a,ll k) { // 快速乘
2
    ll ret=0;
3
    for(;k;(a<<=1)%=mod,k>>=1) if(k&1) (ret+=a)%=mod;
4
    return ret;
5
}
6
ll ksm(ll a,ll k) {  // 快速幂
7
    ll ret=1;
8
    for(;k;(a*=a)%=mod,k>>=1) if(k&1) (ret*=a)%=mod;
9
    return ret;
10
}

2 基础数据结构#

2.1 单链表#

1
int head,e[N],nxt[N],tot;
2
void insert(int x) { // 头插法
3
    e[tot]=x,
4
    nxt[tot]=head,
5
    head=tot;
6
}
7
void remove() {  // 将头结点删除，需要保证头结点存在
8
    head = nxt[head];
9
}

【例】实现单链表及其操作：

H x，表示向链表头插入一个数 $x$ 。
D k，表示删除第 $k$ 个插入的数后面的数（当 $k$ 为 $0$ 时，表示删除头结点）。
I k x，表示在第 $k$ 个插入的数后面插入一个数 $x$ （此操作中 $k$ 均大于 $0$ ）。

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int m,k,x;
5
int e[N],nxt[N],head,tot;
6
void insert(int i,int x) {
7
    e[++tot]=x;
8
    if(!i) nxt[tot]=head, head=tot;
9
    else nxt[tot]=nxt[i], nxt[i]=tot;
10
}
11
void remove(int i) {
12
    if(!i) head=nxt[head];
13
    else nxt[i]=nxt[nxt[i]];
14
}
15
int main() {
16
    scanf("%d",&m);
17
    while(m--) {
18
        char c[1];
19
        scanf("%s",c);
20
        if(c[0]=='H') { scanf("%d",&x); insert(0,x); }
21
        else if(c[0]=='D') { scanf("%d",&k); remove(k); }
22
        else if(c[0]=='I') { scanf("%d%d",&k,&x); insert(k,x); }
23
    }
24
    for(int i=head;i;i=nxt[i]) printf("%d ",e[i]);
25
    return 0;
26
}

2.2 双链表#

1
int e[N],l[N],r[N],tot;
2
const R=N-1;
3
void init() {  // 初始化
4
    r[0]=R,l[R]=0;
5
}
6
void insert(int k, int w) {  // 在节点k的右边插入一个数w
7
    e[++tot]=x;
8
    l[tot]=k,r[tot]=r[k];
9
    l[r[k]]=tot,r[k]=tot;
10
}
11
void remove(int k) {  // 删除节点k
12
    l[r[k]]=l[k];
13
    r[l[k]]=r[k];
14
}

【例】实现双链表及其操作：

L x，表示在链表的最左端插入数 $x$ 。
R x，表示在链表的最右端插入数 $x$ 。
D k，表示将第 $k$ 个插入的数删除。
IL k x，表示在第 $k$ 个插入的数左侧插入一个数。
IR k x，表示在第 $k$ 个插入的数右侧插入一个数。

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100011;
4
const int R=N-10;
5
int m,e[N],l[N],r[N],tot;
6
void insert(int k,int x) {
7
    e[++tot]=x;
8
    l[tot]=k,r[tot]=r[k];
9
    l[r[k]]=tot,r[k]=tot;
10
}
11
void remove(int k) {
12
    l[r[k]]=l[k];
13
    r[l[k]]=r[k];
14
}
15
int main() {
16
    scanf("%d",&m);
17
    r[0]=R,l[R]=0;
18
    while(m--) {
19
        int k,x;
20
        string c;
21
        cin>>c;
22
        if(c=="L") { scanf("%d",&x); insert(0,x); }
23
        else if(c=="R") { scanf("%d",&x); insert(l[R],x); }
24
        else if(c=="D") { scanf("%d",&k); remove(k); }
25
        else if(c=="IL") { scanf("%d%d",&k,&x); insert(l[k],x); }
26
        else { scanf("%d%d",&k,&x); insert(k,x); }
27
    }
28
    for(int i=r[0];i!=R;i=r[i]) printf("%d ",e[i]);
29
    return 0;
30
}

2.3 栈#

2.3.1 模拟栈#

1
int st[N],t=0;
2
bool empty() { return !t; }
3
int size() { return t; }
4
void push(int x) { st[++t]=x; }
5
int pop() {
6
    if(empty()) return -1;
7
    return st[t--];
8
}
9
int top() {
10
    if(empty()) return -1;
11
    return st[t];
12
}
13
void clear() { t=0; }

2.3.2 STL stack#

1
stack<int> s;
2
s.empty();  // 是否为空
3
s.size();  // 返回元素个数
4
s.top();  // 返回栈顶元素
5
s.push(x);  // 压入x
6
s.pop();  // 弹出栈顶（注意STL容器的pop是动作，无返回值）
7
s.clear();   // 不存在！
8
stack<int>().swap(s); // 等同于所谓's.clear()'

2.3.3* 单调栈#

适用问题：找出每个数两侧离它最近的比它大/小的数

1
while(!s.empty()&&check(s.top(),i)) s.pop();
2
s.push(i);

【例】接雨水

给定 $n$ 个非负整数表示每个宽度为 $1$ 的柱子的高度图。

计算按此排列的柱子，下雨之后能接多少雨水。

法一：单调栈

1
int ans=0,n=h.size();
2
stack<int> s;
3
for(int i=0;i<n;i++) {
4
    while(!s.empty()&&h[i]>h[s.top()]) {
5
        int t=s.top(); s.pop();
6
        if(s.empty()) break;
7
        int j=s.top();
8
        ans+=(i-j-1)*(min(h[j],h[i])-h[t]);
9
    }
10
    s.push(i);
11
}
12
printf("%d",ans);

法二：dp

1
const int N=2e4+10;
2
int ans=0,n=h.size();
3
int lm[N]={0},rm[N]={0};
4
for(int i=1;i<=n;i++) {
5
    lm[i]=max(lm[i-1],h[i-1]);
6
  rm[n-i+1]=max(rm[n-i+2],h[n-i]);
7
}
8
for(int i=1;i<=n;i++) ans+=min(lm[i],rm[i])-h[i-1];
9
printf("%d",ans);

法三：双指针

1
int i=0,j=h.size()-1,ans=0;
2
int l=h[i],r=h[j];
3
while(i<j) {
4
    if(h[i]<h[j])
5
        ans+=l-h[i++],l=max(l,h[i]);
6
    else
7
        ans+=r-h[j--],r=max(r,h[j]);
8
}
9
printf("%d",ans);

2.4 队列#

2.4.1 模拟队列#

普通队列（t=-1写法）

1
int q[N],h=0,t=-1;
2
bool empty() { return t<h; }
3
int size() { return t-h+1; }
4
void push(int x) { q[++t]=x; }
5
int pop() {
6
    if(empty()) return -1;
7
    return q[h++];
8
}
9
int front() {
10
    if(empty()) return -1;
11
    return q[h];
12
}
13
void clear() { h=0,t=-1; }

循环队列

1
int q[N],h=0,t=0;
2
bool empty() { return h==t; }
3
int size() { return (t+N-h)%N; }
4
void push(int x) {
5
    q[t++]=x;
6
    if(t==N) t=0;
7
}
8
int pop() {
9
    if(empty()) return -1;
10
    int ret=q[h++];
11
    if(h==N) h=0;
12
    return ret;
13
}
14
int front() {
15
    if(empty()) return -1;
16
    return q[h];
17
}
18
void clear() { h=t=0; }

双端队列

1
int dq[N<<1],h=N,t=N-1;
2
bool empty() { return t<h; }
3
int size() { return t-h+1; }
4
void push_front(int x) { dq[--h]=x; }
5
void push_back(int x) { dq[++t]=x; }
6
int pop_front() {
7
    if(empty()) return -1;
8
    return dq[h++];
9
}
10
int pop_back() {
11
    if(empty()) return -1;
12
    return dq[t--];
13
}
14
int front() {
15
    if(empty()) return -1;
16
    return q[h];
17
}
18
int back() {
19
    if(empty()) return -1;
20
    return q[t];
21
}
22
int clear() { h=N,t=N-1; }

2.4.2 STL queue/deque#

1
queue<int> q;
2
q.empty();  // 是否为空
3
q.size();  // 返回元素个数
4
q.front();  // 返回栈顶元素
5
q.push(x);  // 入队x
6
q.pop();  // 队首出队（注意STL容器的pop是动作，无返回值）
7
q.clear();   // 不存在！
8
queue<int>().swap(q); // 等同于所谓'q.clear()'
9

10
deque<int> dq;  // 双端队列
11
dq.empty();
12
dq.size();
13
dq.front();  dq.back();
14
dq.push_front(); dq.push_back();
15
dq.pop_front(); dq.pop_back();
16
deque<int>().swap(dq);

2.4.3 单调队列#

本质：如果一个选手比你小还比你强，你就可以退役了。

应用：滑动窗口问题，区间最值问题，单调队列优化dp

1
while(!dq.empty()&&check_out(dq.front()) dq.pop_front();
2
while(!dq.empty()&&check_in(dq.back(),i)) dq.pop_back();
3
dq.push_back(i);

【例题】长度不超过 $k$ 的最大连续子区间和

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,k,a[N],s[N],ans=-0x3f3f3f;
5
deque<int> q;
6
int main() {
7
    scanf("%d%d",&n,&k);
8
    for(int i=1;i<=n;i++) scanf("%d",s+i),s[i]+=s[i-1];
9
    q.pop_back(0);
10
    for(int i=1;i<=n;i++) {
11
        while(!q.empty()&&i-q.front()>k) q.pop_front();
12
        while(!q.empty()&&s[q.back()]>s[i]) q.pop_back();
13
        q.push_back(i);
14
        ans=max(ans,s[i]-s[q.front()]);
15
    }
16
    printf("%d",ans);
17
}

2.4.4 优先队列#

1
priority_queue<int> q;  // 大根堆
2
priority_queue<int,vector<int>,greater<int> > q2;  // 小根堆
3
q.top();  // 队头
4
q.empty();   // 是否为空
5
q.size();   // 返回元素个数
6
q.pop();  // 队头出队
7
priority_queue<int>().swap(q);  // q.clear()

应用详见 2.5.2 STL priority_queue

2.5 堆#

2.5.1 模拟堆#

1
// h[N]存储堆中的值, h[1]是堆顶
2
int h[N],tot;
3
void down(int x) {  // 下沉
4
    int t=x;
5
    if((x<<1)<=n && h[x<<1]<h[t]) t=x<<1;
6
    if((x<<1|1)<=n && h[x<<1|1]<h[t]) t=x<<1|1;
7
    if(x!=t) swap(h[x],h[t]),down(t);
8
}
9
void up(int x) {  // 上浮
10
    while((x>>1)&&h[x]<h[x>>1]) {
11
        swap(h[x],h[x>>1]);
12
        x>>=1;
13
    }
14
}
15
void init() {  // 重整堆
16
    for(int i=tot>>1;i;i--) down(i);
17
}
18
int top() { return h[1]; }
19
void push(int x) {  // 压入堆
20
    h[++tot]=x;
21
    up(tot);
22
}
23
int pop() {    // 弹出堆顶
24
    if(!tot) return -1;
25
    int ret=h[1];
26
    h[1]=h[tot--];
27
    down(1);
28
    return ret;
29
}
30
void clear() { tot=0; }

2.5.2 STL priority_queue#

详细定义见 2.4.4 优先队列

应用：堆排序，Dijkstra，Prim

2.5.3 堆排序#

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,m,h[N];
5
void down(int x) {
6
    int t=x;
7
    if((x<<1)<=n&&h[x<<1]<h[t]) t=x<<1;
8
    if((x<<1|1)<=n&&h[x<<1|1]<h[t]) t=x<<1|1;
9
    if(x^t) swap(h[x],h[t]),down(t);
10
}
11
int main() {
12
    scanf("%d%d",&n,&m);
13
    for(int i=1;i<=n;i++) scanf("%d",h+i);
14
    for(int i=n>>1;i;i--) down(i);
15
    while(m--) printf("%d ",h[1]),h[1]=h[n--],down(1);
16
    return 0;
17
}

2.6 并查集#

初始化与合并操作

1
void init() {
2
    for(int i=1;i<N;i++) fa[i]=i /*,size[i]=1,d[i]=0*/;
3
}
4
void Union(int x,int y) {
5
    int a=find(x),b=find(y);
6
    // size[b]+=size[a]  // 同时维护并查集规模
7
    fa[a]=b;
8
}

朴素并查集

1
int find(int x) {
2
    while(fa[x]) x=fa[x];
3
    retun x;
4
}

带路径压缩

1
int find(int x) {
2
    return x==fa[x]?x:(fa[x]=find(fa[x]));
3
}

按秩合并（没什么用）
路径压缩的同时保留到祖宗节点的距离信息

1
int find(int x) {
2
    if(x==fa[x]) return x;
3
    else {
4
    int p=find(fa[x]);
5
        d[x]+=d[fa[x]];
6
        fa[x]=p;
7
    }
8
  return fa[x];
9
}
10
void init() {
11
    for(int i=1;i<N;i++) fa[i]=i,d[i]=0;
12
}
13
void Union(int x,int y) {
14
    int a=find(x),b=find(y);
15
    fa[a]=b;
16
    d[a]=1;
17
}

2.7 邻接表（前向星）#

1
const int N=1010;
2
const int M=N*N<<1;
3
int head[N],tot;
4
int edg[M],ver[M],nxt[M];
5
void add(int x,int y,int z) {
6
    ver[++tot]=y;
7
    edg[tot]=z;
8
    nxt[tot]=head[x];
9
    head[x]=tot;
10
}
11
// 遍历
12
void dfs(int x) {
13
  for(int i=head[x];i;i=nxt[i]) {
14
      int y=ver[i],z=edg[i];
15
        /* */
16
    }
17
}

也有vector写法：

1
const int N=1010;
2
vector<pair<int,int> > e[N];
3
void add(int x,int y,int z) {
4
    e[x].push({y,z});
5
}
6
void dfs(int x) {
7
  for(auto i:e[x]) {
8
        int y=i.first,z=i.second;
9
        /* */
10
    }
11
}

2.8 Trie字典树#

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010,M=1<<7;
4
int m,t[N][M],cnt[N],tot;
5
char opt[1],s[N];
6
int Insert() {
7
    int p=0;
8
    for(int i=0;s[i];i++) {
9
        int idx=s[i]-'a';
10
        if(!t[p][idx]) t[p][idx]=++tot;
11
        p=t[p][idx];
12
    }
13
    return cnt[p]++;
14
}
15
int Query() {
16
    int p=0;
17
    for(int i=0;s[i];i++) {
18
        int idx=s[i]-'a';
19
        p=t[p][idx];
20
        if(!p) return 0;
21
    }
22
    return cnt[p];
23
}
24
int main() {
25
    scanf("%d",&m);
26
    while(m--) {
27
        scanf("%s%s",opt,s);
28
        switch(opt[0]) {
29
            case 'I': Insert();break;
30
            case 'Q': printf("%d\n",Query());break;
31
        }
32
    }
33
}

2.9 哈夫曼树（Huffman）#

下面先放个运用Huffman树思想的例子

1
#include<bits/stdc++.h>
2
using namespace std;
3
int n,ans,t;
4
priority_queue<int,vector<int>,greater<int> > q;
5

6
int main() {
7
    scanf("%d",&n);
8
    for(int i=0;i<n;i++) {
9
        scanf("%d",&t);
10
        q.push(t);
11
    }
12
    for(int i=1;i<n;i++) {
13
        int a=q.top(); q.pop();
14
        int b=q.top(); q.pop();
15
        ans+=a+b; q.push(a+b);
16
    }
17
    printf("%d",ans);
18
}

2.10* ST表#

RMQ (Range Maximum/Minimum Query) 问题，即区间最值问题

ST表（Sparse Table）解决离线查询区间最值

Note

适用场景

静态数组，多次区间最值查询
预处理 $O(n\log n)$ ，查询 $O(1)$
不支持在线修改，若有修改通常考虑线段树

定义 $f(i,j)$ 为以第 $i$ 个数为起点，长度为 $2^j$ 的一段区间中的最大值，则显然状态转移为

f(i,j)=\max{( f(i,j-1),f(i+2^{j-1},j-1))) }

查询同理，只需查询两段（两段中间可能有重叠部分）

1
void pre(int n) {
2
    int t=log2(n)+1;
3
    for(int j=1;j<t;j++)
4
        for(int i=1;i<=n-(1<<j)+1;i++)
5
            f[i][j]=max(f[i][j-1],f[i+(1<<(j-1))][j-1]);
6
}
7
int query(int l,int r) {
8
    int k=log2(r-l+1);
9
    return max(f[l][k],f[r-(1<<k)+1][k]);
10
}
11
int main() {
12
    scanf("%d%d",&n,&m);
13
    for(int i=1;i<=n;i++) scanf("%d",&f[i][0]);
14
    pre(n);
15
    while(m--) {
16
        int l,r;
17
        scanf("%d%d",&l,&r);
18
        printf("%d\n",query(l,r));
19
    }
20
    return 0;
21
}

3 基础搜索#

3.1 DFS#

全排列（搜索回溯）

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1<<5;
4
int n,reg[N],vis[N];
5
void print() {
6
    for(int i=1;i<=n;i++) printf("%d ",reg[i]);
7
    printf("\n");
8
}
9
void dfs(int x) {
10
    if(x>n) { print();return; }
11
    for(int i=1;i<=n;i++)
12
        if(!vis[i]) {
13
            vis[i]=1,reg[x]=i;
14
            dfs(x+1);
15
            vis[i]=0;
16
        }
17
}
18
int main() {
19
    scanf("%d",&n);
20
    dfs(1);
21
}

八皇后

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=10;
4
int n,reg[N];
5
bool row[N],le1[N],le2[N<<1];
6
void print() {
7
    for(int i=1;i<=n;i++,puts(""))
8
        for(int j=1;j<=n;j++)
9
            printf("%c",reg[i]==j?'Q':'.');
10
    puts("");
11
}
12
void dfs(int i) {
13
    if(i>n) { print();return; }
14
    for(int j=1;j<=n;j++) {
15
        if(row[j]||le1[i-j+n]||le2[i+j]) continue;
16
        reg[i]=j;
17
        row[j]=le1[i-j+n]=le2[i+j]=1;
18
        dfs(i+1);
19
        row[j]=le1[i-j+n]=le2[i+j]=0;
20
    }
21
}
22
int main() {
23
    scanf("%d",&n);
24
    dfs(1);
25
}

3.2 BFS#

迷宫问题

平面的遍历一般使用方向向量（注意是四联通还是八联通）

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=110;
4

5
// 四联通方向向量
6
const int dx[4]={0,0,1,-1};
7
const int dy[4]={1,-1,0,0};
8

9
int n,m,mp[N][N],dis[N][N];
10
queue<pair<int,int> > q;
11

12
//判断边界
13
bool inBound(int x,int y) { return x>0&&x<=n&&y>0&&y<=m; }
14

15
void bfs(int sx,int sy) {
16
    dis[sx][sy]=0;
17
    q.push({sx,sy});
18
    while(!q.empty()) {
19
        pair<int,int> p=q.front(); q.pop();
20
        int x=p.first,y=p.second;
21
        for(int i=0;i<4;i++) {
22
            int xx=x+dx[i],yy=y+dy[i];
23
            if(!inBound(xx,yy)||mp[xx][yy]) continue;
24
            if(dis[xx][yy]>dis[x][y]+1) {
25
                dis[xx][yy]=dis[x][y]+1;
26
                q.push({xx,yy});
27
            }
28
            if(xx==ex&&yy==ey) return;
29
        }
30
    }
31
}
32
int main() {
33
    scanf("%d%d",&n,&m);
34
    memset(dis,0x3f3f3f,sizeof dis);
35
    for(int i=1;i<=n;i++)
36
        for(int j=1;j<=m;j++)
37
            scanf("%d",&mp[i][j]);
38
    bfs(1,1);
39
    printf("%d",dis[ex][ey]);
40
    return 0;
41
}

3.3 常用搜索剪枝方法#

可行性剪枝：判断继续搜索是否能得到答案，如果不能，就返回
排除等效冗余：在搜索的几个分支中具有完全相同的效果时，选择其中一个走即可
最优性剪枝：当前搜索到的解已经超过最优解，就返回
顺序剪枝：优化搜索顺序，更快得到解
记忆化：记录每个状态的搜索结果，在后续搜索过程中检索这个状态，可直接使用

4 基础图论#

4.1 拓扑排序#

1
queue<int> q;
2
void topsort() {
3
    while(!q.empty()) {
4
        int x=q.front(); q.pop();
5
        reg[++cnt]=x;
6
        for(int i=head[x];i;i=nxt[i]) {
7
            int y=ver[i];
8
            if(!--deg[y]) q.push(y);
9
        }
10
    }
11
}
12
int main() {
13
    scanf("%d%d",&n,&m);
14
    for(int i=1;i<=m;i++) {
15
        int x,y;
16
        scanf("%d%d",&x,&y);
17
        add(x,y); deg[y]++;
18
    }
19
    for(int i=1;i<=n;i++)
20
        if(!deg[i]) q.push(i);
21
    topsort();
22
    if(cnt==n)
23
        for(int i=1;i<=n;i++) printf("%d ",reg[i]);
24
    else printf("-1");
25
}

4.2 Dijkstra 单源最短路#

1
priority_queue<PII> q;
2
void dijkstra(int st) {
3
    q.push({0,st});
4
    dis[st]=0;
5
    while(!q.empty()) {
6
        auto p=q.top(); q.pop();
7
        int x=p.second;
8
        if(vis[x]) continue;
9
        vis[x]=1;
10
        for(int i=head[x];i;i=nxt[i]) {
11
            int y=ver[i],z=edg[i];
12
            if(dis[y]>dis[x]+z) {
13
                dis[y]=dis[x]+z;
14
                q.push({-dis[y],y});
15
            }
16
        }
17
    }
18
}

4.3 SPFA#

4.3.1 最短路#

1
queue<int> q;
2
void spfa(int st) {
3
    q.push(st);
4
    vis[st]=1,dis[st]=0;
5
    while(!q.empty()) {
6
        int x=q.front(); q.pop();
7
        vis[x]=0;
8
        for(int i=head[x];i;i=nxt[i]) {
9
            int y=ver[i],z=edg[i];
10
            if(dis[y]>dis[x]+z) {
11
                dis[y]=dis[x]+z;
12
                if(!vis[y]) {
13
                    vis[y]=1;
14
                    q.push(y);
15
                }
16
            }
17
        }
18
    }
19
}

4.3.2 负环问题#

将所有点压入队列，跑spfa时记录路径长度

超过总点数 $n$ 则一定在环上跑了不止一次，即存在负环

1
void spfa() {
2
    for(int i=1;i<=n;i++)
3
        vis[i]=1,q.push(i);
4
    while(!q.empty()) {
5
        int x=q.front(); q.pop();
6
        vis[x]=0;
7
        for(int i=head[x];i;i=nxt[i]) {
8
            int y=ver[i],z=edg[i];
9
            if(dis[y]>dis[x]+z) {
10
                dis[y]=dis[x]+z;
11
                cnt[y]=cnt[x]+1;
12
                if(cnt[y]>=n) {
13
                    printf("Yes");
14
                    exit(0);
15
                }
16
                if(!vis[y]) {
17
                    vis[y]=1;
18
                    q.push(y);
19
                }
20
            }
21
        }
22
    }
23
}

4.3.3 Bellman-Ford 边数限制最短路#

1
int n,m,k,dis[N],pre[N];
2
struct Edge{int x,y,z;} e[M];
3
void Bellman_Ford() {
4
    memset(dis,0x3f,sizeof dis);
5
    dis[1]=0;
6
    for(int i=1;i<=k;i++) {
7
        memcpy(pre,dis,sizeof dis);
8
        for(int j=1;j<=m;j++) {
9
            int x=e[j].x,y=e[j].y,z=e[j].z;
10
            dis[y]=min(dis[y],pre[x]+z);
11
        }
12
    }
13
}
14
int main() {
15
    scanf("%d%d%d",&n,&m,&k);  // k为最短路边数限制
16
    for(int i=1;i<=m;i++)
17
        scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].z);
18
    Bellman_Ford();
19
}

4.4 Floyd 最短路#

1
void Floyd() {
2
    for(int k=1;k<=n;k++)
3
        for(int i=1;i<=n;i++)
4
            for(int j=1;j<=n;j++)
5
                dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
6
}
7
int main() {
8
    scanf("%d%d",&n,&m);
9
    memset(dis,0x3f3f3f,sizeof dis);
10
    for(int i=1;i<=n;i++) dis[i][i]=0;
11
    while(m--) {
12
        scanf("%d%d%d",&x,&y,&z);
13
        dis[x][y]=min(dis[x][y],z);    // 防止重边
14
    }
15
    Floyd();
16
}

4.5 最小生成树#

4.5.1 Kruskal#

适用于稀疏图

1
struct Node{int x,y,z;} e[N<<1];
2
bool cmp(Node a,Node b) {return a.z<b.z;}
3
int find(int x) {
4
    if(x==fa[x]) return x;
5
    return fa[x]=find(fa[x]);
6
}
7
void Kruscal() {
8
    sort(e+1,e+m+1,cmp);
9
    for(int i=1;i<=m;i++) {
10
        int x=find(e[i].x),y=find(e[i].y);
11
        if(x==y) continue;
12
        fa[x]=y; tot++; ans+=e[i].z;
13
        if(tot==n-1) return;
14
    }
15
}
16
int main() {
17
    scanf("%d%d",&n,&m);
18
    for(int i=1;i<=n;i++) fa[i]=i;
19
    for(int i=1;i<=m;i++)
20
        scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].z);
21
    Kruscal();
22
    if(tot==n-1) printf("%d",ans);
23
    else puts("Error");
24
}

4.5.2 Prim#

适用于稠密图

1
priority_queue<pair<int,int> > q;
2
void prim() {
3
    memset(dis,0x3f3f3f,sizeof dis);
4
    dis[1]=0;
5
    q.push({0,1});
6
    while(!q.empty()&&cnt<n) {
7
        int d=-q.top().first;
8
        int x=q.top().second;
9
        q.pop();
10
        if(vis[x]) continue;
11
        vis[x]=1;
12
        cnt++,ans+=d;
13
        for(int i=head[x];i;i=nxt[i]) {
14
            int y=ver[i],z=edg[i];
15
            if(dis[y]>z) {
16
                dis[y]=z;
17
                q.push({-dis[y],y});
18
            }
19
        }
20
    }
21
}
22
int main() {
23
    scanf("%d%d",&n,&m);
24
    while(m--) {
25
        int x,y,z;
26
        scanf("%d%d%d",&x,&y,&z);
27
        add(x,y,z); add(y,x,z);
28
    }
29
    prim();
30
    if(cnt==n) printf("%d",ans);
31
    else puts("Illegal");
32
}

4.6 二分图#

4.6.1 二分图判定（染色法）#

1
void dfs(int x) {
2
    for(int i=head[x];i;i=nxt[i]) {
3
        int y=ver[i];
4
        if(f[y]==INF) f[y]=f[x]^1,dfs(y);
5
        else if(f[y]==f[x]) {printf("No");exit(0);}
6
    }
7
}
8
int main() {
9
    memset(f,0x3f3f3f,sizeof f);
10
    scanf("%d%d",&n,&m);
11
    for(int i=1;i<=m;i++) {
12
        int x,y;
13
        scanf("%d%d",&x,&y);
14
        add(x,y); add(y,x);
15
    }
16
    for(int i=1;i<=n;i++) if(f[i]==INF) f[i]=0,dfs(i);
17
    printf("Yes");
18
}

4.6.2 二分图匹配（匈牙利算法）#

（又名找对象算法）

寻找最大匹配

1
/* …… */
2
int find(int x) {
3
    for(int i=head[x];i;i=nxt[i]) {
4
        int y=ver[i];
5
        if(vis[y]) continue;
6
        vis[y]=1;
7
        if(!p[y]||find(p[y]))   //对方还没对象或者对方的对象有备胎
8
            return p[y]=x;      //找到了对象
9
    }
10
    return 0;
11
}
12
int main() {
13
    scanf("%d%d%d",&n1,&n2,&m);
14
    while(m--) {
15
        scanf("%d%d",&x,&y);
16
        add(x,y);
17
    }
18
    for(int i=1;i<=n1;i++) {
19
        memset(vis,0,sizeof vis);
20
        if(find(i)) ans++;
21
    }
22
    printf("%d",ans);
23
}

4.7 LCA（最近公共祖先）#

1
t=log2(n)+1;
2
void dfs(int x,int fa) {
3
    f[x][0]=fa,d[x]=d[fa]+1;
4
    for(int i=1;i<=t;i++) f[x][i]=f[f[x][i-1]][i-1];
5
    for(int i=head[x];i;i=nxt[i]) if(ver[i]^fa) dfs(ver[i],x);
6
}
7
int lca(int x,int y) {
8
    if(d[x]>d[y]) x^=y^=x^=y;
9
    for(int i=t;i>=0;i--) if(d[f[y][i]]>=d[x]) y=f[y][i];
10
    if(x==y) return x;
11
    for(int i=t;i>=0;i--) if(f[x][i]^f[y][i]) x=f[x][i],y=f[y][i];
12
    return f[x][0];
13
}

5 基础数学#

5.1 质数#

5.1.1 试除法判质数#

重要性质：素数的形式均为 $6x+1$ 或 $6x+5$

1
bool isPrime(int x) {  // 加快判断
2
    if(x<=3) return x>1;
3
    else if(x%6!=1&&x%6!=5) return 0;
4
    else for(int i=5;i<=sqrt(x);i+=6)
5
        if(!(x%i)||!(x%(i+2))) return 0;
6
    return 1;
7
}
8
bool isPrime2(int x) {  // 朴素判断
9
    if(x<2) return 0;
10
    for(int i=2;i<=sqrt(x);i++)
11
        if(!(x%i)) return 0;
12
    return 1;
13
}

5.1.2 线性筛素数#

若 $i$ 整除 $p_j$ ，则 $i \times p_k (k>j)$ 一定被 $p_j$ 乘某个数筛掉，即

若 $p_j|i$ ，则对于 $\forall k>j$ , $\exists \lambda’$ 使得 $ip_k=\lambda’ p_j$

Important

线性筛的关键不在“筛掉合数”，而在“每个合数只被它的最小质因子筛一次”。因此遇到 i % p[j] == 0 后必须 break。

简单证明：

$p$ 数组单调递增，设 $i=\lambda p_j$ ，则 $i p_k=(\lambda p_j) p_k=(\lambda p_k) p_j=\lambda’ p_j$

1
void prime() {
2
    for(int i=2;i<=n;i++) {
3
        if(!vis[i]) p[++cnt]=i;
4
        for(int j=1;j<=cnt&&i*p[j]<=n;j++) {
5
            vis[i*p[j]]=1;
6
            if(i%p[j]==0) break;
7
        }
8
    }
9
}

5.1.3 分解质因数#

不需要筛素数，直接分解即可

1
void part(int x){
2
    for(int i=2;i<=x/i;i++){
3
        if(!(x%i)) {
4
            int cnt=0;
5
            while(!(x%i)) x/=i,cnt++;
6
            reg[++tot]={i,cnt};
7
        }
8
    }
9
    if(x>1) reg[++tot]={x,1};
10
}

5.1.4* 短区间二次筛法#

【例】给定两个整数 $L$ 和 $R$ ，在闭区间 $[L,R]$ 内找到距离最接近的两个相邻质数 $C_1$ 和 $C_2$ 和距离最远的两个相邻质数 $D_1$ 和 $D_2$ ，若存在相同距离的其他相邻质数对，则均输出第一对。

$1 \leq L < R \leq 2^{31}-1$

$1 \le R-L \leq 10^6$

先筛小范围质数，再筛 $[L,R]$ 中的质数

（ $n$ 是合数，则 $n$ 一定有 $\sqrt{n}$ 以内的质因子）

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1e6+10;
4
typedef long long ll;
5
int n,l,r,mi,ma,p[N],pp[N],tot,cnt;
6
bool vis[N],st[N];
7
void prime(int m) {
8
    vis[1]=1;
9
    for(int i=2;i<=m;i++) {
10
        if(!vis[i]) p[++tot]=i;
11
        for(int j=1;j<=tot&&p[j]<=m/i;j++) {
12
            vis[i*p[j]]=1;
13
            if(i%p[j]==0) break;
14
        }
15
    }
16
}
17
int main() {
18
    prime(50000);
19
    while(scanf("%d%d",&l,&r)!=EOF) {
20
        memset(st,0,sizeof st);
21
        cnt=0,mi=ma=1;
22
        for(int i=1;i<=tot;i++)
23
            for(int j=max(2ll,((ll)l+p[i]-1)/p[i]);j<=(ll)r/p[i];j++)
24
                st[j*p[i]-l]=1;
25
        for(int i=0;i<=r-l;i++)
26
            if(!st[i]&&l+i>=2)
27
                pp[++cnt]=i+l;
28
        if(cnt<2) {
29
            puts("There are no adjacent primes.");
30
            continue;
31
        }
32
        for(int i=2;i<cnt;i++) {
33
            if(pp[i+1]-pp[i]<pp[mi+1]-pp[mi]) mi=i;
34
            if(pp[i+1]-pp[i]>pp[ma+1]-pp[ma]) ma=i;
35
        }
36
        printf("%d,%d are closest, %d,%d are most distant.\n",pp[mi],pp[mi+1],pp[ma],pp[ma+1]);
37
    }
38
}

5.2 约数#

5.2.1 最大公约数 & 最小公倍数#

辗转相除法

1
int gcd(int x,int y) {return y?gcd(y,x%y):x;}
2
int lcm(int x,int y) {return (ll)x*y/gcd(x,y);}

5.2.2 约数个数#

设 $x$ 有 $n$ 个质因数 $a_i(1\leq i \leq n)$ ，每个质因数 $a_i$ 共有 $k_i$ 个，即 $a_i^{k_i}|x$ ，

则对于 $a_i$ ，有 $(k_i+1)$ 种类选择，即 $0,1,2,\cdots,k_i$

因此 $x$ 的因数个数为 $\prod_i{(k_i+1)}$

5.2.3 约数之和#

若 $N=\prod_{i}{p_i^{c_i}}$
约数个数： $\prod_{i}{(c_i+1)}$
约数之和： $\prod_{i}{\sum_{j=0}^{c_i}{p_i^j}}=\prod_i{\frac{p_i^{c_i+1}-1}{p_i-1}}$

5.3 逆元#

逆元公式推导详解

当求解公式： $(a/b)\bmod m$ 时，因 $b$ 可能会过大，会出现爆精度的情况，所以需变除法为乘法：

设 $b^{-1}$ 是 $b$ 的逆元，则有 $b\cdot b^{-1}\equiv1\pmod m$ ；

则

\begin{align} (a/b) &\equiv (a/b)\cdot 1 \pmod m \\\\ &\equiv (a/b)\cdot b\cdot b^{-1} \pmod m \\\\ &\equiv a\cdot b^{-1} \pmod m \end{align}

即 $a/b$ 的模等于 $a\cdot b^{-1}$ 的模；

5.3.1 费马小定理#

a^p≡a\mod p

也写作

a^{p-1}≡1\mod p

若 $p$ 是素数，且 $inv$ 与 $p$ 互质，则

inv^{p-1}≡(1\%p)

根据逆元的定义可得

x*inv≡1\mod p

得出乘法逆元

inv=x^{p-2}

前提： $p$ 是质数

1
const int mod=1e9+7;
2
ll ksm(ll a,ll b) {
3
    ll ret=1;
4
    for(;b;b>>=1,(a*=a)%=mod) if(b&1) (ret*=a)%=mod;
5
    return ret;
6
}
7
ll inv(ll a) {
8
  return ksm(a,mod-2);
9
}

5.3.2 扩展欧几里得#

对于不完全为 $0$ 的非负整数 $a,b$ ，必然存在整数对 $x,y$ ，使得 $gcd(a,b)=ax+by$ 。

证明：

当 $b=0$ 时， $gcd(a,b)=a$ ，因此 $x=1,y=0$
当 $b \neq 0$ 时， $gcd(a,b)=gcd(b,a\%b)$ ，则
$\begin{align} ax+by &= gcd(a,b)=gcd(b,a\%b)=bx’+(a\%b)y’ \\\\ &= bx’+(a-b*\lfloor a/b\rfloor)y’ \\\\ &= ay’+b(x’-\lfloor a/b\rfloor)y’ \end{align}$
可得：
$\begin{cases} x=y’ \\\\ y=x’-\lfloor a/b\rfloor \end{cases}$

1
ll exgcd(ll a, ll b, ll &x, ll &y) {
2
    if (!b) { x=1,y=0; return a; }
3
    else {
4
        ll r=exgcd(b,a%b,y,x);
5
        y-=x*(a/b);
6
        return r;
7
    }
8
}
9
ll inv(ll a, ll mod) {
10
    ll x,y;
11
    exgcd(a,mod,x,y);
12
    x=(x%mod+mod)%mod;
13
    return x;
14
}

5.3.3 线性求逆元#

已知一个质数 $M$ ，求出 $1→n$ 中每个数关于模 $M$ 的逆元

inv[i]=(M-\lfloor{M/i}\rfloor)*inv[M\%i]%M

1
void init_inv(int n,int p) {
2
    inv[1]=1;
3
    for(int i=2;i<=n;i++) {
4
      inv[i]=(ll)(p-p/i)*inv[p%i]%p;
5
      printf("%d\n",inv[i]);
6
  }
7
}

5.4 组合数#

5.4.1 递推法#

C_n^m=C_{n-1}^{m-1}+C_{n-1}^{m}

1
void init(int n) {
2
    for(int i=0;i<=n;i++)
3
    for(int j=0;j<=i;j++)
4
      C[i][j]=j?(C[i-1][j-1]+C[i-1][j])%mod:1;
5
}

5.4.2 公式法#

C_n^m=\frac{n!}{m!(n-m)!}

线性预处理阶乘和阶乘的逆元

1
const int mod=1e9+7,N=1e5+10;
2
typedef long long ll;
3
ll fac[N]={1ll},inv[N]={1ll};
4
ll ksm(ll a,int k) {
5
    ll res=1ll;
6
    for(;k;(a*=a)%=mod,k>>=1) if(k&1) (res*=a)%=mod;
7
    return res;
8
}
9
void init(int n) {
10
    for(int i=1;i<=n;i++) {
11
        fac[i]=fac[i-1]*i%mod;
12
        inv[i]=inv[i-1]*ksm(i,mod-2)%mod;
13
    }
14
}
15
ll C(ll a,ll b) {
16
    return (ll)fac[a]*inv[b]%mod*inv[a-b]%mod;
17
}

5.4.3 Lucas定理#

C(n,m)=C(n/p,m/p)∗C(n\%p,m\%p)

适用于 $n,m$ 较大时求组合数

1
ll ksm(ll a,int k) {
2
    ll res=1ll;
3
    for(;k;(a*=a)%=mod,k>>=1) if(k&1) (res*=a)%=mod;
4
    return res;
5
}
6
ll C(ll a,ll b) {
7
    if(a<b) return 0;
8
    if(a<mod&&b<mod) return (fac[a]*inv[b]%mod)*(inv[a-b]%mod)%mod;
9
    else return C(a/mod,b/mod)*C(a%mod,b%mod)%mod;
10
}
11
void pre() {
12
    for(int i=1;i<=1e5;i++) {
13
        fac[i]=fac[i-1]*i%mod;
14
        inv[i]=inv[i-1]*ksm(i,mod-2)%mod;
15
    }
16
}
17
int main() {
18
    scanf("%d",&m);
19
    while(m--) {
20
        scanf("%lld%lld%d",&a,&b,&mod);
21
        pre();
22
        printf("%lld\n",C(a,b));
23
    }
24
}

5.4.4 质因数分解法#

$n!$ 中质因数 $p$ 的个数，即统计 $p$ 的出现次数，再统计 $p^2$ 的出现次数，依次类推，即

\begin{align} \text{cnt}(n,p) &=\lfloor \frac{n}{p} \rfloor + \lfloor \frac{n}{p^2} \rfloor + \lfloor \frac{n}{p^3} \rfloor + \cdots \\\\ &= \lfloor \frac{n}{p} \rfloor + \lfloor \frac{n/p}{p} \rfloor + \lfloor \frac{n/p^2}{p} \rfloor + \cdots \end{align} \\\\

C^b_a=\frac{a!}{b!(a−b)!} = \prod_i{p_i^{\text{cnt}(a,p_i)-\text{cnt}(b,p_i)-\text{cnt}(a-b,p_i)}}

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=5010;
4
int a,b,tot,p[N],cnt[N];
5
bool vis[N];
6
vector<int> mul(vector<int> a,int b) { // 高精乘
7
    vector<int> ret;
8
    int t=0;
9
    for(int i=0;i<a.size();i++) {
10
        t+=a[i]*b;
11
        ret.push_back(t%10);
12
        t/=10;
13
    }
14
    while(t) ret.push_back(t%10),t/=10;
15
    while(ret.size()>1&&ret.back()==0) ret.pop_back();
16
    return ret;
17
}
18
void prime(int n) {     // 线性筛素数
19
    for(int i=2;i<=n;i++) {
20
        if(!vis[i]) p[++tot]=i;
21
        for(int j=1;j<=tot&&p[j]<=n/i;j++) {
22
            vis[p[j]*i]=1;
23
            if(!(i%p[j])) break;
24
        }
25
    }
26
}
27
int cal(int n,int p) {     // 计算cnt(n,p)
28
    int res=0;
29
    while(n) res+=n/p,n/=p;
30
    return res;
31
}
32
int main() {
33
    scanf("%d%d",&a,&b);
34
    prime(a);
35
    for(int i=1;i<=tot;i++)
36
        cnt[i]=cal(a,p[i])-cal(a-b,p[i])-cal(b,p[i]);
37
    vector<int> res;
38
    res.push_back(1);
39
    for(int i=1;i<=tot;i++)
40
        for(int j=0;j<cnt[i];j++)
41
            res=mul(res,p[i]);
42
    for(int i=res.size()-1;i>=0;i--) printf("%d",res[i]);
43
}

5.5 欧拉函数#

5.5.1 单点求欧拉函数#

$1∼N$ 中与 $N$ 互质的数的个数被称为欧拉函数，记为 $\varphi(N)$ 。
若在算数基本定理中， $N=p^{a_1}_1p^{a_2}_2 \cdots p^{a_m}_m$ 则：

\varphi(N) = N\times\frac{p_1−1}{p_1}\times \frac{p_2−1}{p_2}\times\cdots\times\frac{p_m−1}{p_m}

1
#include<bits/stdc++.h>
2
using namespace std;
3
int n,tmp;
4
int part(int x) {
5
    int res=x;
6
    for(int i=2;i<=x/i;i++) {
7
        if(x%i==0) (res/=i)*=i-1;
8
        while(x%i==0) x/=i;
9
    }
10
    if(x>1)  (res/=x)*=x-1;
11
    return res;
12
}
13
int main() {
14
    scanf("%d",&n);
15
    while(n--) {
16
        scanf("%d",&tmp);
17
        printf("%d\n",part(tmp));
18
    }
19
    return 0;
20
}

5.5.2* 筛法求欧拉函数#

1
void eular() {
2
    for(int i=2;i<=n;i++) {
3
        if(!vis[i]) p[++tot]=i,e[i]=i-1;
4
        for(int j=1;p[j]<=n/i;j++) {
5
            int t=i*p[j];
6
            vis[t]=1;
7
            if(!(i%p[j])) {
8
                e[t]=e[i]*p[j];
9
                break;
10
            }
11
            e[t]=e[i]*(p[j]-1);
12
        }
13
    }
14
}

5.6 高斯消元#

列主元法消元

每次选择一列中绝对值最大的元素作为主元，保证消元过程收敛

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=110;
4
const double eps=1e-8;
5
int n;
6
double a[N][N];
7
int gauss() {
8
    int c,r;
9
    for(c=0,r=0;c<n;c++,r++) {
10
        //选列主元
11
        int t=r;
12
        for(int i=r;i<n;i++)
13
            if(fabs(a[i][c])>fabs(a[t][c])) t=i;
14
        if(fabs(a[t][c])<eps) {r--;continue;}
15
        //交换主元所在行
16
        for(int i=c;i<=n;i++) swap(a[t][i],a[r][i]);
17
        //单位化
18
        double s=a[r][c]; a[r][c]=1.00;
19
        for(int i=c+1;i<=n;i++) a[r][i]/=s;
20
        //消元
21
        for(int i=r+1;i<n;i++)
22
            for(int j=c+1;j<=n;j++)
23
                a[i][j]-=a[r][j]*a[i][c];
24
    }
25
    if(r<n) {
26
        for(int i=r;i<n;i++)
27
            if(fabs(a[i][n])>eps) return -1; //无解
28
        return 0; //无穷多组解
29
    }
30
    //回代
31
    for(int i=n-1;i>=0;i--)
32
        for(int j=i+1;j<n;j++)
33
            a[i][n]-=a[i][j]*a[j][n];
34
    return 1; //唯一解
35
}
36
int main() {
37
    scanf("%d",&n);
38
    for(int i=0;i<n;i++)
39
        for(int j=0;j<=n;j++)
40
            scanf("%lf",&a[i][j]);
41
    int t=gauss();
42
    if(t==-1) puts("No solution");
43
    else if(!t) puts("Infinite group solutions");
44
    else {
45
        for(int i=0;i<n;i++) {
46
            if(fabs(a[i][n])<eps) a[i][n]=0;
47
            printf("%.2lf\n",a[i][n]);
48
        }
49
    }
50
    return 0;
51
}

6 贪心#

6.1 区间问题#

6.1.1 区间选点#

选多少个点使得每个区间至少有一个点：右端点排序

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,ans,minn=-1e9-10;
5
struct Line {int l,r;} r[N];
6
bool cmp(Line a,Line b) {return a.r<b.r;}
7
int main() {
8
    scanf("%d",&n);
9
    for(int i=1;i<=n;i++) scanf("%d%d",&r[i].l,&r[i].r);
10
    sort(r+1,r+n+1,cmp);
11
    for(int i=1;i<=n;i++)
12
        if(minn<r[i].l)
13
            ans++,minn=r[i].r;
14
    printf("%d\n",ans);
15
    return 0;
16
}

6.1.2 最多不相交区间数量#

选择尽量多的不相交的区间：右端点排序

代码同 6.1.1

6.1.3 区间覆盖#

选择尽量少的区间，将指定线段区间完全覆盖：左端点排序

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,st,ed,ans;
5
bool f;
6
struct Line {int l,r;} a[N];
7
bool cmp(Line a,Line b) {return a.l<b.l;}
8
int main() {
9
    scanf("%d%d%d",&st,&ed,&n);
10
    for(int i=1;i<=n;i++)
11
        scanf("%d%d",&a[i].l,&a[i].r);
12
    sort(a+1,a+n+1,cmp);
13
    for(int i=1;i<=n;i++) {
14
        int j,max_r=-2e9-10;
15
        for(j=i;j<=n&&a[j].l<=st;j++)
16
            max_r=max(max_r,a[j].r);
17
        if(max_r<st) {ans=-1;break;}
18
        ans++;
19
        if(max_r>=ed) {f=1;break;}
20
        st=max_r;
21
        i=j-1;
22
    }
23
    printf("%d\n",f?ans:-1);
24
    return 0;
25
}

6.1.4 区间分组#

将所给区间分为尽量少的若干组，组内区间不相交：左端点排序

每次向空余位置最靠左的分组内放区间

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=100010;
4
int n,m;
5
struct Line {int l,r;} line[N];
6
priority_queue<int,vector<int>,greater<int> > q;
7
bool cmp(Line a,Line b) {return a.l<b.l;}
8
int main() {
9
    scanf("%d",&n);
10
    for(int i=1;i<=n;i++)
11
        scanf("%d%d",&line[i].l,&line[i].r);
12
    sort(line+1,line+n+1,cmp);
13
    for(int i=1;i<=n;i++) {
14
        if(q.empty()||line[i].l<=q.top()) q.push(line[i].r);
15
        else {
16
            q.pop();
17
            q.push(line[i].r);
18
        }
19
    }
20
    printf("%d",q.size());
21
}

也可使用差分寻找区间重叠最厚处即为分组数

1
#include<bits/stdc++.h>
2
using namespace std;
3
map<int,int> d;
4
int n,s,ans;
5
int main() {
6
    scanf("%d",&n);
7
    while(n--) {
8
        int l,r;
9
        scanf("%d%d",&l,&r);
10
        d[l]++; d[r+1]--;
11
    }
12
    for(auto i:d) {
13
        s+=i.second;
14
        ans=max(ans,s);
15
    }
16
    printf("%d",ans);
17
    return 0;
18
}

以及一些奇技淫巧？

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1e5+5;
4
int n,a[N],b[N],j=1,ans;
5
int main() {
6
    scanf("%d",&n);
7
    for(int i=1;i<=n;i++) scanf("%d%d",a+i,b+i);
8
    sort(a+1,a+n+1); sort(b+1,b+n+1);
9
    for(int i=1;i<=n;i++)
10
        (a[i]<=b[j])?ans++:j++;
11
    printf("%d\n",ans);
12
}

6.2 排序不等式#

逆序 $\cdot$ 正序 $\leq$ 乱序 $\cdot$ 正序 $\leq$ 正序 $\cdot$ 正序

排队问题，最短时间优先原理，快者优先

【例】 $n$ 个人排队打水，每个人的打水时间为 $a_i$ ，求最小等待时间之和。

1
#include<bits/stdc++.h>
2
long long n,ans,a[100010];
3
int main() {
4
    scanf("%lld",&n);
5
    for(int i=1;i<=n;i++) scanf("%lld",a+i);
6
    sort(a+1,a+n+1);
7
    for(int i=1;i<=n;i++) a[i]+=a[i-1],ans+=a[i-1];
8
    printf("%lld",ans);
9
}

6.3 二分答案+贪心验证#

当题目中有使最小值最大或使最大值最小的意图时，考虑二分答案

二分答案，将答案进行可行性验证（多数情况为贪心验证）

二分答案的使用条件：答案要随着决策的增减而单调增/减

【例】跳石头

1
#include<bits/stdc++.h>
2
using namespace std;
3
int l,n,m,d[50010],ans;
4
bool check(int x) {
5
  int t=0,cnt=0;
6
  for(int i=1;i<=n+1;i++) {
7
    if(d[i]-t<x) cnt++;
8
    else t=d[i];
9
  }
10
  return cnt<=m;
11
}
12
int main() {
13
  scanf("%d%d%d",&l,&n,&m);
14
  for(int i=1;i<=n;i++) scanf("%d",d+i);
15
  d[n+1]=l;
16
  int L=1,R=l;
17
  while(L<=R) {
18
    int mid=(L+R)>>1;
19
    if(check(mid)) ans=mid,L=mid+1;
20
    else R=mid-1;
21
  }
22
  printf("%d",ans);
23
  return 0;
24
}

7 动态规划（dp）#

8 背包问题#

背包问题作为动态规划的一类重要问题，单独展开。

8.1 01背包#

$N$ 件物品，每件物品 $i$ 体积为 $c_i$ ，价值为 $w_i$
一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

朴素解法：

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1010,M=1010;
4
int n,m,c[N],w[N],f[N][M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) scanf("%d%d",c+i,w+i);
8
    for(int i=1;i<=n;i++)
9
        for(int j=0;j<=m;j++) {
10
            f[i][j]=f[i-1][j];
11
            if(j>=c[i]) f[i][j]=max(f[i][j],f[i-1][j-c[i]]+w[i]);
12
        }
13
    printf("%d",f[n][m]);
14
}

改变遍历顺序，滚动数组空间优化

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int M=1010;
4
int n,m,f[M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        int c,w;
9
        scanf("%d%d",&c,&w);
10
        for(int j=m;j>=c;j--)
11
            f[j]=max(f[j],f[j-c]+w);
12
    }
13
    printf("%d",f[m]);
14
}

8.2 完全背包#

$N$ 种物品，每种物品 $i$ 体积为 $c_i$ ，价值为 $w_i$ ，每种物品有无限件
一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

与01背包相比，每件物品可以装无限次，因此只需改变一下转移方程。

朴素解法：

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=1010,M=1010;
4
int n,m,c[N],w[N],f[N][M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) scanf("%d%d",c+i,w+i);
8
    for(int i=1;i<=n;i++)
9
        for(int j=0;j<=m;j++) {
10
            f[i][j]=f[i-1][j];
11
            if(j>=c[i]) f[i][j]=max(f[i][j],f[i][j-c[i]]+w[i]);  // 与01背包的差别
12
        }
13
    printf("%d",f[n][m]);
14
}

滚动数组优化与01背包的差别在于体积的遍历顺序：

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int M=1010;
4
int n,m,f[M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        int c,w;
9
        scanf("%d%d",&c,&w);
10
        for(int j=c;j<=m;j++)  // 与01背包的差别
11
            f[j]=max(f[j],f[j-c]+w);
12
    }
13
    printf("%d",f[m]);
14
}

8.3 多重背包#

$N$ 种物品，每种物品 $i$ 体积为 $c_i$ ，价值为 $w_i$ ，每种物品有 $s_i$ 件
一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

8.3.1 朴素解法#

遍历每件物品，对每件物品进行决策

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int M=101;
4
int n,m,f[M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        int c,w,s;
9
        scanf("%d%d%d",&c,&w,&s);
10
        for(int k=1;k<=s;k++)
11
            for(int j=m;j>=c;j--)
12
                f[j]=max(f[j],f[j-c]+w);
13
    }
14
    printf("%d",f[m]);
15
}

8.3.2 二进制拆分#

我们对于每个相同种类的物品都进行了一次遍历，而实际上这是冗余的。

我们可以将每种物品的个数进行二进制拆分，这样即可囊括每种物品选择 $0 \to s_i$ 件的情况

最后将问题转化为01背包

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int M=2010;
4
int n,m,f[M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        int c,w,s,k=1;
9
        scanf("%d%d%d",&c,&w,&s);
10
        for(int k=1;s>=k;s-=k,k<<=1)
11
            for(int j=m;j>=c*k;j--)
12
                f[j]=max(f[j],f[j-k*c]+k*w);
13
        if(s)
14
            for(int j=m;j>=s*c;j--)
15
                f[j]=max(f[j],f[j-s*c]+s*w);
16
    }
17
    printf("%d",f[m]);
18
}

8.4 混合背包#

$N$ 类物品，每类物品 $i$ 体积为 $c_i$ ，价值为 $w_i$ ，以及一个属性 $s_i$

$s_i=-1$ 表示该类物品仅有一件

$s_i=0$ 表示该类物品有无限件

$s_i>0$ 表示该类物品有 $s_i$ 件

一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=10010,M=1010;
4
int n,m,f[M];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        int c,w,s;
9
        scanf("%d%d%d",&c,&w,&s);
10
        if(!s) {        // 完全背包
11
            for(int j=c;j<=m;j++)
12
                f[j]=max(f[j],f[j-c]+w);
13
        } else if(s<0||s==1) {  // 01背包
14
            for(int j=m;j>=c;j--)
15
                f[j]=max(f[j],f[j-c]+w);
16
        } else {        // 多重背包
17
            for(int k=1;s>=k;s-=k,k<<=1)
18
                for(int j=m;j>=c*k;j--)
19
                    f[j]=max(f[j],f[j-k*c]+k*w);
20
            if(s)
21
                for(int j=m;j>=s*c;j--)
22
                    f[j]=max(f[j],f[j-s*c]+s*w);
23
        }
24
    }
25
    printf("%d",f[m]);
26
}

8.5 分组背包#

$N$ 组物品，每组中有 $s_i$ 件
第 $i$ 组中的物品 $j$ 体积为 $c_{ij}$ ，价值为 $w_{ij}$ ，同一组内的物品只能选一个
一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

对于一个组内的物品进行枚举，关键在于组内只能选择一个

因此需要注意枚举背包体积和组内决策的内外层关系

若内外层枚举顺序反置，则失去了组内约束，变成了普通01背包

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=101;
4
int n,m,c[N][N],w[N][N],s[N],f[N];
5
int main() {
6
    scanf("%d%d",&n,&m);
7
    for(int i=1;i<=n;i++) {
8
        scanf("%d",s+i);
9
        for(int j=1;j<=s[i];j++)
10
            scanf("%d%d",&c[i][j],&w[i][j]);
11
    }
12
    for(int i=1;i<=n;i++)
13
        for(int j=m;j>=0;j--)
14
            for(int k=1;k<=s[i];k++)
15
                if(j>=c[i][k])
16
                    f[j]=max(f[j],f[j-c[i][k]]+w[i][k]);
17
    printf("%d",f[m]);
18
}

8.6 二维费用背包#

$N$ 件物品，每件物品 $i$ 体积为 $c_i$ ，重量为 $w_i$ ，价值为 $v_i$
一个容积为 $V$ ，最大承重为 $M$ 的背包，使得总价值最大

1
#include<bits/stdc++.h>
2
using namespace std;
3
int n,m1,m2,c1,c2,w,f[110][110];
4
int main() {
5
    scanf("%d%d%d",&n,&m1,&m2);
6
    for(int i=1;i<=n;i++) {
7
        scanf("%d%d%d",&c1,&c2,&w);
8
        for(int j=m1;j>=c1;j--)
9
            for(int k=m2;k>=c2;k--)
10
                f[j][k]=max(f[j][k],f[j-c1][k-c2]+w);
11
    }
12
    printf("%d",f[m1][m2]);
13
}

8.7 有依赖的背包#

$N$ 件物品，每件物品 $i$ 体积为 $c_i$ ，重量为 $w_i$ ，价值为 $v_i$ ，其依赖物品编号为 $p_i$
即只有购置了 $p_i$ 物品，才能购置 $i$ 物品（ $p_i$ 是 $i$ 的前驱）
一个容量为 $M$ 的背包，物品总体积不超过背包容量，且总价值最大

树形结构，使用记忆化搜索更新

1
#include<bits/stdc++.h>
2
using namespace std;
3
const int N=110;
4
int n,m,r,f[N][N],c[N],w[N];
5
vector<int> e[N];
6
void dp(int x) {
7
    for(int i=c[x];i<=m;i++) f[x][i]=w[x];  // 必须买自己，才能买其附属物品
8
    for(int y:e[x]) {
9
        dp(y);
10
        for(int j=m;j>=c[x];j--)    // 枚举装入后的体积
11
            for(int k=0;k<=j-c[x];k++)    // 枚举新装入的子物品体积
12
                f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
13
    }
14
}
15
int main() {
16
    scanf("%d%d",&n,&m);
17
    for(int i=1;i<=n;i++) {
18
        int p;
19
        scanf("%d%d%d",c+i,w+i,&p);
20
        if(p==-1) r=i;
21
        else e[p].push_back(i);
22
    }
23
    dp(r);
24
    printf("%d",f[r][m]);
25
}

基础算法模版

基础算法模板#

1 基础算法#

1.1 归并排序#

1.2 快速排序#

1.3 二分#

1.4 前缀和 & 差分#

1.5 有序数组定位#

1.5.1 lower_bound & upper_bound#

1.5.2 离散化#

1.6 位运算#

1.6.1 数字二进制位数#

1.6.2 数字二进制中’1’的个数#

1.6.3 二进制枚举#

1.7 高精度#

1.8 KMP#

1.9 Hash#

1.9.1 拉链法#

1.9.2 开放寻址法（线性探查法）#

1.9.3 字符串Hash#

1.10 快速乘 快速幂#

2 基础数据结构#

2.1 单链表#

2.2 双链表#

2.3 栈#

2.3.1 模拟栈#

2.3.2 STL stack#

2.3.3* 单调栈#

2.4 队列#

2.4.1 模拟队列#

2.4.2 STL queue/deque#

2.4.3 单调队列#

2.4.4 优先队列#

2.5 堆#

2.5.1 模拟堆#

2.5.2 STL priority_queue#

2.5.3 堆排序#

2.6 并查集#

2.7 邻接表（前向星）#

2.8 Trie字典树#

2.9 哈夫曼树（Huffman）#

2.10* ST表#

3 基础搜索#

3.1 DFS#

3.2 BFS#

3.3 常用搜索剪枝方法#

4 基础图论#

4.1 拓扑排序#

4.2 Dijkstra 单源最短路#

4.3 SPFA#

4.3.1 最短路#

4.3.2 负环问题#

4.3.3 Bellman-Ford 边数限制最短路#

4.4 Floyd 最短路#

4.5 最小生成树#

4.5.1 Kruskal#

4.5.2 Prim#

4.6 二分图#

4.6.1 二分图判定（染色法）#

4.6.2 二分图匹配（匈牙利算法）#

4.7 LCA（最近公共祖先）#

5 基础数学#

5.1 质数#

5.1.1 试除法判质数#

5.1.2 线性筛素数#

5.1.3 分解质因数#

5.1.4* 短区间二次筛法#

5.2 约数#

5.2.1 最大公约数 & 最小公倍数#

5.2.2 约数个数#

5.2.3 约数之和#

5.3 逆元#

5.3.1 费马小定理#

5.3.2 扩展欧几里得#

5.3.3 线性求逆元#

5.4 组合数#

5.4.1 递推法#

5.4.2 公式法#

5.4.3 Lucas定理#

5.4.4 质因数分解法#

1.10 快速乘快速幂#