-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.cpp
1896 lines (1831 loc) · 62.5 KB
/
main.cpp
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#include <iostream>
#include <vector>
#include <stack>
#include <queue>
//#include <bits/unordered_map.h>
#include <unordered_map>
#include <algorithm>
#include <set>
#include <map>
using namespace std;
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x):val(x), left(NULL), right(NULL){
}
};
struct RandomListNode{
int label;
RandomListNode* next;
RandomListNode* random;
RandomListNode(int x):label(x), next(NULL), random(NULL){}
};
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
class Solution{
public:
/*
* 重建二叉树
*/
/*
* 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
* 假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
* 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
*/
TreeNode* reConstructBinaryTree(vector<int> preArr, vector<int> inArr){
if(preArr.size() ==0 || inArr.size() ==0 || preArr.size() != inArr.size())
return NULL;
return reConstructBinaryTree(preArr, inArr, 0, preArr.size() - 1, 0, inArr.size() - 1);
}
TreeNode* reConstructBinaryTree(vector<int> preArr, vector<int> inArr, int preStart, int preEnd, int inStart, int inEnd){
if(preEnd - preStart < 0 || inEnd - inStart < 0)
return NULL;
TreeNode* pRoot = new TreeNode(preArr[preStart]);
int curIndexInInArr = findIndexOfVal(inArr, inStart, inEnd, preArr[preStart]);
int leftNums = curIndexInInArr - inStart; // 以cur为根的左子树结点个数
int rightNums = inEnd - curIndexInInArr;
if(leftNums > 0){
pRoot->left = reConstructBinaryTree(preArr, inArr, preStart + 1, preStart + leftNums, inStart, inStart + leftNums - 1);
}
if(rightNums > 0){
pRoot->right = reConstructBinaryTree(preArr, inArr, preStart + leftNums + 1, preEnd, inStart + leftNums + 1, inEnd);
}
return pRoot;
}
int findIndexOfVal(vector<int> arr, int start, int end, int val){ // 返回某值在数组中的下标, 没找到返回 -1
if(end - start >= 0){
for(int i=start; i<=end; i++){
if(arr[i] == val){
return i;
}
}
}
return -1;
}
/*
* 用两个栈实现队列
*/
/*
* 用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
* 思路:pop: 只从 stack2 中弹出元素,若 stack2 为空,将 stack1 中的元素全部倒入 stack2
* push: 直接将元素压入 stack1
*/
int pop(){
int res;
if(stack2.empty()){
while (!stack1.empty()){
stack2.push(stack1.top());
stack1.pop();
}
}
res = stack2.top();
stack2.pop();
return res;
}
void push(int x){
stack1.push(x);
}
/*
* 旋转数组的最小数字
*/
/*
* 把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。
* 输入一个非减排序的数组的一个旋转,输出旋转数组的最小元素。
* 例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。
* NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
* 思路:设置两个指针,一个指针指向前一个数组中最后一个元素,一个指针指向后一个数组中第一个元素,当两个指针相遇或相邻时即能找到最小值
* 当 arr[mid] > arr[right] , mid 指向前一半数组中的元素
* 当 arr[mid] <= arr[right], mid 指向后一半数组中的元素
* 特殊情况: arr[mid] == arr[left] == arr[right] 只能顺序查找数组最小值
*/
/*
* 迭代方法
*/
int minNumberInRotateArray(vector<int> rotateArr){
if(rotateArr.size() == 0)
return 0;
int left = 0, right = rotateArr.size() - 1;
while (right - left > 1){
int mid = (left + right) / 2;
if(rotateArr[mid] == rotateArr[left] && rotateArr[left] == rotateArr[right])
return findMinNumberBySeq(rotateArr, left, right);
if(rotateArr[mid] > rotateArr[right]){
left = mid + 1;
} else{ // if(rotateArr[mid] <= rotateArr[right])
right = mid;
}
}
if(right - left == 1 || right == left){
return rotateArr[left] < rotateArr[right] ? rotateArr[left] : rotateArr[right];
}
}
/*
* 递归方法
*/
int minNumberInRotateArray2(vector<int> rotateArr){
if(rotateArr.size() == 0 )
return 0;
return minNumberInRotateArray2(rotateArr, 0, rotateArr.size() - 1);
}
int minNumberInRotateArray2(vector<int> rotateArr, int left, int right){
if(right - left > 1){
int mid = (right + left) / 2;
if(rotateArr[mid] == rotateArr[left] && rotateArr[left] == rotateArr[right]){
return findMinNumberBySeq(rotateArr, left, right);
}
if(rotateArr[mid] > rotateArr[right]){
return minNumberInRotateArray2(rotateArr, mid + 1, right);
} else if(rotateArr[mid] <= rotateArr[right]){
return minNumberInRotateArray2(rotateArr, left, mid);
}
}
if(right == left)
return rotateArr[left];
if(right - left == 1)
return rotateArr[right] < rotateArr[left] ? rotateArr[right] : rotateArr[left];
}
int findMinNumberBySeq(vector<int> arr, int left, int right){
int res = 0;
for(int i=left; i<=right; i++){
if(arr[i] < res){
res = arr[i];
}
}
return res;
}
/*
* 数值的整数次方
*/
double Power(double base, int exponent) {
if(exponent == 0)
return 1;
if(base == 0)
return 0;
bool isExponentNeg = false;
if(exponent < 0){
isExponentNeg = true;
exponent = -exponent;
}
double res = base;
bool isOdd = false;
if(exponent % 2 == 1)
isOdd = true;
while(exponent > 1 && exponent / 2){
res *= res;
exponent = exponent / 2;
}
if(isOdd)
res = res * base;
if(isExponentNeg)
res = 1.0 / res;
return res;
}
/*
* 树的子结构
* 输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
* 思路:两个函数, HasSubtree 判断B是不是A的子结构
* isSubtree 判断B是不是A的同根子结构
*/
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) // 树 pRoot1 中是否有子树 pRoot2
{
if(pRoot1 && pRoot2){
if(isSubtree(pRoot1, pRoot2) == false){
return HasSubtree(pRoot1->left, pRoot2) ||
HasSubtree(pRoot1->right, pRoot2);
}else{
return true;;
}
}
return false;
}
bool isSubtree(TreeNode* pRoot1, TreeNode* pRoot2){ // 树 pRoot2 是否是 pRoot1 以 pRoot1 为根的子树
if(pRoot2 == NULL)
return true;
if(pRoot1 == NULL)
return false;
if(pRoot1->val == pRoot2->val)
return isSubtree(pRoot1->left, pRoot2->left) &&
isSubtree(pRoot1->right, pRoot2->right);
return false;
}
/*
* 二叉树的镜像
* 操作给定的二叉树,将其变换为源二叉树的镜像。
*/
void Mirror(TreeNode* pRoot){
if(pRoot != NULL){
SwapNode(pRoot->left, pRoot->right);
if(pRoot->left)
Mirror(pRoot->left);
if(pRoot->right)
Mirror(pRoot->right);
}
}
void SwapNode(TreeNode* &p1, TreeNode* &p2){
TreeNode* temp = p1;
p1 = p2;
p2 = temp;
}
/*
* 顺时针打印矩阵
*/
vector<int> printMatrix(vector<vector<int >> matrix){
vector<int> res;
if(matrix.size() == 0)
return res;
int row1 = 0, row2 = matrix.size() - 1;
int col1 = 0, col2 = matrix[0].size() - 1;
while (row2 - row1 >= 0 && col2 - col1 >= 0){
printMatrix(matrix, res, row1++, row2--, col1++, col2--);
}
return res;
}
void printMatrix(vector<vector<int >> matrix, vector<int> &res, int row1, int row2, int col1, int col2){
if(row2 - row1 >= 0 && col2 - col1 >= 0){
// 只有一行
if(row1 == row2){
for(int i=col1; i<=col2; i++){
res.push_back(matrix[row1][i]);;
}
return;
}
// 只有一列
if(col1 == col2){
for(int i=row1; i<=row2; i++){
res.push_back(matrix[i][col1]);
}
return;
}
// 从左到右
for(int i=col1; i<col2; i++){
res.push_back(matrix[row1][i]);
}
// 从上到下
for(int i=row1; i<row2; i++){
res.push_back(matrix[i][col2]);
}
// 从右到左
for(int i=col2; i>col1; i--){
res.push_back(matrix[row2][i]);
}
// 从下到上
for(int i=row2; i>row1; i--){
res.push_back(matrix[i][col1]);
}
}
}
/*
* 包含min函数的栈
* 定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
* 思路:
*/
void push1(int x){
if(minStack.empty() || x <= minStack.top()){
minStack.push(x);
}
dataStack.push(x);
}
void pop1(){
if(dataStack.top() == minStack.top()){
minStack.pop();
}
dataStack.pop();
}
int top(){
return dataStack.top();
}
int min(){
return minStack.top();
}
/*
* 栈的压入、弹出序列
*
* 输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。
* 假设压入栈的所有数字均不相等。
* 例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,
* 但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
*
* 思路:使用辅助栈,将pushV的元素一个个压栈,
* 当遇到栈顶与popV中元素相等时,弹栈,指向popV的下标后移
*/
bool isPopOrder(vector<int> pushV, vector<int> popV){
if(pushV.size() != popV.size())
return false;
stack<int> s;
for(int i=0, j=0; i<pushV.size() && j<popV.size(); i++){
s.push(pushV[i]);
while (j<popV.size() && !s.empty() && s.top() == popV[j]){ // 保证j不越界,栈不空,栈顶和popV元素相等
s.pop();
j++;
}
}
if(s.empty())
return true;
return false;
}
/*
* 从上往下打印二叉树(层序打印二叉树,借助队列)
*
* 从上往下打印出二叉树的每个节点,同层节点从左至右打印。
*/
vector<int> PrintFromTopToBottom(TreeNode* pRoot){
vector<int> res;
if(pRoot == NULL)
return res;
queue<TreeNode*> q;
TreeNode* p;
q.push(pRoot);
while (!q.empty()){
p = q.front();
q.pop();
res.push_back(p->val);
if(p->left)
q.push(p->left);
if(p->right)
q.push(p->right);
}
return res;
}
/*
* 二叉搜索树的后序遍历序列 (BST, postOrder)
*
* 输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。
* 如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
*
* 二叉排序树的性质:左子树上所有节点的值均小于它的根节点;右子树上所有节点的值均大于它的根节点。
* 二叉排序树后序遍历的性质:序列最后一个数字是根节点,序列剩余部分分成两部分,前一部分是左子树,后一部分是右子树。
*/
bool VerifySquenceOfBST(vector<int> sequence){
if(sequence.size() == 0)
return false;
int rootIndex = sequence.size() - 1;
int start = 0;
int end = sequence.size() - 2;
return IsPostOrderOfBST(sequence, rootIndex, start, end);
}
bool IsPostOrderOfBST(vector<int> sequence, int rootIndex, int start, int end){
if(end >= start){
int smallSet = start - 1; // smallset 指向 start 前一个
int bigSet = end + 1; // bigSet 指向 end 后一个
int i = start, j = end;
while (i <= end && sequence[i] < sequence[rootIndex]){
smallSet++;
i++;
}
while (j>= start && sequence[j] > sequence[rootIndex]){
bigSet--;
j--;
}
if(smallSet == bigSet - 1) // 如果 smallset 和 bigset 是相邻的,通过当前检查,继续检查左右子区间
return IsPostOrderOfBST(sequence, smallSet, start, smallSet - 1) && IsPostOrderOfBST(sequence, end, bigSet, end-1);
return false;
}
return true;
}
/*
* 二叉树中和为某一值的路径
*
* 输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。
* 路径定义:从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
* 每条满足条件的路径都是以根节点开始,叶子结点结束,
* 如果想得到所有根节点到叶子结点的路径(不一一定满足和为某整数的条件),需要遍历整棵树,还要先遍历根节点,所以采用先序遍历
* (注意: 在返回值的list中,数组长度大的数组靠前)
*/
vector<vector<int >> FindPath(TreeNode* pRoot, int expectNumber){
vector<vector<int >> res;
if(pRoot == NULL)
return res;
vector<int> path;
int curNumber = 0;
FindPath(pRoot, res, path, curNumber, expectNumber);
return res;
}
// 深度优先遍历(DFS) 树的深度优先遍历即先根遍历
void FindPath(TreeNode* pRoot, vector<vector<int >> &res, vector<int> &path, int curNumber, int expectNumber){
if(pRoot == NULL)
return;
path.push_back(pRoot->val);
curNumber += pRoot->val;
bool isLeaf = pRoot->left == NULL && pRoot->right == NULL;
if(isLeaf && curNumber == expectNumber){ // 如果当前节点是叶节点,且满足条件,将该路径加入res(先处理根)
res.push_back(path);
}
if(curNumber < expectNumber ){ // 当满足条件,继续遍历下去(先左后右遍历左右子树)
if(pRoot->left){
FindPath(pRoot->left, res, path, curNumber, expectNumber);
}
if(pRoot->right){
FindPath(pRoot->right, res, path, curNumber, expectNumber);
}
}
path.pop_back(); // *移除当前节点*
}
/*
* 复杂链表的复制
*
* 输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),
* 返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
*
* 思路1:使用 hashSet, 时复 O(n), 空复 O(n)
* (key, value) = (oldNode, newNode)
* newNode->val = oldNode->val;
* newNode->next = oldNode->next;
* pNew->random = mymap[pOld->random];
*/
// RandomListNode* Clone(RandomListNode* pHead){
// if(pHead == NULL)
// return NULL;
// unordered_map<RandomListNode*, RandomListNode*> mymap;
// RandomListNode* resHeadPre = new RandomListNode(0); // 头结点的前一个结点(为方便编写,浪费一个结点)
// RandomListNode* pOld = pHead;
// RandomListNode* pNew = resHeadPre;
// while (pOld){
// RandomListNode* node = new RandomListNode(pOld->label);
// pNew->next = node;
// pNew = node;
// mymap.insert(make_pair(pOld, pNew));
// pOld = pOld->next;
// }
// // 复制 random 指针
// pNew = resHeadPre->next;
// pOld = pHead;
// while (pOld){
// pNew->random = mymap[pOld->random];
// pNew = pNew->next;
// pOld = pOld->next;
// }
// return resHeadPre->next;
// }
/*
* 复杂链表的复制
*
* 输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针指向任意一个节点),
* 返回结果为复制后复杂链表的head。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
*
* 思路2:不使用 hashSet, 时复 O(n), 空复 O(1)
* 例:1->2->3->null
* ① 1->1'->2->2'->3->3'->null
* ② 连接random指针:1'->random = 1->random->next
* ③ 拆分链表
*/
// RandomListNode* Clone2(RandomListNode* pHead){
// if(pHead == NULL)
// return NULL;
// // 复制结点并设置 next 和 label
// RandomListNode* p = pHead;
// RandomListNode* pNext = p->next;
// while (p){
// RandomListNode* node = new RandomListNode(p->label);
// node->next = pNext;
// p->next = node;
// p = pNext;
// if(pNext)
// pNext = pNext->next;
// }
// // 设置 random 11'22'33'null
// p = pHead;
// pNext = p->next;
// while (p){
// if(p->random){
// pNext->random = p->random->next;
// }
// if(pNext){
// p = pNext->next;
// }
// if(p){
// pNext = p->next;
// }
// }
// // 拆分链表 11'22'33'null
// p = pHead;
// pNext = p->next;
// RandomListNode* resHead = pNext;
// while (p){
// if(pNext){
// p->next = pNext->next;
// p = p->next;
// }
// if(p){
// pNext->next = p->next;
// pNext = pNext->next;
// }
// }
// return resHead;
// }
/*
* 二叉搜索树与双向链表
*
* 输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。
* 要求不能创建任何新的结点,只能调整树中结点指针的指向。
*
* 思路1(递归中序遍历):二叉搜索树中序遍历结果是升序的,在中序遍历二叉搜索树的过程中,记录当前结点和下一个结点,修改两个结点的指针
* 1 2 3
* 1->right = 2;
* 2->left = 1;
* 2->right = 3;
* 3->left = 2;
*/
TreeNode* BinaryTreeConvertList(TreeNode* pRoot){
if(pRoot ==NULL)
return NULL;
TreeNode* pCur = pRoot;
TreeNode* preCur = NULL;
BinaryTreeConvertList(pCur, preCur);
while (pCur->left){
pCur = pCur->left;
}
return pCur;
}
void BinaryTreeConvertList(TreeNode* pCur, TreeNode* &preCur){
// 第一个用指针的原因:pCur的值不需要改变,只需改变pCur指向的值
// 第二个用指针引用原因:preCur的值需要改变
if(pCur == NULL)
return;
if(pCur->left){
BinaryTreeConvertList(pCur->left, preCur);
}
pCur->left = preCur;
if(preCur)
preCur->right = pCur;
preCur = pCur;
if(pCur->right){
BinaryTreeConvertList(pCur->right, preCur);
}
}
/*
* 二叉搜索树与双向链表
*
* 输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。
* 要求不能创建任何新的结点,只能调整树中结点指针的指向。
*
* 思路1(非递归中序遍历):二叉搜索树中序遍历结果是升序的,在中序遍历二叉搜索树的过程中,记录当前结点和下一个结点,修改两个结点的指针
* 1 2 3
* 1->right = 2;
* 2->left = 1;
* 2->right = 3;
* 3->left = 2;
*/
TreeNode* Convert(TreeNode* pRoot){
if(pRoot == NULL)
return NULL;
TreeNode* pre = NULL;
TreeNode* cur = pRoot;
stack<TreeNode*> s;
while (cur || !s.empty()){
while (cur){
s.push(cur);
cur = cur->left;
}
if(!s.empty()){
cur = s.top();
s.pop();
// 修改指针
cur->left = pre;
if(pre){
pre->right = cur;
}
pre = cur;
// 修改指针
cur = cur->right;
}
}
// 找到链表头, 没必要从尾节点开始找, 可以直接从根节点开始找
cur = pRoot;
while (cur->left){
cur = cur->left;
}
return cur;
}
/*
* 字符串的排列
*
* 输入一个字符串,按字典序打印出该字符串中字符的所有排列。
* 例如输入字符串abc,则打印出由字符a,b,c所能排列出来的所有字符串abc,acb,bac,bca,cab和cba。
* 输入一个字符串,长度不超过9(可能有字符重复),字符只包括大小写字母。
*
* 思路:把一个字符串看成两部分组成:第一部分为第一个字符,第二部分为后面的所有字符。
* 求整个字符串的排列,可以看出两步:首先求所有可能出现在第一个位置的字符,
* 即把第一个字符和后面的所有字符交换;然后固定第一个字符,求后面所有字符的排序。
* 此时仍把后面的字符看成两部分,第一个字符和后面的字符,然后重复上述步骤。(递归)
*/
vector<string> Permutation(string str){
vector<string> res;
int curIndex = 0;
// char* cur = &str[0];
// Permutation(str, cur, res);
Permutation(str, curIndex, res);
sort(res.begin(), res.end());
return res;
}
/*
* 方法1:使用指针
*/
void Permutation(string &str, char* cur, vector<string> &res){
if(*cur == '\0'){
res.push_back(str);
return;
}
for(char* i = cur; *i != '\0'; i++){
if(i != cur && *i == *cur)
continue;
char temp = *i;
*i = *cur;
*cur = temp;
Permutation(str, cur + 1, res);
// 换回来
temp = *i;
*i = *cur;
*cur = temp;
}
}
/*
* 方法2:使用下标
*/
void Permutation(string &str, int curIndex, vector<string> &res){
if(curIndex == str.length()){
res.push_back(str);
return;
}
for(int i = curIndex; i<str.length() && curIndex < str.length(); i++){
if(i != curIndex && str[i] == str[curIndex])
continue;
char temp = str[i];
str[i] = str[curIndex];
str[curIndex] = temp;
Permutation(str, curIndex + 1, res);
// 换回来
temp = str[i];
str[i] = str[curIndex];
str[curIndex] = temp;
}
}
/*
* 数组中出现次数超过一半的数字
*
* 数组中有一个数字出现的次数超过数组长度的一半,请找出这个数字。
* 例如输入一个长度为9的数组{1,2,3,2,2,2,5,4,2}。
* 由于数字2在数组中出现了5次,超过数组长度的一半,因此输出2。如果不存在则输出0。
*/
int MoreThanHalfNum(vector<int> arr){
if(arr.size() == 0)
return 0;
int res = arr[0];
int count = 1;
for(int i=1; i<arr.size(); i++){
if(arr[i] == res){
count++;
}else{
if(count > 1){
count--;
} else{
res = arr[i];
count = 1;
}
}
}
if( count > 1 ){
return res;
}
count = 0;
for(int i=0; i<arr.size(); i++){ // verify
if(arr[i] == res)
count++;
}
if(count * 2 > arr.size())
return res;
return 0;
}
/*
* 最小的K个数
*
* 输入n个整数,找出其中最小的K个数。
* 例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
*
* 思路1:partition, 需要改变数组,且找到的最小 k 个数是乱序的
*/
vector<int> minKNumbers(vector<int> arr, int k){
vector<int> res;
if(k > arr.size())
return res;
int begin = 0;
int end = arr.size() - 1;
int index = Partition(arr, begin, end);
if(index != k - 1){
if(index < k - 1){
index = Partition(arr, index + 1, end);
} else{
index = Partition(arr, begin, index - 1);
}
}
for(int i=0; i<k; i++){
res.push_back(arr[i]);
}
return res;
}
int Partition(vector<int> &arr, int begin, int end){
int pivot = arr[begin];
int i = begin, j = end;
while (i < j){
while (i < j && arr[j] >= pivot ){
j--;
}
while (i < j && arr[i] <= pivot){
i++;
}
if(i < j)
swap(arr[i], arr[j]);
}
swap(arr[begin], arr[i]); // 将枢值放在应该放的位置上
return i;
}
/*
* 最小的K个数
*
* 输入n个整数,找出其中最小的K个数。
* 例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
*
* 思路2:不改变数组,排序
* 思路3: 不改变数组,堆
*/
vector<int> GetLeastNumbers(vector<int> arr, int k){
vector<int> res;
if(k > arr.size())
return res;
priority_queue<int> minK; // 建立最小k个数的大根堆
for(int i=0; i<k; i++){
minK.push(arr[i]);
}
for(int i=k; i<arr.size(); i++){
if(!minK.empty() && arr[i] < minK.top()){
minK.pop();
minK.push(arr[i]);
}
}
while (!minK.empty()){
res.push_back(minK.top());
minK.pop();
}
return res;
}
vector<int> GetLeastNumbers_Solution(vector<int> arr, int k){
vector<int> res;
if(k > arr.size())
return res;
multiset<int, greater<int>> minK;
for(int i = 0; i < k; i++){
minK.insert(arr[i]);
}
for (int i = k; i < arr.size(); ++i) {
if(minK.size() > 0 && arr[i] < *minK.begin()){
minK.erase(*minK.begin());
minK.insert(arr[i]);
}
}
multiset<int>::iterator it = minK.begin();
for (; it != minK.end(); ++it) {
res.push_back(*it);
}
return res;
}
/*
* 使用大根堆对数组进行排序
* priority_queue<Type, Container, Functional>
* 如果我们把后面俩个参数缺省的话,优先队列就是大顶堆,队头元素最大。
*/
vector<int> maxHeapSort(vector<int> arr){
vector<int> res;
priority_queue<int> maxheap;
for(int i=0; i<arr.size(); i++){
maxheap.push(arr[i]);
}
while (!maxheap.empty()){
res.push_back(maxheap.top());
maxheap.pop();
}
return res;
}
/*
* 使用小根堆对数组进行排序
*
* priority_queue<Type, Container, Functional>
*/
vector<int> minHeapSort(vector<int> arr){
vector<int> res;
priority_queue<int, vector<int>, greater<int> > minheap;
for(int i=0; i<arr.size(); i++){
minheap.push(arr[i]);
}
while (!minheap.empty()){
res.push_back(minheap.top());
minheap.pop();
}
return res;
}
/*
* STL使用堆方法:
* 1. 优先队列,默认大顶堆 priority_queue<int>
* 小顶堆 priority_queue<int, vector<int>, greater<int> >
* 2. 利用头文件 <algothrim> 中的函数进行堆操作 (1)make_heap(_First, _Last, _Comp)构造堆:默认是建立最大堆的。对int类型,可以在第三个参数传入greater<int>()得到最小堆。
* (2)push_heap (_First, _Last)添加元素到堆:先在容器中加入,再调用push_heap()
* (3)pop_heap(_First, _Last)从堆中移出元素:先调用pop_heap(),再在容器中删除
* (4)sort_heap(_First, _Last)对整个堆排序:排序之后就不再是一个合法的heap结构了, 容器变成有序数组
*/
/*
* 对称的二叉树
*
* 请实现一个函数,用来判断一颗二叉树是不是对称的。
* 注意,如果一个二叉树同此二叉树的镜像是同样的,定义其为对称的。
*
* 思路:先序遍历左右子树
*/
bool isSymmetrical2(TreeNode* pRoot) {
if(pRoot == NULL)
return true;
return isSymmetrical(pRoot->left, pRoot->right);
}
bool isSymmetrical(TreeNode* pRoot1, TreeNode* pRoot2) {
if(pRoot1 == NULL && pRoot2 == NULL)
return true;
if(pRoot1 && pRoot2){
if(pRoot1->val == pRoot2->val){
return isSymmetrical(pRoot1->left, pRoot2->right) && isSymmetrical(pRoot1->right, pRoot2->left);
}else{
return false;
}
}
return false;
}
/*
* 连续子数组的最大和
*
* 例如:{6,-3,-2,7,-15,1,2,2},连续子向量的最大和为8(从第0个开始,到第3个为止)。给一个数组,返回它的最大连续子序列的和(子向量的长度至少是1)
*
* 思路:res 记录当前连续子向量最大和; curSum 记录当前连续子向量的和
* 遍历数组, curSum + arr[i] >= 0 时, 继续使用当前的 curSum, 否则重置 curSum 为 0
* 过程中一旦发现 curSum > res , 更新 res 的值
*/
int FindGreatestSumOfSubArray(vector<int> arr){
if(arr.size() == 0)
return 0;
int res = 0;
int curSum = 0;
bool hasPos = false;
for(int i=0; i<arr.size(); i++){
if(curSum + arr[i] >= 0){
curSum += arr[i];
}else{
curSum = 0;
}
if(curSum > res){
res = curSum;
hasPos = true;
}
}
if(hasPos == false){ // 如果数组中全是负数,返回最大值
res = *max_element(arr.begin(), arr.end());
}
return res;
}
/*
* 二叉搜索树的第k个结点
*
* 给定一棵二叉搜索树,请找出其中的第k小的结点。
* 例如,(5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
*
* 思路:中序遍历
*/
TreeNode* KthNode(TreeNode* pRoot, int k){
if(pRoot == NULL || k <= 0)
return NULL;
TreeNode* p = pRoot;
stack<TreeNode*> s;
int cur = 0;
while (p || !s.empty()){
while (p){
s.push(p);
p = p->left;
}
if(!s.empty()){
p = s.top();
s.pop();
cur++;
if(cur == k){
return p;
}
p = p->right;
}
}
return NULL;
}
/*
* 把二叉树打印成多行
*
* 从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
*
* 思路:先求出二叉树高度,初始化数组,再层序遍历一行一行放
*/
vector<vector<int> > PrintBTByLevel(TreeNode* pRoot){
int height = BTHeight(pRoot);
vector<vector<int> > res(height);
if(pRoot == NULL)
return res;
TreeNode* p;
queue<TreeNode*> q;
q.push(pRoot);
int curLine = 0;
while (!q.empty()){
int curCount = 0, eleSum = q.size(); // curCount: 当前行加入元素计数 eleSum:当前行应该加入元素总数
while(curCount < eleSum) {
p = q.front();
q.pop();
res[curLine].push_back(p->val);
curCount++;
if(p->left){
q.push(p->left);
}
if(p->right){
q.push(p->right);
}
}
curLine++;
}
return res;
}
int BTHeight(TreeNode* pRoot){
if(pRoot == NULL)
return 0;
return max(BTHeight(pRoot->left), BTHeight(pRoot->right)) + 1;
}
/*
* 把二叉树打印成多行
*
* 从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
*
* 思路:不求二叉树高度,每层先存在一个vector<int>中, 再一行一行放进 vector<vector<int> > 中
*/
vector<vector<int> > PrintBTByLevel2(TreeNode* pRoot){
vector<vector<int> > res;
if(pRoot == NULL)
return res;
queue<TreeNode*> q;
q.push(pRoot);
while (!q.empty()){
vector<int> curLine;
int start = 0, end = q.size();
while (start < end){
TreeNode* p = q.front();
q.pop();
curLine.push_back(p->val);
if(p->left){
q.push(p->left);
}
if(p->right){
q.push(p->right);
}