LCP 08. 剧情触发时间
在战略游戏中,玩家往往需要发展自己的势力来触发各种新的剧情。一个势力的主要属性有三种,分别是文明等级(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
代码
用二分法搜寻
class Solution {
public:
vector<int> getTriggerTime(vector<vector<int>>& increase, vector<vector<int>>& requirements) {
int rowr = requirements.size(); //int colr = requirements[0].size();
vector<int>res(rowr, -1);//对返回数组进行初始化,都设置为未触发-1.
increase.insert(increase.begin(), { 0,0,0 });//把初始第0天加入,防止requirements[0]={0,0,0}.也方便于之后给res赋值的时候不用给下标+1。
int rowi = increase.size();//int coli = increase[0].size();
for (int i = 1; i < rowi; ++i) {
for (int j = 0; j < 3; ++j)
increase[i][j] += increase[i - 1][j];//increase各行叠加。
}
//二分法搜寻
for (int i = 0; i < rowr; ++i) {
int l = 0, h = rowi;
int j;
while (l < h) {
j = (h + l) / 2;
if (increase[j][0] >= requirements[i][0] && increase[j][1] >= requirements[i][1] && increase[j][2] >= requirements[i][2])
h = j;
else
l = j + 1;
}
if (l != rowi)res[i] = l;//判断是否触发。
}
//myprint2(increase);
//myprint2(requirements);
//myprint1(res);
return res;
}
};
测试:
int main() {
// [[2, 8, 4], [2, 5, 0], [10, 9, 8]]
int a[][3] = { 2, 8, 4,2, 5,0,10, 9, 8 };
//[[2, 11, 3], [15, 10, 7], [9, 17, 12], [8, 1, 14]]
int b[][3] = { 2, 11, 3 ,15, 10, 7 ,9, 17, 12 ,8, 1, 14 };
vector<vector<int>>vec(3, vector<int>(3));
vector<vector<int>>vec1(4, vector<int>(3));
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 3; ++j)
vec[i][j] = a[i][j];
}
for (int i = 0; i < 4; ++i) {
for (int j = 0; j < 3; ++j)
vec1[i][j] = b[i][j];
}
Solution S;
S.getTriggerTime(vec, vec1);//结果为2 -1 3 -1
return 0;
}