数据结构(一)

单链表

头插法

  • 将x插到头结点
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
// head 表示头结点的下标
// e[i] 表示节点i的值
// ne[i] 表示节点i的next指针是多少
// idx 存储当前已经用到了哪个点

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

void add_to_head(int x)
{
e[idx] = x;
ne[idx] = head;// 注意这里的head代表的是第一个节点
head = idx;
idx ++ ;
}

一般插入法

1
2
3
4
5
6
7
8
// 将x插到下标是k的点后面
void add(int k, int x)
{
e[idx] = x;
ne[idx] = ne[k];
ne[k] = idx;
idx ++ ;
}

将下标是k的点后面的点删掉

1
2
3
4
5
6
// 将下标是k的点后面的点删掉
void remove(int k)
{
ne[k] = ne[ne[k]];
}

把以上的东西结合起来

实现一个单链表,链表初始为空,支持三种操作:

  1. 向链表头插入一个数;
  2. 删除第 $k$ 个插入的数后面的数;
  3. 在第$k$ 个插入的数后插入一个数。

现在要对该链表进行 $M$ 次操作,进行完所有操作后,从头到尾输出整个链表。

注意:题目中第 $k$ 个插入的数并不是指当前链表的第 $k$ 个数。例如操作过程中一共插入了 $n$ 个数,则按照插入的时间顺序,这 $n$ 个数依次为:第 1 个插入的数,第 2 个插入的数,…第 $n$ 个插入的数。

输入格式

第一行包含整数 $M$,表示操作次数。

接下来 $M$ 行,每行包含一个操作命令,操作命令可能为以下几种:

  1. H x,表示向链表头插入一个数 $x$。
  2. D k,表示删除第 k 个插入的数后面的数(当 $k$ 为 0 时,表示删除头结点)。
  3. I k x,表示在第 k 个插入的数后面插入一个数 $x$(此操作中 $k$ 均大于 0)。

输出格式

共一行,将整个链表从头到尾输出。

数据范围

$1 \leq M \leq 100000$
所有操作保证合法。

输入样例:

1
2
3
4
5
6
7
8
9
10
11
10
H 9
I 1 1
D 1
D 0
H 6
I 3 6
I 4 5
I 4 5
I 3 4
D 6

输出样例:

1
6 4 6 5

AC代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include <iostream>

using namespace std;

const int N = 100010;


// head 表示头结点的下标
// e[i] 表示节点i的值
// ne[i] 表示节点i的next指针是多少
// idx 存储当前已经用到了哪个点
int head, e[N], ne[N], idx;

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

// 将x插到头结点
void add_to_head(int x)
{
e[idx] = x, ne[idx] = head, head = idx ++ ;
}

// 将x插到下标是k的点后面
void add(int k, int x)
{
e[idx] = x, ne[idx] = ne[k], ne[k] = idx ++ ;
}

// 将下标是k的点后面的点删掉
void remove(int k)
{
ne[k] = ne[ne[k]];
}

int main()
{
int m;
cin >> m;

init();

while (m -- )
{
int k, x;
char op;

cin >> op;
if (op == 'H')
{
cin >> x;
add_to_head(x);
}
else if (op == 'D')
{
cin >> k;
if (!k) head = ne[head];// 如果k=0,那么!k=1,执行删除头结点的操作
else remove(k - 1);// 因为0号点是第一个插入的点1号点事第2个插入的点
}
else
{
cin >> k >> x;
add(k - 1, x);
}
}

for (int i = head; i != -1; i = ne[i]) cout << e[i] << ' ';
cout << endl;

return 0;
}

双链表

在节点a的右边插入一个数x

1
2
3
4
5
6
7
8
// 在节点a的右边插入一个数x
void insert(int a, int x)
{
e[idx] = x;
l[idx] = a, r[idx] = r[a];
l[r[a]] = idx, r[a] = idx;
idx ++ ;
}

删除节点a

1
2
3
4
5
6
// 删除节点a
void remove(int a)
{
l[r[a]] = l[a];
r[l[a]] = r[a];
}

实现一个双链表,双链表初始为空,支持 $5$ 种操作:

  1. 在最左侧插入一个数;
  2. 在最右侧插入一个数;
  3. 将第 $k$ 个插入的数删除;
  4. 在第 $k$ 个插入的数左侧插入一个数;
  5. 在第 $k$ 个插入的数右侧插入一个数

现在要对该链表进行 $M$ 次操作,进行完所有操作后,从左到右输出整个链表。

注意:题目中第 $k$ 个插入的数并不是指当前链表的第 $k$ 个数。例如操作过程中一共插入了 $n$ 个数,则按照插入的时间顺序,这 $n$ 个数依次为:第 $1$ 个插入的数,第 $2$ 个插入的数,…第 $n$ 个插入的数。

输入格式

第一行包含整数 $M$,表示操作次数。

接下来 $M$ 行,每行包含一个操作命令,操作命令可能为以下几种:

  1. L x,表示在链表的最左端插入数 $x$。
  2. R x,表示在链表的最右端插入数 $x$。
  3. D k,表示将第 $k$ 个插入的数删除。
  4. IL k x,表示在第 $k$ 个插入的数左侧插入一个数。
  5. IR k x,表示在第 $k$ 个插入的数右侧插入一个数。

输出格式

共一行,将整个链表从左到右输出。

数据范围

$1 \leq M \leq 100000$
所有操作保证合法。

输入样例:

1
2
3
4
5
6
7
8
9
10
11
10
R 7
D 1
L 3
IL 2 10
D 3
IL 2 7
L 8
R 9
IL 4 7
IR 2 2

输出样例:

1
8 7 7 3 2 9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <iostream>

using namespace std;

const int N = 100010;

int m;
int e[N], l[N], r[N], idx;

// 在节点a的右边插入一个数x
void insert(int a, int x)
{
e[idx] = x;
l[idx] = a, r[idx] = r[a];
l[r[a]] = idx, r[a] = idx ++ ;
}

// 删除节点a
void remove(int a)
{
l[r[a]] = l[a];
r[l[a]] = r[a];
}

int main()
{
cin >> m;

// 初始化,0是左端点,1是右端点
r[0] = 1, l[1] = 0;// r[0]是0的右边
idx = 2;// 因为0和1已经被占用过了

while (m -- )
{
string op;
cin >> op;
int k, x;
if (op == "L")
{
cin >> x;
insert(0, x);
}
else if (op == "R")
{
cin >> x;
insert(l[1], x);
}
else if (op == "D")
{
cin >> k;
remove(k + 1);// 因为0和1已经被占用过了,所以可以使用的从2开始,例如第3个插入的节点在从0开始的数组中实际上是4
}
else if (op == "IL")
{
cin >> k >> x;
insert(l[k + 1], x);// 因为我只写了从右边插入,所以想从左边插入就只能往前再推一位了
}
else
{
cin >> k >> x;
insert(k + 1, x);
}
}

for (int i = r[0]; i != 1; i = r[i]) cout << e[i] << ' ';
cout << endl;

return 0;
}

先进后出

实现一个栈,栈初始为空,支持四种操作:

  1. push x – 向栈顶插入一个数 xx;
  2. pop – 从栈顶弹出一个数;
  3. empty – 判断栈是否为空;
  4. query – 查询栈顶元素。

现在要对栈进行 MM 个操作,其中的每个操作 $3$ 和操作 $4$ 都要输出相应的结果。

输入格式

第一行包含整数 $M$,表示操作次数。

接下来 $M$ 行,每行包含一个操作命令,操作命令为 push xpopemptyquery 中的一种。

输出格式

对于每个 emptyquery 操作都要输出一个查询结果,每个结果占一行。

其中,empty 操作的查询结果为 YESNOquery 操作的查询结果为一个整数,表示栈顶元素的值。

数据范围

$1 \leq M \leq 100000$
$1 \leq x \leq 10^9$
所有操作保证合法。

输入样例:

1
2
3
4
5
6
7
8
9
10
11
10
push 5
query
push 6
pop
query
pop
empty
push 4
query
empty

输出样例:

1
2
3
4
5
5
5
YES
4
NO
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include <iostream>

using namespace std;

const int N = 100010;

int m;
int stk[N], tt;// tt是栈顶

int main()
{
cin >> m;
while (m -- )
{
string op;
int x;

cin >> op;
if (op == "push")
{
cin >> x;
stk[ ++ tt] = x;// 插入
}
else if (op == "pop") tt -- ;// 弹出
else if (op == "empty") cout << (tt ? "NO" : "YES") << endl;// 判断tt是不是大于0,如果大于0就是不空
else cout << stk[tt] << endl;
}

return 0;
}

队列

先进先出

实现一个队列,队列初始为空,支持四种操作:

  1. push x – 向队尾插入一个数 xx;
  2. pop – 从队头弹出一个数;
  3. empty – 判断队列是否为空;
  4. query – 查询队头元素。

现在要对队列进行 MM 个操作,其中的每个操作 33 和操作 44 都要输出相应的结果。

输入格式

第一行包含整数 $M$,表示操作次数。

接下来 $M$ 行,每行包含一个操作命令,操作命令为 push xpopemptyquery 中的一种。

输出格式

对于每个 emptyquery 操作都要输出一个查询结果,每个结果占一行。

其中,empty 操作的查询结果为 YESNOquery 操作的查询结果为一个整数,表示队头元素的值。

数据范围

$1 \leq M \leq 100000$
$1 \leq x \leq 10^9$
所有操作保证合法。

输入样例:

1
2
3
4
5
6
7
8
9
10
11
10
push 6
empty
query
pop
empty
push 3
push 4
pop
query
push 6

输出样例:

1
2
3
4
NO
6
YES
4

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#include <iostream>

using namespace std;

const int N = 100010;

int m;
int q[N], hh=0, tt = -1;

int main()
{
cin >> m;

while (m -- )
{
string op;
int x;

cin >> op;
if (op == "push")
{
cin >> x;
q[ ++ tt] = x;// 从队尾插入
}
else if (op == "pop") hh ++ ;从队头弹出
else if (op == "empty") cout << (hh <= tt ? "NO" : "YES") << endl;// 判断tt是不是大于0,如果大于0就是不空
else cout << q[hh] << endl;// 取出队头元素
}

return 0;
}

单调栈

给定一个长度为 $N$ 的整数数列,输出每个数左边第一个比它小的数,如果不存在则输出 $−1$。

输入格式

第一行包含整数 $N$,表示数列长度。

第二行包含 $N$ 个整数,表示整数数列。

输出格式

共一行,包含 $N$ 个整数,其中第 $i$ 个数表示第 $i$ 个数的左边第一个比它小的数,如果不存在则输出 $−1$。

数据范围

$1 \leq N \leq 10^5$
$1 \leq 数列中元素 \leq 10^9$

输入样例:

1
2
5
3 4 2 7 5

输出样例:

1
-1 3 -1 2 2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#include <iostream>

using namespace std;

const int N = 100010;

int stk[N], tt;// stk[N]是栈的元素,tt是栈顶节点

int main()
{
int n;
cin >> n;// 一共输入几个数据
while (n -- )
{
int x;
scanf("%d", &x);// 读入数据
while (tt && stk[tt] >= x) tt -- ;// 栈要不空,如果栈顶数据的大小比读入的数据大,那么栈顶元素就没必要存在了
// 因为从第i个数表示第i个数的左边第一个比它小的数,就相当于从栈顶开始往下寻找比第i个数小且最近的数
// 如果栈顶都能满足条件,那么比栈顶小的才读入的数据也一定满足,此时栈顶应该是左边第二个小的,我们需要第一个小的
// 也就是刚读入的那个数据,原来的栈顶删除,刚读入的数据是新的栈顶,后面再读入的数据再找从栈顶比它小的最近一个
// 也肯定用不上之前的删除的那个栈顶了
if (!tt) printf("-1 ");// tt=0,!tt=1,也就是说没有栈里面没有一个数据比读进来的数据小,按照题目要求输出−1
else printf("%d ", stk[tt]);// 找到比读入数据小的左边最近的数据了,并且输出符合条件的数据
stk[ ++ tt] = x;// 把数据x插到我们的栈里面去
}

return 0;
}


单调队列

给定一个大小为 $n \leq 10^6$的数组。

有一个大小为 $k$ 的滑动窗口,它从数组的最左边移动到最右边。

你只能在窗口中看到 $k$ 个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子:

该数组为 [1 3 -1 -3 5 3 6 7],$k$ 为 $3$。

窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7

你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。

输入格式

输入包含两行。

第一行包含两个整数 $n$ 和 $k$,分别代表数组长度和滑动窗口的长度。

第二行有 $n$ 个整数,代表数组的具体数值。

同行数据之间用空格隔开。

输出格式

输出包含两个。

第一行输出,从左至右,每个位置滑动窗口中的最小值。

第二行输出,从左至右,每个位置滑动窗口中的最大值。

输入样例:

1
2
8 3
1 3 -1 -3 5 3 6 7

输出样例:

1
2
-1 -3 -3 -3 3 3
3 3 5 5 6 7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <iostream>

using namespace std;

const int N = 1000010;

int a[N], q[N];// q[N]里边存的是下标

int main()
{
int n, k;// 分别代表数组长度和滑动窗口的长度
scanf("%d%d", &n, &k); // 读入数组长度和滑动窗口的长度
for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);// 把所有数据都读进来

int hh = 0, tt = -1;// 队头是0,队尾是-1
for (int i = 0; i < n; i ++ )
{
if (hh <= tt && i - k + 1 > q[hh]) hh ++ ;
// i是我们枚举的右端点,也就是窗口最右边的,k是窗口长度
// 窗口的范围是在i-k+1到i之间,q[N]里边存的是下标,我们的窗口是往右边移动的
// 先判断队列是否为空,i-k+1是起点,如果大于q[hh],那么说明我们的q[hh]已经出了我们的窗口了,然后删除
while (hh <= tt && a[q[tt]] >= a[i]) tt -- ;
// 如果我们新插入的数比队尾的数据要小,那么队尾的数据永远满足不了条件,永远输出不了,就可以把队尾的数据删掉
q[ ++ tt] = i;// 把当前的数据的下标插到q[]里面去

if (i >= k - 1) printf("%d ", a[q[hh]]);// 它是从前k个数据开始输出的,当我们的数不足k和就不用输出答案
}

puts("");// 输出一个回车
// 以上是用来输出滑动窗口的最小值

// 下面是输出滑动窗口的最大值


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) printf("%d ", a[q[hh]]);
}

puts("");

return 0;
}


kmp

暴力做法

1
2
3
4
5
6
7
8
9
10
11
// 暴力做法
s[N], p[M]
for (int i = 1; i <= n; i++) {
bool flag = true;
for (int j = 1; j <= m; j++) {
if (s[i + j - 1] != p[j]) {
flag = false;
break;
}
}
}

试题

给定一个字符串 $S$,以及一个模式串 $P$,所有字符串中只包含大小写英文字母以及阿拉伯数字。

模式串 $P$ 在字符串 $S$ 中多次作为子串出现。

求出模式串 $P$ 在字符串 $S$ 中所有出现的位置的起始下标。

输入格式

第一行输入整数 $N$,表示字符串 $P$ 的长度。

第二行输入字符串 $P$。

第三行输入整数 $M$,表示字符串 $S$ 的长度。

第四行输入字符串 $S$。

输出格式

共一行,输出所有出现位置的起始下标(下标从 $0$ 开始计数),整数之间用空格隔开。

数据范围

$1 \leq N \leq 10^5$
$1 \leq M \leq 10^6$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include <iostream>

using namespace std;

const int N = 100010, M = 1000010;

int n, m;
int ne[N];
char s[M], p[N];// s数组是长的那个目标,p数组则是短的那个去跟s比较是否相等的

int main()
{
cin >> n >> p + 1 >> m >> s + 1;// 我们下标从1开始

// 求ne[]数组
for (int i = 2, j = 0; i <= n; i ++ )
// j表示匹配成功的长度,i表示q数组中的下标
// 因为q数组的下标是从1开始的,只有1个时,没有所谓的前后缀,ne[](或者说是j)一定为0,所以i从2开始

{
while (j && p[i] != p[j + 1]) j = ne[j];
// 不匹配,就让j往回退一步

if (p[i] == p[j + 1]) j ++ ;
// 匹配了,就让j加上1
// 不断刷新数组中最长的公共前后缀的长度
ne[i] = j;
}

// kmp的匹配过程
for (int i = 1, j = 0; i <= m; i ++ )
// 试图和Si匹配的是Pj+1,因此我们j总是往前错一位
// 注意哈,i是从1开始跑的,j是从0开始的
// 两个数组想要在同列相比就必须让j加上1
{
while (j && s[i] != p[j + 1]) j = ne[j];
// j没有退回起点(如果退回,意味着我们需要重新匹配)
// 试图和Si匹配的是Pj+1,因此我们j总是往前错一位
// j没有退回起点和匹配失败(不相等),j就要更新为j长的数组中最长的公共前后缀
// 假如数组前3位比较下来都一样(第4位不一样),那么数组中最长的公共前后缀长度就是3
// 而第4位不一样了,正常我们要把j退回起点从头开始比较
// 但是j = ne[j]帮助我们把j跑到了p数组的第3位
// 接下来我们只需要从第四位开始比较
// 所以j = ne[j]这步相当于让j往回退一步比如从4退到3

// 这个循环结束条件,要么j被退回起点了,要么就是两个数组匹配成功了

if (s[i] == p[j + 1]) j ++ ;
// 如果真的匹配成功了,我们的j就可以移动到下一个位置了
// 从总体上看也就是匹配成功后i++,s数组向后移动一位,p数组向后移动一位
// 开始进行下一轮比较

if (j == n)
// 匹配成功
// j是j长的数组中最长的公共前后缀

{
printf("%d ", i - n);
j = ne[j];
}
}

return 0;
}