c++常见算法模板

快速排序

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;
int n;
int a[N];

void quick_sort(int l,int r)
{
	if(l >= r) return;
	int x = a[(l + r) >>1];
	int i = l - 1, j = r + 1;
	while(i < j)
	{
		do i ++; while(a[i] < x);
		do j --; while(a[j] > x);
		if(i < j) swap(a[i],a[j]);
	}
	quick_sort(l,j);
	quick_sort(j + 1,r);
}
int main()
{
	cin >> n;
	for(int i = 0; i < n; i ++) cin >> a[i];
	quick_sort(0,n-1);
	for(int i = 0; i < n; i ++) cout << a[i] << ' ';
	return 0;
}

归并排序

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;
int q[N];
int n;

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 i = l, j = mid + 1,k = 0;
	int a[N];
	while(i <= mid && j <= r){
		if(q[i] <= q[j]) a[k ++] = q[i ++];
		else a[k ++] = q[j ++];
	}
	while(i <= mid){
		a[k ++] = q[i ++];
	}
	while(j <= r){
		a[k ++] = q[j ++];
	}
	for(int i = l,j = 0;i <= r; i ++, j ++) q[i] = a[j];
}


int main(){
	cin >> n;
	for(int i = 0; i < n; i ++) cin >> q[i];
	merge_sort(q,0, n -1);
	for(int i = 0; i < n; i ++) cout << q[i] << " ";
	return 0;
}

二分模板

#include <bits.stdc++.h>


using namespace std;


/**
二分的第一个模板


void erfen()
{
	int l = 0, r = n - 1;
	while(l < r)
	{
		int mid = (l + r) >> 1;
		if(check(mid)) r = mid;
		else l = mid + 1;
	}
}


二分的第二个模板(边界问题)
void erfen()
{
	int l = 0, r = n - 1;
	while(l < r)
	{
		int mid = (l + r + 1) >> 1;
		if(check(mid)) l = mid;
		else r = mid - 1;
	}
}

前缀和

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;

int a[N],s[N];
int n,m;
int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) cin >> a[i];
	for(int i = 1; i <= n; i ++) s[i] = s[i - 1] + a[i];
	while(m -- )
	{
		int l,r;
		cin >> l >> r;
		cout << s[r] - s[l - 1] << endl;
	}
	return 0;
}

差分

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;

int a[N],b[N];
int n,m;

void insert(int l,int r,int c)
{
    b[l] += c;
    b[r + 1] -= c;
}
int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++) cin >> a[i];
    for(int i = 1; i <= n; i++) insert(i,i,a[i]);
    while(m --)
    {
    	int l,r,c;
    	cin >> l >> r >> c;
    	insert(l,r,c);
    }
    for(int i = 1;i <= n; i++) b[i] += b[i - 1];
    for(int i = 1;i <= n; i++) cout << b[i] << " ";
    return 0;
}

位运算

#include<bits/stdc++.h>

//n的2进制表示中，第k位是几
//lowbit(x) 返回x的最后一位1应用统计x里面1的个数
using namespace std;

int main(){
	int n = 10;
	for(int k = 3; k >= 0; k --) cout << (n >> k & 1);
	return 0;
}

lowbit操作

#include <bits/stdc++.h>
//用来统计x中最后一位1的个数
using namespace std;

int lowbit(int x)
{
	return x & -x;
}
int main()
{
	int n;
	cin >> n;
	while(n -- )
	{
		int x;
		cin >> x;

		int res = 0;
		while(x) x -= lowbit(x), res ++;
		cout << res << " ";
	}
	return 0;
}

kmp算法

#include<bits/stdc++.h>
using namespace std;

const int N = 100010,M = 100010;
int n,m;
char p[N],s[M];
int ne[N];


int main()
{
	cin >> n >> p + 1 >> m >> s + 1;

	//求ne过程
	for(int i = 2,j = 0; i <= n; i ++)
	{
		while(j && p[i] != p [j + 1]) j = ne[j];
		if(p[i] == p[j + 1]) j++;
		ne[i] = j;
	}

	//kmp匹配过程
	for(int i = 1,j = 0; i <= m; i ++)
	{
		while(j && s[i] != p[j + 1])
		{
			j = ne[j];
		}
		if(s[i] == p[j + 1]) j ++;
		if(j == n)
		{
			//匹配成功
			cout << i - n << ' ';
			j = ne[j];
		}
	}
	return 0;
}

滑动窗口

#include<bits/stdc++.h>
using namespace std;
//滑动窗口也叫作单调队列

//两个关键点一是维护窗口的长度而是判断入队的条件
const int N = 1000010;
int n,k;
int a[N],q[N],hh,tt = -1;

int main()
{
	cin.tie(0);
	cin >> n >> k;
	for(int i = 0; i < n; i ++) cin >> a[i];

	for(int i = 0; i < n; i ++)
	{
		//判断对头是否已经划出窗口
		if(hh <= tt && i - k + 1 > q[hh]) hh ++;
		while(hh <= tt && a[q[tt]] >= a[i]) tt--;
		q[++ tt] = i; 
		if(i >= k - 1) cout << a[q[hh]] << ' ';
	}
	cout << endl;
	hh = 0; tt = -1;

	for(int i = 0; i < n; i ++)
	{
		if(hh <= tt && i - k + 1 > q[hh]) hh ++;
		while(hh <= tt && a[q[tt]] <= a[i]) tt--;
		q[++ tt] = i;
		if(i >= k - 1) cout << a[q[hh]] << ' '; 
	}
	return 0;
}

单调栈

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;

int n;
int stk[N],tt;

int main()
{
	cin.tie(0);
	cin >> n;
	for(int i = 0; i < n; i ++)
	{
		int x;
		cin >> x;
		while(tt && stk[tt] >= x) tt--;
		if(tt) cout << stk[tt] << ' ';
		else cout << "-1" << ' ';

		stk[ ++ tt] = x;
	}
	return 0;
}

trie树

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
//Trie树的基本用法就是用来高效的存储和查找字符串集合的数据结构
//字母类型和字母个数不是很多
int son[N][26],cnt[N],idx;//下表是0的点既是根节点又是空节点
char str[N];
void insert(char str[])
{
	int p = 0;
	for(int i = 0; str[i]; i ++)
	{
		int u = str[i] - 'a';
		if(!son[p][u]) son[p][u] = ++ idx;
		p = son[p][u];
	}
	cnt[p] ++;
}

int query(char str[])
{
	int p = 0;
	for(int i = 0; str[i]; i ++)
	{
		int u = str[i] - 'a';
		if(!son[p][u]) return 0;
		p = son[p][u];
	}
	return cnt[p];
}
int main()
{
	int n;
	cin >> n;
	while(n -- )
	{
		char op[2];
		cin >> op >> str;
		if(op[0] == 'I') insert(str);
		else cout << query(str) << ' '; 
	}
	return 0;
}

并查集

#include<bits/stdc++.h>
using namespace std;

//并查集支持的操作第一个是将两个集合合并，第二个是询问两个元素是否在一个集合当中

//近乎o(1)的时间复杂度之内快速的支持这两个操作
/**
基本原理:每个集合用一棵树来表示，树根的编号就是集合的编号,每个节点存储他的父节点 */
//并查集的优化(路径压缩)
const int N = 100010;
int n,m;
int p[N];
int find(int x)//返回x所在集合的编号+路径压缩
{
	if(p[x] != x) p[x] = find(p[x]);
	return p[x];
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) p[i] = i;
	while(m -- )
	{
		char op[2];
		int a,b;
		cin >> op >> a >> b;
		if(op[0] == 'M') p[find(a)] = find(b);
		else 
		{
			if(find(a) == find(b)) puts("Yes");
			else puts("No");
		}
	}
	return 0;
}