leetcode(18)python基本操作
数据结构
整型
正无穷:
a = float('inf')
负无穷:
b = float('-inf')
向下取整:
mid = (l + r) // 2
防止溢出:
mid = left + (right - left) // 2
字符型
反转字符串
ans = ans[::-1]
列表list
nums[: i] + nums[i + 1:] # 拼接两个数组
path + [nums[i]] # 拼接一个数组和一个数字
向列表中添加列表:
result.append(path[:])
path[:]和 path.copy()才是等价的
python是动态语言,变量名都只是对对象的引用,对于数组,字典,set等可变对象,后面的Path和结果数组里面的Path是同一个对象,一个改变,另一个也改变,了解一下copy和deepcopy
字典类型dict
对于普通的dict,key不存在就报错。
但是对于defaultdict,key不存在就会返回默认值。
# defaultdic(list) 是为了方便直接append
tickets_dict = defaultdict(list)
for item in tickets:
tickets_dict[item[0]].append(item[1])
'''
tickets_dict里面的内容是这样的
{'JFK': ['SFO', 'ATL'], 'SFO': ['ATL'], 'ATL': ['JFK', 'SFO']})
'''
332. 重新安排行程
hashmap
使hashmap里的元素频率+1:
hashmap[s[end]] = hashmap.get(s[end], 0) + 1
169. 多数元素
用一个循环遍历数组 nums 并将数组中的每个元素加入哈希映射中。遍历哈希映射中的所有键值对,返回值最大的键。
counts = collections.Counter(nums)
return max(counts.keys(), key=counts.get)
hashset
lookup = set()
while s[i] in lookup:
lookup.remove(s[left])
lookup.add(s[i])
双端队列
from collections import deque
que = deque([root])
cur = que.popleft() #返回并删除队首元素
cur = que.pop() #返回并删除队尾元素
优先队列
import heapq
dummy = ListNode(0)
heapq.heappush(head, (lists[i].val, i))
val, idx = heapq.heappop(head)
while l:
heapq.heappush(minHeap, l.val) #把l中的数据逐个加到堆中
l = l.next
val = heapq.heappop(minHeap) #依次弹出最小堆的数据
tuple、list、dictionary操作
运算符
/ 和 //
/
表示浮点整除法,返回浮点结果,也就是结果为浮点数;
而//
表示整数除法,返回大于结果的一个最大的整数,意思就是除法结果向下取整。
内置函数
range()
从后往前遍历到0:
for i in range(length - 1, -1, -1):
dp = [[0]*(len2 + 1) for _ in range(len1 + 1)]
先列再行:
dp = [[0 for _ in range(len2 + 1)] for _ in range(len1 + 1)]
sort()与sorted()
nums.sort()
sorted(str1)
stones = sorted(zip(rocks, capacity), key=lambda x: (x[1] - x[0])) # 自定义排序函数
stockPrices.sort(key=lambda x: x[0]) # 二维数组按照第0列排序
6075. 装满石头的背包的最大数量
sum()
对数组求和
left_count = sum(1 for i in range(lo, hi + 1) if nums[i] == left)
sum_ = sum(nums)
decimal.Decimal()
高精度(默认28位,可调更高)
import decimal
斜率 = decimal.Decimal(y1-y0)/decimal.Decimal(x1-x0)
6076. 表示一个折线图的最少线段数
zip()
for x, y in zip(capacity, rocks)
split()
默认会将字符串用空白或者换行符分割,并且返回一个列表
注意:字符串中间无论有多少个空格,它都会按一个空白进行分割。
s = '1 2 3 4 5 6'
a = s.split(' ')
# 输出:['1', '2', '3', '4', '5', '', '', '', '6']
strip()
默认将字符串首部与尾部的空白去掉,直到遇到非空的元素,并且返回一个字符串
s = '1 2 3 4 5 6'
b = s.strip()
# 输出:'1 2 3 4 5 6'
iloc() ilx() loc()
参考资料:
菜鸟教程