题目描述:
在战略游戏中,玩家往往需要发展自己的势力来触发各种新的剧情。一个势力的主要属性有三种,分别是文明等级(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 。
题解:
首先想到暴力解法,每次从0递增C、R和H,每次遍历时将满足条件的剧情触发时间修改即可。这样时间复杂度变为O(mn)。并不是一个好的解法,面对数据过多的时候可能会超时。其实我们可以看到C、R和H随着天数都是递增的。因此我们可以构造包含C、R和R的递增数列。然后遍历剧情,在循环内部二分查找满足剧情的最低天数即可,这样时间复杂度就变成了O(mlogn)。
代码如下:
class Solution {
public:
vector<int> getTriggerTime(vector<vector<int>>& increase, vector<vector<int>>& requirements) {
vector<vector<int>> s(increase.size()+1,vector<int>(3,0));
for(int i=0;i<increase.size();i++){
for(int j=0;j<3;j++){
s[i+1][j] = s[i][j] + increase[i][j];
}
}
vector<int> ans;
for(auto v:requirements){
int l=0, r = increase.size();
while(l<r){
int m = (l+r)/2;
if(s[m][0]>=v[0]&&s[m][1]>=v[1]&&s[m][2]>=v[2])
r = m;
else
l = m+1;
}
if(s[l][0]>=v[0]&&s[l][1]>=v[1]&&s[l][2]>=v[2])
ans.push_back(l);
else
ans.push_back(-1);
}
return ans;
}
};