-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMiddleproblem-set.py
1312 lines (1302 loc) · 62.5 KB
/
Middleproblem-set.py
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
#kmp算法应用:给你一个文本串 T ,一个非空模板串 S ,问 S 在 T 中出现了多少次
#空间复杂度 O(len(S)),时间复杂度 O(len(S)+len(T))
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 计算模板串S在文本串T中出现了多少次
# @param S string字符串 模板串
# @param T string字符串 文本串
# @return int整型
#
class Solution:
def kmp(self , S , T ):
i, j = 0, 0
lenthS, lenthT = len(S), len(T)
if lenthS > lenthT: return 0
cnt = 0
nextArr = getNext(S)
while lenthT - i >= lenthS - j:
if j != -1 and T[i] == S[j]:
i += 1
j += 1
elif j != -1 and T[i] != S[j]: j = nextArr[j]
else:
i += 1
j += 1
if j == lenthS: #匹配到一个模板串后,i,j更新规则如下:
i -= 1 #i应为当前匹配子串的最后一个字符
j = nextArr[-1] #j应为nextArr对应最后一个字符的值
cnt += 1
return cnt
def getNext(S):
nextList = [-1, 0]
for i in range(2, len(S)):
tmp = nextList[i-1]
while tmp != -1:
if S[i-1] == S[tmp]: break
tmp = nextList[tmp]
nextList.append(tmp + 1)
return nextList
# write code here
###################################################
#尺取法应用:给定一个长度为 n 的数组 num 和滑动窗口的大小 size ,找出所有滑动窗口里数值的最大值
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param num int整型一维数组
# @param size int整型
# @return int整型一维数组
#
class Solution:
def maxInWindows(self , num: List[int], size: int) -> List[int]:
lenth = len(num)
res = []
if not size or size > lenth: return res
deque = [] #双端队列,记录下标
left, right = -1, 0
while right < len(num):
while deque and num[right] > num[deque[-1]]:
deque.pop() #右边界当前数大于队尾,出队直到队尾大于等于当前数
deque.append(right)
right += 1 #deque[0]始终为(left,right)的最大值下标
if deque[0] == left: #判断当前窗口最大值是否为左边界数
deque.pop(0)
if right - left > size: #当达到指定窗口大小时,记录结果并更新left
res.append(num[deque[0]])
left += 1
return res
# write code here
###################################################
#对于一个给定的链表,返回环的入口节点,如果没有环,返回None
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
#
# @param head ListNode类
# @return ListNode类
###################################################
class Solution:
def detectCycle(self , head ):
slow = head
fast = head
while fast and fast.next:
slow = slow.next
fast = fast.next.next
if fast == slow: #利用快慢指针判断是否有环并找相遇点
break
if fast and fast.next: #找到相遇点后,将slow指向头节点
slow = head
while fast != slow: #slow与fast以相同速度移动,直到相遇,相遇点即为入口节点
slow = slow.next
fast = fast.next
return slow
return None
# write code here
###################################################
#给定两个字符串str1和str2,输出两个字符串的最长公共子串
# longest common substring
# @param str1 string字符串 the string
# @param str2 string字符串 the string
# @return string字符串
#遍历其中一个字符串,maxlen记录当前已找到公共子串的最大长度,找长度为maxlen+1的第一个公共子串,找到后,更新maxlen,ans
#奥义:每次更新maxlen和ans后,长度更长(即maxlen+1)的第一个公共子串只会以当前字符之后的字符为结尾,即按照我们的更新规则,当前字符前不会存在长度更长的公共子串
#因此复杂度为O(len(str1))
class Solution:
def LCS(self , str1 , str2 ):
if len(str1) < len(str2):
str1,str2 = str2,str1
maxlen = 0
ans = ''
for i in range(len(str1)):
if str1[i - maxlen:i + 1] in str2: #for遍历比in判断的复杂度可能低一些,因此在开始先判断字符串长度大小关系进行优化
ans = str1[i-maxlen:i+1] #注意ans与maxlen的更新顺序不能颠倒
maxlen += 1
return ans
# write code here
################################################### |
#给定字符串A以及它的长度n,请返回最长回文子串的长度 |
#sol1:可类比上题(最长公共子串)idea,关键是需要注意更新maxlen的条件 |sol2:中心扩散
#从头到尾扫描字符串,每次更新maxlen和ans后,第一个长度更长(maxlen+1 or maxlen+2)的回文子串只会以当前字符之后的字符为结尾 |
class Solution: | class Solution:
def getLongestPalindrome(self, A, n): | def getLongestPalindrome(self, A, n):
# write code here | maxlen = 0
max_len = 0 | for i in range(n):
for i in range(n): | maxlen = max(temp,spread(A,i, i, n),spread(A,i, i+1, n)) #注意奇长度回文串与偶长度回文串的参数略有不同
oddNum = A[i-max_len-1:i+1] | return
evenNum = A[i-max_len:i+1] | def spread(A,left,right,n): #以A[left]和A[right]为中心向左右两边扩散,返回扩散的最大长度
if i-max_len-1>=0 and oddNum == oddNum[::-1]: | while left >= 0 and right <= n-1 and A[left] == A[right]: #利用回文子串正反一样的特点进行扩散,注意left和right的边界条件
max_len+=2 | left -= 1
elif i-max_len>=0 and evenNum == evenNum[::-1]: | right += 1
max_len+=1 | return right - left - 1
return max_len
###################################################
#根据快速排序的思路,找出数组中第K大的数
class Solution:
def findKth(self, a, n, K):
l = 0
r = n - 1
while l <= r:
p = partion(a,l,r)
if p+1 == K:
return a[p]
elif p+1 > K:
r = p-1
else:
l = p+1
def partion(alist,l,r):
num = alist[l]
left = l + 1
right = r
while left <= right:
while left <= right and alist[left] >= num:
left += 1
while right >= left and alist[right] <= num:
right -= 1
if left <= right:
alist[left], alist[right] = alist[right], alist[left]
alist[right],alist[l] = num,alist[right]
return right
# write code here
###################################################
#请写一个整数计算器,支持加减乘除运算和括号
# @param s string字符串 待计算的表达式
# @return int整型
class Solution:
def solve(self , s ): #此题亦可采用先转为后缀表达式再求值
partStack = [] #keyidea:用列表保存各部分的值,整个表达式的值即为各部分的和 (ex:原表达式形如(A*(B-C))*D,则列表保存(A*(B-C))的值和D的值)
sign = '+' #使用sign记录运算符,初始化为 '+'
number = 0
i = 0
while i < len(s):
if s[i] in '0123456789':
number = number * 10 + ord(s[i]) - ord('0') #number记录字符串中的数字部分
if s[i] == '(': #遇到左括号时递归求这个括号里面的表达式的值
counterPartition = 1
j = i
while counterPartition > 0:
j += 1
if s[j] == '(':
counterPartition += 1
if s[j] == ')':
counterPartition -= 1 #先遍历找到对应的右括号,counterPartition统计括号对数直到变量为0
S = Solution()
number = S.solve(s[i+1:j]) #利用递归来达到去括号的目的,去括号后的表达式具有A+B*C-D的形式
i = j #更新当前扫描到字符的索引
if (s[i] in '+-*') or (i == len(s) - 1): #遇到运算符时或者到表达式末尾时,说明该操作符之前的子表达式已计算完成,可根据sign的值作相应的运算
if sign == '+': #如果是当前sign为+,直接将当前number push 进去
partStack.append(number)
if sign == '-': #如果是-,push 进去-number
partStack.append(-number)
if sign == '*': #如果是 ×、÷ ,pop 出一个运算数和当前数作计算
partStack.append(partStack.pop() * number)
if sign == '/':
partStack.append(partStack.pop() / number)
sign = s[i] #然后保存新的运算符到sign
number = 0 #并将number置零,即完成一个部分的计算
i += 1
return sum(partStack)
# write code here
###################################################
#给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
#从根节点往下递归:
#1. 若该节点是第一个值为o1或o2的节点,则该节点是最近公共祖先;
#2. 否则,看左子树是否包含o1或o2:
#2.1 若左子树包含o1或o2,则看右子树有没有:
#a. 若右子树没有,则公共祖先在左子树
#b. 若右子树有,则o1和o2肯定是左右子树一边一个,则公共祖先是根节点;
#2.2 若左子树不包含o1和o2,则公共祖先在右子树或者该根子树不包含o1和o2。(两种情况都取决于右子树)
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# @param root TreeNode类
# @param o1 int整型
# @param o2 int整型
# @return int整型
class Solution:
def lowestCommonAncestor(self , root , o1 , o2 ):
return self.dfs(root,o1,o2).val
# 该子树是否包含o1或o2
def dfs(self,root,o1,o2):
if root is None: return None
if root.val == o1 or root.val == o2: return root
left = self.dfs(root.left,o1,o2)
# 左子树没有,则在右子树
# 若右子树没有,则右子树返回 None
if left == None: return self.dfs(root.right,o1,o2)
# 左子树有,则看右子树有没有
right = self.dfs(root.right,o1,o2)
if right == None: return left
# 左子树右子树都有,则最近的祖先节点是root
return root
# write code here
###################################################
#给定数组arr,输出arr的最长递增子序列。(如果有多个答案,请输出其中字典序最小的)
# retrun the longest increasing subsequence
# @param arr int整型一维数组 the array
# @return int整型一维数组
#
class Solution:
def LIS(self , arr ):
monoList = [arr[0]] #长度为 i+1 的子序列的最后一位的最小值(不是解,只是长度关联),单调递增
maxlen = [1] #maxlen[i]记录以arr[i]结尾的最长递增子列长度
for i in arr[1:]:
if i >= monoList[-1]:
monoList.append(i)
maxlen.append(len(monoList))
else:
pos = findPosition(monoList, i) #二分查找monoList中第一个 >= arr[i]的元素位置(必然存在)
monoList[pos] = i #找到位置后替换掉,而非插入
maxlen.append(pos+1) #monoList序列的总长不变,但是为了复用原序列一些 < arr[i]的结果,把arr[i]替换到合适的位置
length = len(monoList)
ans = [0]*length
for j in range(-1,-len(arr)-1,-1): #倒着遍历arr,找到满足长度的maxlen就记录,然后更新。(即同样值的maxlen,选尽量靠右边的)
if length > 0:
if maxlen[j] == length:
ans[length-1] = arr[j]
length -= 1
else:break
return ans
def findPosition(li, t):
if li[-1] < t: return None
l = 0
r = len(li) - 1
while l < r:
mid = (l + r) >> 1
if li[mid] >= t:
r = mid
else:
l = mid + 1
return r
# write code here
###################################################
#给定一个二叉树和一个值sum,请找出所有的根节点到叶子节点的节点值之和等于sum的路径
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# @param root TreeNode类
# @param sum int整型
# @return int整型二维数组
#
class Solution:
def pathSum(self , root , target ):
if not root: return []
paths = preOrder(root, [], [])
return [p for p in paths if sum(p) == target]
def preOrder(root, path, paths):
path.append(root.val) #将当前节点值加入path
if root.left:
preOrder(root.left, path, paths) #递归左子树
if root.right:
preOrder(root.right, path, paths) #递归右子树
if not root.left and not root.right: #当递归到没有子树的时候就说明遍历完一条路径path
paths.append(path.copy())
path.pop() #回溯很重要,从path1转到path2时须先移除path1最后一个元素,才能继续添加path2的节点
return paths
# write code here
###################################################
#给出n对括号,请编写一个函数来生成所有的由n对括号组成的合法组合
#关键:当前位置左括号不少于右括号
#构造图,其中节点:目前位置左括号和右括号数(x,y)(x>=y) 边:从(x,y)到(x+1,y)和(x,y+1),x==y时,没有(x,y+1)这条边
#解是从(0,0)出发到(n,n)的全部路径(类比上题)
# @param n int整型
# @return string字符串一维数组
#
class Solution:
def generateParenthesis(self , n ):
return dfs(n, 0, 0, '', [])
def dfs(n, x, y, s, ans):
if x == n and y == n: ans.append(s)
if x < n: dfs(n, x+1, y, s+'(', ans)
if x > y: dfs(n, x, y+1, s+')', ans)
return ans
# write code here
###################################################
#现在有一个只包含数字的字符串,将该字符串转化成IP地址的形式,返回所有可能的情况
# @param s string字符串
# @return string字符串一维数组
#关于ip: 1.包含四段字符串,每一段≤”255”,≥0且至多包含3个字符;2.如果一个字符包括2个或3个字符,它的第一个字符不能为0,不能为010或者03等
class Solution:
def restoreIpAddresses(self , s ):
return dfs(s, [], [])
def dfs(s, tmp, ans):
if not s and len(tmp) == 4: ans.append('.'.join(tmp))
for i in range(1, len(s)+1): #i为每一段的长度,可以为1,2,3
if i > 1 and s[0] == '0': break #判断是否合法
if int(s[0:i]) > 255: break
tmp.append(s[0:i]) #tmp列表保存每一步分割后的字符串
dfs(s[i:], tmp, ans) #递归
tmp.pop() #回溯
return ans
# write code here
###################################################
#在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。
#输入一个数组,求出这个数组中的逆序对的总数P
# -*- coding:utf-8 -*-
class Solution:
def InversePairs(self, data):
return inversePairsHelp(data, 0, len(data)-1)
def inversePairsHelp(data, l, r): #借助归并排序
if l == r: return 0
mid = l + ((r - l) >> 1)
left = inversePairsHelp(data, l, mid)
right = inversePairsHelp(data, mid + 1, r)
ans = left + right #整个数组的逆序对 = 左子数组逆序对+右子数组逆序对 + 左右子数组共同组成逆序对 (递归)
tmp = data[l:r+1] #当左右数组有序时可在线性时间复杂度内得到第三项,考虑到归并排序也有递归的思想,因此可在一个递归下同时做这两件事情:排序和计算逆序对(O(nlogn))
i, j, k = mid - l, r - l, r
while i >= 0 and j >= mid+1-l: #每次让data[l:r+1]中的最大数归位
if tmp[i] > tmp[j]:
ans += j - mid + l
data[k] = tmp[i]
i -= 1
else:
data[k] = tmp[j]
j -= 1
k -= 1
if i >= 0: #让剩下的数归位
data[l:k+1] = tmp[0:i+1]
if j >= mid+1-l:
data[l:k+1] = tmp[mid+1-l:j+1]
return ans
# write code here
###################################################
#有一个源源不断的吐出整数的数据流,假设你有足够的空间来保存吐出的数。请设计一个名叫MedianHolder的结构,MedianHolder可以随时取得之前吐出所有数的中位数。
#[要求]
#1. 如果MedianHolder已经保存了吐出的N个数,那么将一个新数加入到MedianHolder的过程,其时间复杂度是O(logN)。
#2. 取得已经吐出的N个数整体的中位数的过程,时间复杂度为O(1)
#keypoint:中位数即某个能将所有数根据大小关系划分为左右两半的数,只要能够不断的维护好这两部分的数,中位数就能在O(1)的时间内求得
# @param operations int整型二维数组 ops
# @return double浮点型一维数组
#
import heapq
class Solution:
def __init__(self):
self.maxHeap, self.minHeap = [], [] #maxHeap存放小于等于中位数, minHeap存放大于中位数
def flowmedian(self , operations ):
ans = []
for op in operations:
if op[0] == 1:
self.addNum(op[1])
if op[0] == 2:
ans.append(self.findMedia())
return ans
def addNum(self, num): #trick:让数据在两个堆中流动一遍
heapq.heappush(self.maxHeap, -num)
heapq.heappush(self.minHeap, -heapq.heappop(self.maxHeap))
if len(self.maxHeap) < len(self.minHeap): # 平衡左右两个堆的大小,总是保证len(maxHeap) >= len(minHeap)且至多多1
heapq.heappush(self.maxHeap, -heapq.heappop(self.minHeap))
def findMedia(self):
if len(self.maxHeap) == 0: return -1
if len(self.maxHeap) == len(self.minHeap): return (self.minHeap[0] - self.maxHeap[0]) / 2 #偶数个数据
else: return -self.maxHeap[0] #奇数个数据
# write code here
###################################################
#一个整型数组里除了两个数字只出现一次,其他的数字都出现了两次。请写程序找出这两个只出现一次的数字
# @param array int整型一维数组
# @return int整型一维数组
#由于存在a != b, tmp == a ^ b且tmp不等于0, 依据tmp二进制表示中右起第一位不为1的位置将数组划分为两部分
#则a, b必分属不同部分,再次对两部分分别异或即得a, b
class Solution:
def FindNumsAppearOnce(self , array ):
ans = [0] * 2
tmp = 0
i = 1
for num in array: tmp = tmp ^ num # ^异或: a^0 = a, a^a = 0
while tmp & i == 0: # &按位与
i <<= 1
for num in array:
if num & i == 0: ans[0] ^= num
else: ans[1] ^= num
if ans[0] > ans[1]: ans.reverse()
return ans
# write code here
###################################################
#把只包含质因子2、3和5的数称作丑数(Ugly Number),习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数
class Solution:
def GetUglyNumber_Solution(self, index):
if not index: return 0
uglys = [1]
p2, p3, p5 = 0, 0, 0
for i in range(1, index): #前面生成的每一个丑数都应该乘以2,3,5以产生新的丑数
uglys.append(min(uglys[p2] * 2, uglys[p3] * 3 ,uglys[p5] * 5))#但是我们不必比较前面的每一个丑数, 例如假设前面的某个丑数a 因为乘以2 得到了当前的丑数b,下一次再生成下一个丑数时,就不用再考察a以及a前面的丑数*2的情况
if uglys[i] % 2 == 0: p2 += 1
if uglys[i] % 3 == 0: p3 += 1
if uglys[i] % 5 == 0: p5 += 1
return uglys[-1]
# write code here
###################################################
#一个重复字符串是由两个相同的字符串首尾拼接而成,例如abcabc便是长度为6的一个重复字符串,而abcba则不存在重复字符串。
#给定一个字符串,请编写一个函数,返回其最长的重复字符子串。若不存在任何重复字符子串,则返回0。
# @param a string字符串 待计算字符串
# @return int整型
#
class Solution:
def solve(self , a ):
n = len(a)
if n < 2:
return 0
count = 0
for window in range(n // 2, -1, -1):
for j in range(n - window):
if a[j] == a[j+window]:
count += 1
else:
count = 0
if count == window:
return 2 * window
return 0
# write code here
###################################################
#在一个二维数组array中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序
#请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param target int整型
# @param array int整型二维数组
# @return bool布尔型
#
class Solution:
def Find(self , target: int, array: List[List[int]]) -> bool:
end_row, start_col = len(array) - 1, 0
return FindHelp(target, array, end_row, start_col)
def FindHelp(target, array, end_row, start_col):
#IMPORTANT: 基于array的特点,从左下角(记为ld)开始查找
#当target=ld,直接返回True; 当target>ld,只须在ld所在列右方查找; 当target<ld只需在ld所在行上方查找,递归
if start_col >= len(array[0]) or end_row < 0: return False #超出范围还未找到,返回False
if target == array[end_row][start_col]: return True
if target > array[end_row][start_col]:
return FindHelp(target, array, end_row, start_col + 1)
if target < array[end_row][start_col]:
return FindHelp(target, array, end_row - 1, start_col)
# write code here
###################################################
#给定两个递增数组arr1和arr2,已知两个数组的长度都为N,求两个数组中所有数的上中位数(time:O(logN),space:O(1))
#note:上中位数:假设递增序列长度为n,为第n/2个数
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# find median in two sorted array
# @param arr1 int整型一维数组 the array1
# @param arr2 int整型一维数组 the array2
# @return int整型
#
class Solution:
def findMedianinTwoSortedAray(self , arr1: List[int], arr2: List[int]) -> int:
return aux(arr1, arr2, 0, len(arr1)-1, 0, len(arr2)-1)
def aux(arr1, arr2, l1, r1, l2, r2):
if l1 == r1: return min(arr1[l1], arr2[l2]) #终止条件1
mid1 = (l1 + r1) // 2
mid2 = (l2 + r2) // 2
lenth = r1 - l1 + 1
if arr1[mid1] == arr2[mid2]: #终止条件2
return arr1[mid1]
if arr1[mid1] > arr2[mid2]:
r1 = mid1
l2 = mid2 if lenth % 2 else mid2+1 #IMPORTANT:数组长度分奇偶性讨论, 确保两个数组删除的元素个数相同
if arr1[mid1] < arr2[mid2]:
r2 = mid2
l1 = mid1 if lenth % 2 else mid1+1
return aux(arr1, arr2, l1, r1, l2, r2) #递归求解
# write code here
###################################################
#现在有一个没有重复元素的整数集合S,求S的所有子集
#注意:给出的子集中的元素必须按升序排列; 给出的解集中不能出现重复的元素
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param S int整型一维数组
# @return int整型二维数组
#
class Solution:
def subsets(self , S: List[int]) -> List[List[int]]:
if not S: return []
S.sort() #排序,满足子集内升序
allsubs = []
for lenth in range(len(S)+1): #以子集长度分类添加
dfs(S, [], allsubs, 0, lenth)
return allsubs
def dfs(S, tmpSet, res, start, Len): #更新start参数确保不会有重复集合
if len(tmpSet) == Len: res.append(tmpSet.copy()) #满足长度条件,加入res
else:
for i in range(start, len(S)):
tmpSet.append(S[i])
dfs(S, tmpSet, res, i+1, Len)
tmpSet.pop()
return res
# write code here
###################################################
#给定一个非负整数 n ,返回 n! 结果的末尾为 0 的数量
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# the number of 0
# @param n long长整型 the number
# @return long长整型
#
#由因数分解知识可将问题转为: n!能拆出多少对(2,5)
#IMPORTANT: 可以发现,有5因子的数比有2因子的数要少,所以只需求能拆出来多少个因子5即可(因为一定能有足够数量的因子2来匹配)
#又发现,5的倍数可以至少产生1个5,25的倍数可以产生至少2个5,125的倍数可以产生至少3个5...,因此答案即为 n/5+n/25+n/25...
class Solution:
def thenumberof0(self , n: int) -> int:
res = 0
div = 5
while n // div:
res += n // div
div *= 5
return res
# write code here
###################################################
#给定一个长度为n的数组nums,请你找到峰值并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个所在位置即可。
#1.峰值元素是指其值严格大于左右相邻值的元素。严格大于即不能有等于
#2.假设 nums[-1] = nums[n] = −∞
#3.对于所有有效的 i 都有 nums[i] != nums[i + 1]
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param nums int整型一维数组
# @return int整型
#最优解法:二分
class Solution:
def findPeakElement(self , nums: List[int]) -> int:
l = 0
r = len(nums) - 1 #[l,r]为一定存在峰值的区间
while l < r: #不断二分压缩区间,最后的单点集即为峰值索引
mid = (l + r) >> 1
if nums[mid] < nums[mid+1]:
l = mid + 1 #IMPORTANT: l, r的更新必须保证[l,r]一定存在峰值
else:
r = mid
return l
# write code here
###################################################
#给定无序数组arr,返回其中最长的连续序列的长度(要求值连续,位置可以不连续,例如 3,4,5,6为连续的自然数)
#
# max increasing subsequence
# @param arr int整型一维数组 the array
# @return int整型
#
class Solution:
def MLS(self , arr ):
long=0
arr=set(arr)
for num in arr:
if num-1 not in arr:
cur=num #找到每一段连续子列的最小值
cur_long=1
while cur+1 in arr: #计算该段连续子列的长度
cur+=1
cur_long+=1
long=max(long, cur_long)
return long
# write code here
###################################################
#一座大楼有 n+1 层,地面算作第0层,最高的一层为第 n 层。已知棋子从第0层掉落肯定不会摔碎,从第 i 层掉落可能会摔碎,也可能不会摔碎
#给定整数 n 作为楼层数,再给定整数 k 作为棋子数,返回如果想找到棋子不会摔碎的最高层数,即使在最差的情况下扔的最小次数。一次只能扔一个棋子
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 返回最差情况下扔棋子的最小次数
# @param n int整型 楼层数
# @param k int整型 棋子数
# @return int整型
#反过来考虑: dp[i][j]为i个棋子扔j次能够辨别的最大n值
#key1:动态方程dp[i][j] = dp[i-1][j-1]+1 + dp[i][j-1],可使用一维数组和一个记录变量节省空间复杂度
#记dp[i-1][j-1]为k,第1颗棋子在k+1层扔下,若碎,剩下i-1颗棋子扔j-1次可得结果;否则剩下i颗棋子扔j-1次
import math
class Solution:
def solve(self , n: int, k: int) -> int:
b = math.log2(n) + 1 #key2:采取二分策略时最差情况所需次数,优化时间复杂度
if k >= b:
return int(b)
dp = [0] * k
res = 0
while True:
tmp = 0
res += 1
for i in range(k):
dp[i], tmp = tmp + dp[i] + 1, dp[i]
if dp[i] >= n: return res #依次增加扔的次数并判断是否达到要求
# write code here
###################################################
#给定一个由'0'和'1'组成的2维矩阵,返回该矩阵中最大的由'1'组成的正方形的面积,输入的矩阵是字符形式而非数字形式
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 最大正方形
# @param matrix char字符型二维数组
# @return int整型
class Solution:
def solve(self , matrix: List[List[str]]) -> int:
if not matrix or not matrix[0]: return 0 #判断矩阵是否为空
rows, cols = len(matrix), len(matrix[0])
dp = [[0] * cols for i in range(rows)] #IMPORTANT: dp[i][j]表示的是在矩阵中以坐标(i,j)为右下角的最大正方形边长
res = 0
for r in range(rows):
if matrix[r][0] == '1': #base case直接求
res = 1
dp[r][0] = 1
for c in range(cols):
if matrix[0][c] == '1':
res = 1
dp[0][c] = 1
for row in range(1, rows):
for col in range(1, cols):
if matrix[row][col] == '0':
dp[row][col] = 0
else:
dp[row][col] = min(dp[row-1][col], dp[row][col-1], dp[row-1][col-1]) + 1 #IMPORTANT:动态方程,dp[row][col-1]为水平方向可延伸的最大长度,dp[row-1][col]为垂直方向可延伸的最大长度
res = max(pow(dp[row][col], 2), res)
return res
# write code here
###################################################
#有 n 个活动即将举办,每个活动都有开始时间与活动的结束时间,第 i 个活动的开始时间是 starti ,第 i 个活动的结束时间是 endi ,举办某个活动就需要为该活动准备一个活动主持人
#一位活动主持人在同一时间只能参与一个活动。并且活动主持人需要全程参与活动,换句话说,一个主持人参与了第 i 个活动,那么该主持人在 (starti,endi) 这个时间段不能参与其他任何活动
#求为了成功举办这 n 个活动,最少需要多少名主持人
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 计算成功举办活动需要多少名主持人
# @param n int整型 有n个活动
# @param startEnd int整型二维数组 startEnd[i][0]用于表示第i个活动的开始时间,startEnd[i][1]表示第i个活动的结束时间
# @return int整型
#
import heapq
class Solution:
def minmumNumberOfHost(self , n: int, startEnd: List[List[int]]) -> int:
startEnd.sort() #按开始时间升序,此步不可省略
endTime = [float('inf')] #结束时间优先队列,初始化为无穷大
for party in startEnd:
if party[0] >= endTime[0]: #若开始时间≥最早结束时间
n -= 1 #则有一个主持人可以共用
heapq.heappop(endTime) #弹出最早结束时间
heapq.heappush(endTime, party[1]) #更新优先队列
return n
# write code here
###################################################
#给定一个长度为n的数组nums,数组由一些非负整数组成,现需要将他们进行排列并拼接,每个数不可拆分,使得最后的结果最大,返回值需要是string类型,否则可能会溢出
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 最大数
# @param nums int整型一维数组
# @return string字符串
#定义新的排序规则,使得排序后的字符串数组 依次拼接 所得总串的字典序在所有拼接方案中最大
from functools import cmp_to_key
#cmp_to_key将旧式比较函数转换为关键函数 key function
#具体用法参考https://my.oschina.net/u/4346073/blog/3673092
class Solution:
def solve(self , nums: List[int]) -> str:
nums = [str(num) for num in nums]
nums.sort(key= cmp_to_key(cmp))
return ''.join(nums) if nums[0] != '0' else '0'
def cmp(a, b): #IMPORTANT: 新的排序规则
if a + b == b + a:
return 0 #返回0,表示在新的规则下a==b
elif a + b < b + a:
return 1 #返回1,表示在新的规则下a>b
else:
return -1 #返回-1,表示在新的规则下a<b
# write code here
###################################################
#圆圈中最后剩下的数,即约瑟夫环问题, 空间复杂度O(1)
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param n int整型
# @param m int整型
# @return int整型
#KEY: f(n,m) = (f(n-1,m) + m) % n, f(n, m) 表示最终留下元素的索引
#若x = f(n - 1, m),则长度为 n 的序列最后一个删除的元素,应当是从 m % n 开始数的第 x 个元素,因此有 f(n, m) = (m % n + x) % n = (m + x) % n
class Solution:
def LastRemaining_Solution(self , n: int, m: int) -> int:
i = 1
ans = 0
while i < n:
i += 1
ans = (ans + m) % i
return ans
# write code here
###################################################
#给定一棵完全二叉树的头节点head,返回这棵树的节点个数(完全二叉树指:设二叉树的深度为h(根节点记为第一层),则 [1,h-1] 层的节点数都满足 2^(i-1))
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param head TreeNode类
# @return int整型
#二分:注意到 一棵完全二叉树的左右子树一定有一边是满二叉树,另一边至少也是完全二叉树
#所以当左右子树高度相等时(即满二叉树)可直接计算, 因此在递归的过程中实际上只会沿着其中的一个方向不断递归,时间复杂度logN * logN
class Solution:
def nodeNum(self , head: TreeNode) -> int:
if not head: return 0
lh = rh = 0
curNode = head
while curNode != None:
curNode = curNode.left
lh += 1
curNode = head
while curNode != None:
curNode = curNode.right #由于是完全二叉树,计算高度可不用递归
rh += 1
if lh == rh: return pow(2, lh) - 1
return 1 + self.nodeNum(head.left) + self.nodeNum(head.right)
###################################################
#给你一个 1 到 n 的排列和一个栈,并按照排列顺序入栈
#在不打乱入栈顺序的情况下,输出字典序最大的出栈序列
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 栈排序
# @param a int整型一维数组 描述入栈顺序
# @return int整型一维数组
#KEY: 按照读入顺序入栈,如果当前入栈第i个数字比将要入栈的剩余元素都要大, 则不断弹出栈中元素,直到栈顶元素小于比将要入栈的剩余元素最大值, 遍历完后弹出栈中所有元素
class Solution:
def solve(self , a: List[int]) -> List[int]:
if not a: return []
monoStack = [0] * len(a)
maxNum = -1
for i in range(len(a)):
monoStack[-i-1] = maxNum = max(maxNum, a[-i-1]) #从后往前遍历, monoStack[i]为a[i:n]中最大值
res = []
stack = []
for i in range(len(a)):
stack.append(a[i])
while i < len(a)-1 and stack and stack[-1] > monoStack[i+1]: res.append(stack.pop()) #IMPORTANT:只需对前n-1个元素判断
while stack:
res.append(stack.pop())
return res
# write code here
###################################################
#输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。
#如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同
#法一:分治+递归, O(N^2)
#法二:单调栈, O(N), 算法流程如下:
#初始化: 单调栈 stackstack ,父节点值 root = +∞ (初始值为正无穷大,可把树的根节点看为此无穷大节点的左孩子);
#倒序遍历 postorder :记每个节点为 r_i
#判断: 若 r_i>root,说明此后序遍历序列不满足二叉搜索树定义,直接返回 false
#更新父节点 rootroot : 当栈不为空 且 r_i<stack.peek()时,循环执行出栈,并将出栈节点赋给 root
#入栈: 将当前节点 r_i入栈;
#若遍历完成,则说明后序遍历满足二叉搜索树定义,返回 true
class Solution_recur:
def verifyPostorder(self, postorder: List[int]) -> bool:
return self.partition(postorder, 0, len(postorder)-1)
def partition(self, postorder, l, r):
flag = True
if l >= r: return flag
left, right = l, r-1
while flag and left <= right:
while flag and left <= right and postorder[left] < postorder[r]:
left += 1
while flag and left <= right and postorder[right] > postorder[r]:
right -= 1
if left < right: flag = False
return flag and self.partition(postorder, l, right) and \
self.partition(postorder, right+1, r-1)
class Solution_monoStack:
def verifyPostorder(self, postorder: List[int]) -> bool:
#单调栈O(n)
stack, root = [], float('inf')
postorder.reverse()
for num in postorder:
if num > root: return False
while stack and num < stack[-1]:
root = stack.pop()
stack.append(num)
return True
###################################################
#已知一棵节点个数为 n 的二叉树的中序遍历单调递增, 求该二叉树能能有多少种树形, 输出答案对 109 +7 取模
#法一:动态规划
#法二:卡特兰数
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# 计算二叉树个数
# @param n int整型 二叉树结点个数
# @return int整型
#
class Solution_dp:
def numberOfTree(self , n: int) -> int:
# write code here
dp = [0] * (n + 1) #dp[i]表示有i个节点时,共有多少种树形
dp[0] = 1
for i in range(1, n+1): #对根节点大小分类
for j in range(0, i):
dp[i] = (dp[i] + dp[j] * dp[i-1-j] % 1000000007) % 1000000007
return dp[-1]
class Solution_cartelan:
#卡特兰数参考https://blog.nowcoder.net/n/46bb5cecb4e1444489ed20a796ef70a5?f=comment
factor = [1] * 200010
mod = 1000000007
def qp(self, a, b):
ans = 1
a %= self.mod
while b:
if b & 1:ans = ans * a % self.mod
b >>= 1
a = a * a % self.mod
return ans
def C(self, n, m):
if n < m: return 0
return self.factor[n] * self.qp(self.factor[m] * self.factor[n-m], self.mod-2) % self.mod
def Lucas(self, n, m):
if m == 0: return 1
return self.C(n % self.mod, m % self.mod) * self.Lucas(n // self.mod, m // self.mod)
def numberOfTree(self , n: int) -> int:
# write code here
self.factor[0] = self.factor[1] = 1
for i in range(2, 2 * n + 1):
self.factor[i] = self.factor[i-1]*i%self.mod
ans = (self.Lucas(2*n,n) - self.Lucas(2*n,n+1)) % self.mod
if ans < 0: ans += self.mod
return ans
###################################################
#将整数 n 分成 k 份,且每份不能为空,任意两个方案不能相同(不考虑顺序),问有多少种不同的分法, 答案对 109 + 7 取模
#法一动态规划:dp[i][j]表示数字i分为j份总共有多少种分法,根据每份不能为空,我们可以将划分的情况分为两种:
#1. 至少有一份分配了数字1; 2. 所有的份数都没有分配数字1(如果没有分配数字1,必然分配了大于1的数字)
#如果没有分配数字1,那么我们一定可以给j份都提前放一个数字1,剩下的就是dp[i][j-i]
#
#法二:隔板法直接计算
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# @param n int整型 被划分的数
# @param k int整型 化成k份
# @return int整型
#
class Solution:
def divideNumber(self , n: int, k: int) -> int:
# write code here
dp = [[0] * n for i in range(k)] #此处dp[i][j]表示j+1分成i+1份的方案数
dp[0] = [1] * n
for i in range(1, k):
for j in range(i, n):
dp[i][j] = (dp[i-1][j-1] % 1000000007 + dp[i][j-i-1] % 1000000007) % 1000000007
return dp[-1][-1]
###################################################
#给定一个二叉树root和一个整数值 sum ,求该树有多少路径的的节点值之和等于 sum 。
#该题路径定义不需要从根节点开始,也不需要在叶子节点结束,但是一定是从父亲节点往下到孩子节点
#法一:两次dfs,基于路径是否包含根节点讨论
#法二:一次dfs+hash表记录当前路径和
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# @param root TreeNode类
# @param sum int整型
# @return int整型
#
class Solution_dfs:
def FindPath(self , root: TreeNode, sum: int) -> int:
ret = 0
if not root: return 0
#若路径不包含根节点
if root.left: ret += self.FindPath(root.left, sum)
if root.right: ret += self.FindPath(root.right, sum)
ret += self.dfs(root, sum) #若路径以root开始
return ret
def dfs(self, root, sum):
'''以root开始的路径和为sum的路径条数'''
if not root: return 0
ret = 0
if root.val == sum: ret += 1 #若路径只包含一个节点
#若路径包含两个以上节点
if root.left: ret += self.dfs(root.left, sum - root.val)
if root.right: ret += self.dfs(root.right, sum - root.val)
return ret
class Solution_hashDFS:
def __init__(self):
#记录路径和及条数
self.mp = dict()
#last为到上一层为止的累加和
def dfs(self, root: TreeNode, sum: int, last: int) -> int:
#空结点直接返回
if root is None:
return 0
res = 0
#到目前结点为止的累加和
temp = root.val + last;
#如果该累加和减去sum在哈希表中出现过,相当于减去前面的分支
if (temp - sum) in self.mp:
#加上有的路径数
res += self.mp[temp - sum]
#增加该次路径和
if temp in self.mp:
self.mp[temp] += 1
else:
self.mp[temp] = 1
#进入子结点
res += self.dfs(root.left, sum, temp)
res += self.dfs(root.right, sum, temp)
#回退该路径和,因为别的树枝不需要这边存的路径和
self.mp[temp] -= 1
return res
def FindPath(self , root: TreeNode, sum: int) -> int:
#路径和为0的有1条
self.mp[0] = 1
return self.dfs(root, sum, 0)
###################################################
#给定一个字符串s,字符串s只包含以下三种字符: (,*,),请你判断 s是不是一个合法的括号字符串。合法括号字符串有如下规则:
#1.左括号'('必须有对应的右括号')';2.右括号')'必须有对应的左括号'(';3.左括号必须在对应的右括号前面;
#4.*可以视为单个左括号,也可以视为单个右括号,或者视为一个空字符;5.空字符串也视为合法的括号字符串
#法一:利用两个栈存储 ( 和 * 并模拟匹配过程
#注1:如果遇到右括号,则需要有一个左括号或星号和右括号匹配,由于星号也可以看成右括号或者空字符串,因此当前的右括号应优先和左括号匹配,没有左括号时才和星号匹配
#注2:最终遍历完,可能两个栈中还有值。为了将每个左括号匹配,需要将星号看成右括号,且每个左括号必须出现在其匹配的星号之前。
#当两个栈都不为空时,每次从左括号栈和星号栈分别弹出栈顶元素,对应左括号下标和星号下标,若左括号下标小于星号下标则可匹配,否则返回 false;最终判断左括号栈是否为空
#
#法二:贪心匹配
#正序和倒序遍历两遍,分别把星号当做左括号和右括号。
#从左往右遍历,把星号视为左括号,在遍历的过程中如果出现左括号数量比右括号数量少就返回false(说明即使将 * 作为左括号时,右括号还有剩余)
#然后从右往左遍历,把星号视为右括号,在遍历的过程中如果出现右括号数量比左括号少就返回false(说明即使将 * 作为右括号时,左括号还有剩余)
#若未返回false,则说明匹配成功,多余的 * (如果有的话)可以作为空字符,返回true
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# @param s string字符串
# @return bool布尔型
#
class Solution_stack:
def isValidString(self , s: str) -> bool:
# write code here
left, star = [], []
for i in range(len(s)):
if s[i] == '(': left.append(i)
elif s[i] == '*': star.append(i)
else:
if left: left.pop()
elif star: star.pop()
else: return False
while left and star: #类似于((())(**(()*()
if left.pop() > star.pop(): return False
return not left
class Solution_greedy:
def isValidString(self , s: str) -> bool:
flag = 0
for i in range(len(s)):
if s[i] == ')': flag -= 1
else: flag += 1
if flag < 0: return False
flag = 0
for i in range(len(s)-1,-1,-1):
if s[i] == '(': flag -= 1
else: flag += 1
if flag < 0: return False
return True
###################################################
# 给定一个字符串形式的数字序列,请问能否由这个字符串拆分成一个累加序列。
# 累加序列:至少包含三个数,除了最开始的两个数外,每个数都是前两个数之和。
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
# @param arr string字符串
# @return bool布尔型
#
class Solution:
def AdditiveArray(self , arr: str) -> bool:
# write code here