Extend your head command to take an integer n and return the first n lines of lines, joined by newline characters (\n). For this version you may assume 0 ≤ n ≤ lines.size(). If n is 0 the result is the empty string.
Function signature:
string head(vector<string> lines, int n)
Example:
Input: lines = ["alpha", "beta", "gamma", "delta", "epsilon"], n = 3 Output: "alpha\nbeta\ngamma" When printed: alpha beta gamma
Loop from
0 to n - 1, accumulating each line. Insert a "\n" before every line except the first.
for (int i = 0; i < n; i++) {
if (i > 0) result += "\n";
result += lines[i];
}
The full solution:
class Solution {
public:
string head(vector<string> lines, int n) {
string result;
for (int i = 0; i < n; i++) {
if (i > 0) result += "\n";
result += lines[i];
}
return result;
}
};
Free plan: 30 tutor messages. Premium is unlimited.
Saved
Ctrl + Enter to check
