1. 问题描述:
在战略游戏中,玩家往往需要发展自己的势力来触发各种新的剧情。一个势力的主要属性有三种,分别是文明等级(C),资源储备(R)以及人口数量(H)。在游戏开始时(第 0 天),三种属性的值均为 0。随着游戏进程的进行,每一天玩家的三种属性都会对应增加,我们用一个二维数组 increase 来表示每天的增加情况。这个二维数组的每个元素是一个长度为 3 的一维数组,例如 [[1,2,1],[3,4,2]] 表示第一天三种属性分别增加 1,2,1 而第二天分别增加 3,4,2。所有剧情的触发条件也用一个二维数组 requirements 表示。这个二维数组的每个元素是一个长度为 3 的一维数组,对于某个剧情的触发条件 c[i], r[i], h[i],如果当前 C >= c[i] 且 R >= r[i] 且 H >= h[i] ,则剧情会被触发。根据所给信息,请计算每个剧情的触发时间,并以一个数组返回。如果某个剧情不会被触发,则该剧情对应的触发时间为 -1 。
示例 1:
输入: increase = [[2,8,4],[2,5,0],[10,9,8]] requirements = [[2,11,3],[15,10,7],[9,17,12],[8,1,14]]
输出: [2,-1,3,-1]
解释:
初始时,C = 0,R = 0,H = 0
第 1 天,C = 2,R = 8,H = 4
第 2 天,C = 4,R = 13,H = 4,此时触发剧情 0
第 3 天,C = 14,R = 22,H = 12,此时触发剧情 2
剧情 1 和 3 无法触发。
示例 2:
输入: increase = [[0,4,5],[4,8,8],[8,6,1],[10,10,0]] requirements = [[12,11,16],[20,2,6],[9,2,6],[10,18,3],[8,14,9]]
输出: [-1,4,3,3,3]
示例 3:
输入: increase = [[1,1,1]] requirements = [[0,0,0]]
输出: [0]
限制:
1 <= increase.length <= 10000
1 <= requirements.length <= 100000
0 <= increase[i] <= 10
0 <= requirements[i] <= 100000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ju-qing-hong-fa-shi-jian
2. 思路分析:
一开始的时候没有想到什么好的思路,于是看了一下力扣的题解,发现可以使用二分查找的思路进行解决,仔细想想使用二分查找的思路真是妙呀(一开始的时候感觉每一次都需要一一比对requirements的三个属性值感觉好麻烦所以是没有想到二分查找的思路的)。二分查找的思路如下:一开始的时候可以计算出每一天increase数组中三种属性的值,可以在循环中累加上一天的值就可以得到当前天数对应的属性值(累加的时候increase数组肯定是升序的所以可以使用二分查找),然后我们可以遍历requirements数组,使用二分查找找出当前的剧情i所需要的最短触发时间(左边的范围为0,右边的范围为increase数组的长度,二分查找的中间值对应的是天数),在二分查找的时候我们需要比较当前的中间天数mid对应的increase三个属性值是否大于等于了当前剧情requirements[i]的三个属性值,只有当都大于等于的时候才表示当前的天数是可以触发的,然后我们就可以缩短触发的时间,也就是二分查找对应的右边的范围要缩小,否则需要增加触发的时间,当循环结束的时候判断当前二分查找左边的范围l对应的increase[l]三个属性值是否都大于了requirements[i]的三个属性值,假如满足说明是可以触发相应的剧情的,不满足则说明最大的天数都不能够满足当前剧情i的触发时间的
3. 代码如下:
from typing import List
class Solution:
def getTriggerTime(self, increase: List[List[int]], requirements: List[List[int]]) -> List[int]:
# 计算当前天数i对应的三个属性值
for i in range(1, len(increase)):
increase[i][0] += increase[i - 1][0]
increase[i][1] += increase[i - 1][1]
increase[i][2] += increase[i - 1][2]
res = list()
for i in range(len(requirements)):
# 当触发条件对应的三个值都为0的时候那么当前的结果为0
if requirements[i][0] == 0 and requirements[i][1] == 0 and requirements[i][2] == 0:
res.append(0)
else:
l, r = 0, len(increase) - 1
# 寻找当前达到requirements的最少的天数
while l <= r:
mid = (l + r) // 2
if increase[mid][0] < requirements[i][0] or increase[mid][1] < requirements[i][1] or increase[mid][
2] < requirements[i][2]:
l = mid + 1
else:
r = mid - 1
# 只有当当前查找的最终天数对应的increase数组对应的三个属性值都大于了requirements之后说明才能够触发的
if l < len(increase) and increase[l][0] >= requirements[i][0] and increase[l][1] >= requirements[i][1] and increase[l][2] >= requirements[i][2]:
res.append(l + 1)
else:
res.append(-1)
return res