階段の数と各階段のステップ数をカウントするC++プログラム

n個の要素を持つ配列Aがあるとします。アマルが高層ビルの階段を上っていくようにしましょう。登るたびに、1から数え始めます。たとえば、3段と4段の2つの階段を上る場合、1、2、3、1、2、3、4のような数字を話します。配列Aでは、数字はアマルが言った階段の数字を表しています。彼が登った階段の数を数え、各階段の階段の数も印刷する必要があります。

したがって、入力がA =[1、2、3、1、2、3、4、5]の場合、出力は2、[3、5]

ステップ

これを解決するには、次の手順に従います-

p = 0
n := size of A
for initialize i := 0, when i < n, update (increase i by 1), do:
   if A[i] is same as 1, then:
      (increase p by 1)
print p
for initialize i := 1, when i < n, update (increase i by 1), do:
   if A[i] is same as 1, then:
      print A[i - 1]
print A[n - 1]

例

理解を深めるために、次の実装を見てみましょう-

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

void solve(vector<int> A) {
   int i, p = 0;
   int n = A.size();
   for (i = 0; i < n; i++) {
      if (A[i] == 1)
         p++;
   }
   cout << p << endl;
   for (i = 1; i < n; i++) {
      if (A[i] == 1)
         cout << A[i - 1] << ", ";
   }
   cout << A[n - 1];
}
int main() {
   vector<int> A = { 1, 2, 3, 1, 2, 3, 4, 5 };
   solve(A);
}

入力

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

出力

2
3, 5