枚举算法
简介(第一天)
枚举算法的核心思想是:通过列举问题的所有状态,将它们逐一与目标状态进行比较,从而得到满足条件的解。
解题思路与优化
采用枚举算法解题的一般思路如下:
- 确定枚举对象、枚举范围和判断条件,并判断条件设立的正确性。
- 一一枚举可能的情况,并验证是否是问题的解。
- 考虑提高枚举算法的效率。
我们可以从下面几个方面考虑提高算法的效率:
- 抓住问题状态的本质,尽可能缩小问题状态空间的大小。
- 加强约束条件,缩小枚举范围。
- 根据某些问题特有的性质,例如对称性等,避免对本质相同的状态重复求解。
应用
01百钱百鸡问题
确定范围 0<= x,y,z <= 100
满足条件:1.5*x + 3*y + z//3
2. x+y+z = 100
def buyChicken():
for x in range(100):
for y in range(100):
for z in range(100):
if 5*x + 3*y + z//3 and x+y+z = 100:
print("公鸡 %s 母鸡 %s 小鸡 %s"%(x,y,z))
针对百钱百鸡问题的优化:
假设全买公鸡则 x为20,若全买母鸡则y为33.333近似为33
z可以用x和y表示
def buyChicken():
for x in range(21): #21取不到
for y in range(34): #34取不到
z = 100 - x -y
if 5*x + 3*y + z//3:
print("公鸡 %s 母鸡 %s 小鸡 %s"%(x,y,z))
运行结果
02两数相加
def twoSum(nums,target):
for i in range(len(nums)):
for j in range(i+1,len(nums)): #从i+1取值不需要再与j进行比较
if nums[i] == nums[j]:
return [i,j]
return []
运行结果
03计数质数
def isPrime(x):
for i in range(2,int(sqrt(x))+1):
if x % i == 0:
return False
return True
def countPrime(n):
count = 0
for i in range(2,n+1): #此处要从2处取值,若取值为1,则再判断是否为质数时,出现1%2=1,返回True,与事实结果不符
if isPrime(i):
count += 1
return count
运行结果
04统计平方和三元组的数目
def countTriples(n):
count = 0
for a in range(1,n+1):
for b in range(1,n+1):
c = int(sqrt(pow(a,2)+pow(b,2)+1))
if c < n and pow(a,2) + pow(b,2) == pow(c,2):
count += 1
return count
运行结果
练习(第二天)
01公因子的数目
解题思路:
1.假定给定 a 和 b ,添加一步判断如果a < b,则不变;a > b,则将b的值赋给 a,始终保持 a 的值是小的一方
2.确定小的一方后,遍历时可以缩小遍历范围,即遍历[2,a]区间的值
def countCommonFactor(a,b):
count = 0
if a > b:
a , b = b , a
for i in range(2,a+1):
if a % i == 0 and b % i == 0:
count += 1
return count
运行结果
02和为S的连续正数序列
解题思路:
1.外层循环遍历[1,target]
2.内层循环遍历 i 之后的值,直到 i 之后sum = target
def sumSort(target):
alist = []
for i in range(1,target):
sum = 0
j = i
while sum < target:
sum += j
j += 1
if sum == target:
seq = list(range(i,j))
alist.append(seq)
return alist
运行结果
03统计圆内格点数
解题思路:
1.首先要取出二维列表circles的值 x i , y i, r i
2.再对[xi-ri,xi+ri]和[yi-ri,ti+ri]的值遍历
3.引入集合去除重复点位
def countCircle(circles):
count = 0
points = set() #引入集合去除重复
for x in range(len(circles)):
xi,yi,ri = circles[x]
for i in range(xi-ri,xi+ri+1):
for j in range(yi-ri,yi+ri+1):
if sqrt(pow(i-xi,2)+pow(j-yi,2)) <= ri:
point = (i,j)
if point not in points:
count += 1
points.add(point)
return count
运行结果