链表
单链表
https://www.acwing.com/problem/content/828/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
//head:头节点的指向 e[i]:当前节点i的值 ne[i]:当前节点i的next指针 idx:当前存储的点
int head, e[N], ne[N], idx;
//初始化
void init(){
head = -1;
idx = 1;
}
//将x插入到头节点
void add_head(int x){
e[idx] = x;
ne[idx] = head;
head = idx;
idx++;
}
//将x插入到下标是k的后面
void add(int k, int x){
e[idx] = x;
ne[idx] = ne[k];
ne[k] = idx;
idx++;
}
//删除第k+1个节点
void remove(int k){
ne[k] = ne[ne[k]];
}
int main(){
init();
int m;
cin>>m;
char c;
while(m--){
cin>>c;
int k, x;
if(c == 'H'){
cin>>x;
add_head(x);
}else if(c == 'D'){
cin>>k;
//删除头节点
if(k == 0){
head = ne[head];
}else{
remove(k);
}
}else{
cin>>k>>x;
add(k, x);
}
}
for(int i = head; i != -1; i = ne[i]){
cout<<e[i]<<" ";
}
return 0;
}
双链表
https://www.acwing.com/problem/content/829/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int e[N], l[N], r[N], idx;
//初始化,0表示左端点,1表示右端点,下标从2开始
void init(){
r[0] = 1;
l[1] = 0;
idx = 2;
}
//在第k个点的右边插入x
void add(int k, int x){
e[idx] = x;
r[idx] = r[k];
l[idx] = k;
l[r[k]] = idx;
r[k] = idx;
idx++;
}
//删除第k个点
void remove(int k){
r[l[k]] = r[k];
l[r[k]] = l[k];
}
int main(){
init();
int m;
cin>>m;
string s;
while(m--){
cin>>s;
int k, x;
if(s == "L"){
cin>>x;
add(0, x);
}else if(s == "R"){
cin>>x;
add(l[1], x);
}else if(s == "D"){
cin>>k;
//下标从2开始
remove(k + 1);
}else if(s == "IL"){
cin>>k>>x;
add(l[k + 1], x);
}else{
cin>>k>>x;
add(k + 1, x);
}
}
for(int i = r[0]; i != 1; i = r[i]){
cout<<e[i]<<" ";
}
return 0;
}
栈
先进后出
模拟栈
https://www.acwing.com/problem/content/830/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int st[N], tt;
//栈顶插入一个元素
void push(int x){
st[++tt] = x;
}
//删除栈顶
void pop(){
tt--;
}
//判断栈是否为空
void empty(){
cout<<(tt > 0 ? "NO" : "YES")<<endl;
}
//查询栈顶元素
void query(){
cout<<st[tt]<<endl;
}
int main(){
int m;
cin>>m;
string s;
while(m--){
int x;
cin>>s;
if(s == "push"){
cin>>x;
push(x);
}else if(s == "pop"){
pop();
}else if(s == "empty"){
empty();
}else{
query();
}
}
return 0;
}
表达式求值
https://www.acwing.com/problem/content/3305/
中缀表达式,需要括号;后缀表达式,不需要括号
#include<iostream>
#include<unordered_map>
using namespace std;
const int N = 1e5 + 10;
int st1[N], tt1, tt2; //存储数字
char st2[N]; //存储运算符
unordered_map<char, int> mp{{'+', 1}, {'-', 1}, {'*', 2}, {'/', 2}}; //运算符优先级
string s;
void f(){
int x1 = st1[tt1];
tt1--;
int x2 = st1[tt1];
tt1--;
char c = st2[tt2];
tt2--;
int res = 0;
if(c == '-'){
res = x2 - x1;
}else if(c == '+'){
res = x2 + x1;
}else if(c == '*'){
res = x2 * x1;
}else{
res = x2 / x1;
}
st1[++tt1] = res;
}
int main(){
cin>>s;
int n = s.size();
for(int i = 0; i < n; i++){
//如果是数字
if(isdigit(s[i])){
int x = 0, j = i;
//计算数字
while(j < n && isdigit(s[j])){
x = x * 10 + (s[j] - '0');
j++;
}
//存储到数字栈中
st1[++tt1] = x;
//因为for循环i还要++
i = j - 1;
}else if(s[i] == '('){
//存储到运算符栈中
st2[++tt2] = s[i];
}else if(s[i] == ')'){
//当栈顶不等于左括号时,就计算括号里的
while(st2[tt2] != '('){
f();
}
//弹出左括号
tt2--;
}else{
//当运算符栈中有元素,当前元素小于等于栈顶优先级,则先计算之前的
while(tt2 > 0 && mp[st2[tt2]] >= mp[s[i]]){
f();
}
//存储到运算符栈中
st2[++tt2] = s[i];
}
}
//防止算了括号里的就结束了
while(tt2 > 0){
f();
}
cout<<st1[tt1];
return 0;
}
队列
先进先出
模拟队列
https://www.acwing.com/problem/content/831/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int q[N], hh, tt = -1;
//插入元素x
void push(int x){
q[++tt] = x;
}
//弹出队头
void pop(){
hh++;
}
//判断队列是否为空
void empty(){
cout<<(hh <= tt ? "NO" : "YES")<<endl;
}
//查询队头
void query(){
cout<<q[hh]<<endl;
}
int main(){
int m;
cin>>m;
string s;
while(m--){
cin>>s;
int x;
if(s == "push"){
cin>>x;
push(x);
}else if(s == "pop"){
pop();
}else if(s == "empty"){
empty();
}else{
query();
}
}
return 0;
}
单调栈
https://www.acwing.com/problem/content/832/
存储一个单调递增的序列
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int st[N], tt;
int n, x;
int main(){
cin>>n;
for(int i = 0; i < n; i++){
cin>>x;
//如果栈顶大于等于当前元素,那就弹出栈顶
while(tt && st[tt] >= x){
tt--;
}
//如果栈顶小于当前元素,那就输出
if(tt){
cout<<st[tt]<<" ";
}else{
cout<<-1<<" ";
}
//入栈
st[++tt] = x;
}
return 0;
}
单调队列
https://www.acwing.com/problem/content/156/
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int a[N], q[N]; //q数组存储下标
int n, k, x;
int main(){
cin>>n>>k;
for(int i = 0; i < n; i++){
cin>>a[i];
}
int hh = 0, tt = -1;
for(int i = 0; i < n; i++){
cin>>x;
//如果队头已经出窗口了,那就出队
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++){
cin>>x;
//如果队头已经出窗口了,那就出队
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;
}
KMP
https://www.acwing.com/problem/content/833/
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int ne[N];
int n, m;
char s[N], p[N];
int main(){
//下标从1开始
cin >> m >> p + 1 >> n >> 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;
}
//s串和p串匹配
for(int i = 1, j = 0; i <= n; i++){
//如果j没有回到起点,且不相等,那就到标记的位置
while(j && s[i] != p[j + 1]){
j = ne[j];
}
//如果匹配,j向后移
if(s[i] == p[j + 1]){
j++;
}
//如果完全匹配p串,那就到下一个位置再尝试
if(j == m){
cout << i - m << " ";
j = ne[j];
}
}
return 0;
}
//下标从0开始
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1000010;
int n, m;
char s[N], p[N];
int ne[N];
int main()
{
cin >> m >> p >> n >> s;
ne[0] = -1;
for (int i = 1, j = -1; i < m; i ++ )
{
while (j >= 0 && p[j + 1] != p[i]) j = ne[j];
if (p[j + 1] == p[i]) j ++ ;
ne[i] = j;
}
for (int i = 0, j = -1; i < n; i ++ )
{
while (j != -1 && s[i] != p[j + 1]) j = ne[j];
if (s[i] == p[j + 1]) j ++ ;
if (j == m - 1)
{
cout << i - j << ' ';
j = ne[j];
}
}
return 0;
}
Trie树
高效的存储和查找字符串集合
Trie字符串统计
https://www.acwing.com/problem/content/837/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int son[N][26], cnt[N], idx; //0是根节点也是空节点
int n;
//插入字符串
void insert(char str[N]){
int p = 0;
//字符串结尾是'\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(){
cin>>n;
char c, s[N];
while(n--){
cin>>c;
if(c == 'I'){
cin>>s;
insert(s);
}else{
cin>>s;
cout<<query(s)<<endl;
}
}
return 0;
}
最大异或对
https://www.acwing.com/problem/content/145/
字典树不单单可以高效存储和查找字符串集合, 还可以存储二进制数字
将每个数以二进制方式存入字典树, 找的时候从最高位去找有无该位的异
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int son[31 * N][2], idx;
int n, x;
int ans, res;
void insert(int x){
int p = 0;
for(int i = 30; i >= 0; i--){
int u = x >> i & 1;
if(!son[p][u]){
son[p][u] = ++idx;
}
p = son[p][u];
}
}
int query(int x){
int p = 0;
int sum = 0;
for(int i = 30; i >= 0; i--){
int u = x >> i & 1;
if(son[p][!u]){
p = son[p][!u];
sum = 2 * sum + !u;
}else{
p = son[p][u];
sum = 2 * sum + u;
}
}
return sum;
}
int main(){
cin>>n;
for(int i = 0; i < n; i++){
cin>>x;
insert(x);
res = query(x);
ans = max(ans, res ^ x);
}
cout<<ans;
return 0;
}
并查集
合并集合
https://www.acwing.com/problem/content/838/
开始时每个集合都是一个独立的集合,并且都是等于自己本身下标的数
例如:p[5]=5,p[3]=3;
如果是M操作的话那么就将集合进行合并,合并的操作是:
p[3]=p[5]=5;
所以3的祖宗节点便成为了5
此时以5为祖宗节点的集合为{5,3}
如果要将p[9]=9插入到p[3]当中,应该找到3的祖宗节点,
然后再把p[9]=9插入其中,所以p[9]=find(3);(find()函数用于查找祖宗节点)
也可以是p[find(9)]=find(3),因为9的节点本身就是9
此时以5为祖宗节点的集合为{5,3,9};
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int p[N];
int n, m;
//找到根节点,路径压缩
int find(int x){
if(x != p[x]){
p[x] = find(p[x]);
}
return p[x];
}
int main(){
cin>>n>>m;
//初始化每个节点的根节点
for(int i = 1; i <= n; i++){
p[i] = i;
}
char c;
int a, b;
while(m--){
cin>>c;
if(c == 'M'){
cin>>a>>b;
//将a合并到b的集合中
p[find(a)] = find(b);
}else{
cin>>a>>b;
cout<<(find(a) == find(b) ? "Yes" : "No")<<endl;
}
}
return 0;
}
连通块中点的数量
https://www.acwing.com/problem/content/839/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int p[N], cnt[N];
int n, m;
int find(int x){
if(x != p[x]){
p[x] = find(p[x]);
}
return p[x];
}
int main(){
cin>>n>>m;
for(int i = 0; i < n; i++){
p[i] = i;
cnt[i] = 1;
}
string s;
while(m--){
cin>>s;
int a, b;
if(s == "C"){
cin>>a>>b;
//如果是一个集合,就不用再加了
if(find(a) == find(b)){
continue;
}
cnt[find(b)] += cnt[find(a)];
p[find(a)] = find(b);
}else if(s == "Q1"){
cin>>a>>b;
cout<<(find(a) == find(b) ? "Yes" : "No")<<endl;
}else{
cin>>a;
cout<<cnt[find(a)]<<endl;
}
}
return 0;
}
食物链
https://www.acwing.com/problem/content/description/242/
#include<iostream>
using namespace std;
const int N = 5e4 + 10;
int p[N], d[N]; //p存储根节点,d存储到根节点的距离
int n, k;
int res;
int find(int x){
if(x != p[x]){
int t = find(p[x]);
d[x] += d[p[x]];
p[x] = t;
}
return p[x];
}
int main(){
cin>>n>>k;
for(int i = 1; i <= n; i++){
p[i] = i;
}
int c, x, y;
while(k--){
cin>>c>>x>>y;
if(x > n || y > n){
res++;
}else{
//找到x和y的根节点
int px = find(x), py = find(y);
if(c == 1){
//如果父节点相同,但是距离不同,就是假话
if(px == py && (d[x] - d[y]) % 3){
res++;
}else if(px != py){
//不在一个集合,放在一个集合中
p[px] = py;
d[px] = d[y] - d[x]; //d[px] + d[x]和d[y]同余
}
}else{
if(px == py && (d[x] - d[y] - 1) % 3){
res++;
}else if(px != py){
p[px] = py;
d[px] = d[y] - d[x] + 1;
}
}
}
}
cout<<res;
return 0;
}
堆
用一个一维数组存储一颗二叉树
1.插入一个元素
2.求集合当中最小值(最大值)
3.删除最小值(最大值)
4.删除任意一个元素
5.修改任意一个元素
小根堆:根节点小于等于左右子节点
堆排序
https://www.acwing.com/problem/content/840/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int h[N], cnt;
int n, m;
void down(int u){
int t = u;
//如果根节点大于左右子节点,那就交换下标
if(2 * u <= cnt && h[2 * u] < h[t]){
t = 2 * u;
}
if(2 * u + 1 <= cnt && h[2 * u + 1] < h[t]){
t = 2 * u + 1;
}
//如果已经交换了下标,那就递归子节点,u为根节点,t为子节点
if(h[t] != h[u]){
swap(h[t], h[u]);
down(t);
}
}
int main(){
cin>>n>>m;
cnt = n;
for(int i = 1; i <= n; i++){
cin>>h[i];
}
for(int i = n / 2; i > 0; i--){
down(i);
}
while(m--){
cout<<h[1]<<" ";
h[1] = h[cnt];
cnt--;
down(1);
}
return 0;
}
模拟堆
https://www.acwing.com/problem/content/841/
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int h[N], cnt, m; //m:第几个插入的数
int hp[N], ph[N]; //hp[k]:第k个插入的点是几号,ph[j]:j号是第几个插入的点
int n;
void head_swap(int a, int b){
swap(hp[ph[a]], hp[ph[b]]);
swap(ph[a], ph[b]);
swap(h[a], h[b]);
}
void down(int u){
int t = u;
if(2 * u <= cnt && h[2 * u] < h[t]){
t = 2 * u;
}
if(2 * u + 1 <= cnt && h[2 * u + 1] < h[t]){
t = 2 * u + 1;
}
if(h[t] != h[u]){
head_swap(t, u);
down(t);
}
}
void up(int u){
while(u / 2 > 0 && h[u / 2] > h[u]){
head_swap(u / 2, u);
u /= 2;
}
}
int main(){
cin>>n;
string s;
while(n--){
cin>>s;
int k, x;
if(s == "I"){
cin>>x;
m++;
cnt++;
hp[m] = cnt;
ph[cnt] = m;
h[cnt] = x;
up(cnt);
}else if(s == "PM"){
cout<<h[1]<<endl;
}else if(s == "DM"){
head_swap(1, cnt);
cnt--;
down(1);
}else if(s == "D"){
cin>>k;
k = hp[k];
head_swap(k, cnt);
cnt--;
down(k);
up(k);
}else{
cin>>k>>x;
k = hp[k];
h[k] = x;
down(k);
up(k);
}
}
return 0;
}
哈希表
模拟散列表
https://www.acwing.com/problem/content/842/
//拉链法
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 3; //大于1e5的第一个质数
int h[N], e[N], ne[N], idx;
int n;
void insert(int x){
//加N防止模出来的数是负数
int k = (x % N + N) % N;
e[idx] = x;
ne[idx] = h[k];
h[k] = idx;
idx++;
}
int find(int x){
int k = (x % N + N) % N;
for(int i = h[k]; i != -1; i = ne[i]){
if(e[i] == x){
return 1;
}
}
return 0;
}
int main(){
cin>>n;
memset(h, -1, sizeof(h));
char c;
while(n--){
cin>>c;
int x;
if(c == 'I'){
cin>>x;
insert(x);
}else{
cin>>x;
if(find(x)){
cout<<"Yes"<<endl;
}else{
cout<<"No"<<endl;
}
}
}
return 0;
}
//开放寻地址法
#include<iostream>
#include<cstring>
using namespace std;
const int N = 2e5 + 3, null = 0x3f3f3f3f; //第一个大于2e5的质数
int h[N];
int n;
int find(int x){
int k = (x % N + N) % N;
//如果k位置不等于x,那就找下一个数
while(h[k] != x && h[k] != null){
k++;
//到结尾也没找到,就从头开始找
if(k == N){
k = 0;
}
}
return k;
}
int main(){
memset(h, 0x3f, sizeof(h));
cin>>n;
char c;
while(n--){
cin>>c;
int x;
if(c == 'I'){
cin>>x;
//找到第一个空位置
int k = find(x);
h[k] = x;
}else{
cin>>x;
//查询是否有x存在
int k = find(x);
cout<<(h[k] == x ? "Yes" : "No")<<endl;
}
}
return 0;
}
字符串哈希
https://www.acwing.com/problem/content/843/
#include<iostream>
using namespace std;
const int N = 1e5 + 10, p = 131; //p为131或13331,2的64次方,溢出int就等效于取模了
unsigned long long a[N], b[N]; //a存储前缀和,b存储要乘的进位数
char s[N];
int n, m;
//初始化
void init(){
b[0] = 1;
for(int i = 1; i <= n; i++){
a[i] = a[i - 1] * p + s[i];
b[i] = b[i - 1] * p;
}
}
unsigned long long get(int l, int r){
return a[r] - a[l - 1] * b[r - l + 1];
}
int main(){
cin>>n>>m>>s + 1;
init();
int l1, r1, l2, r2;
while(m--){
cin>>l1>>r1>>l2>>r2;
cout<<(get(l1, r1) == get(l2, r2) ? "Yes" : "No")<<endl;
}
return 0;
}