C++程序:统计从列表左右两端可解决的问题数量


假设我们有一个包含n个元素的数组A和另一个数字k。假设一场比赛中有n道题。Amal的解题能力为k。Amal总是从列表的任意一端解题。他不能解决难度大于k的问题。当左右两端问题的难度都大于k时,他停止解题。我们必须计算他可以解决多少问题。A[i]表示第i道题的难度。

因此,如果输入类似于A = [4, 2, 3, 1, 5, 1, 6, 4]; k = 4,则输出为5,因为最初从左边解决难度为4的问题,然后从右边解决难度为4的问题,然后无法再解决右边的问题,然后从左边解决难度为2、3和1的问题。总共可以解决5道题。

步骤

为了解决这个问题,我们将遵循以下步骤:

n := size of A
l := 0
r := n - 1
for initialize i := 0, when i < n, update (increase i by 1), do:
   if A[i] <= k and l is same as i, then:
      (increase l by 1)
while A[r] <= k, do:
   (decrease r by 1)
if l is same as n, then:
   return n
Otherwise
   return n - 1 - r + l

示例

让我们看看下面的实现来更好地理解:

#include <bits/stdc++.h>
using namespace std;

int solve(vector<int> A, int k) {
   int n = A.size();
   int l = 0, r = n - 1;
   for (int i = 0; i < n; ++i) {
      if (A[i] <= k && l == i)
         ++l;
   }
   while (A[r] <= k)
      --r;
   if (l == n)
      return n;
   else
      return n - 1 - r + l;
}
int main() {
   vector<int> A = { 4, 2, 3, 1, 5, 1, 6, 4 };
   int k = 4;
   cout << solve(A, k) << endl;
}

输入

{ 4, 2, 3, 1, 5, 1, 6, 4 }, 4

输出

5

更新于:2022年3月3日

226 次浏览

开启你的职业生涯

完成课程获得认证

开始
广告
© . All rights reserved.