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.