C++程式用於統計樓梯數量以及每段樓梯的臺階數
假設我們有一個包含n個元素的陣列A。假設Amal在一個多層建築裡爬樓梯。每次他爬樓梯時,都從1開始計數。例如,如果他爬了兩段樓梯,分別有3個臺階和4個臺階,他就會說出像1、2、3、1、2、3、4這樣的數字。在陣列A中,這些數字代表Amal所說的樓梯號碼。我們需要統計他爬了多少段樓梯,並列印每段樓梯的臺階數。
所以,如果輸入像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
廣告