forked from mandliya/algorithms_and_data_structures
-
Notifications
You must be signed in to change notification settings - Fork 0
/
length_of_last_word.cpp
56 lines (48 loc) · 1.75 KB
/
length_of_last_word.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/*
* Given a string consisting of lower/upper case alphabets and empty space character ' '.
* Return the length of last word in the string.
*
* A word is character sequence containing non-space characters only.
*
* Return 0, if no last word exist.
*
* Approach:
*
* Start from end of the string, and trim the whitespace.
* When we find the first non-space character, find the next space or till we reach beginning of string.
*
*/
#include <iostream>
int get_length_of_last_word(const std::string& str)
{
if (str.length() == 0) {
return 0;
}
int n = str.length();
int i = n - 1;
int j = 0;
// Trim the whitespace in the end
while(i >= 0 && str[i] == ' ') i--;
// Check if we already reached beginning of the string.
if (i < 0) {
return 0;
}
// Now start from the first non-space character
for (j = i; j >= 0; --j) {
if (str[j] == ' ') break;
}
return i - j;
}
int main()
{
std::string str1 = "Hello";
std::string str2 = "Hello ";
std::string str3 = "Hello World!";
std::cout << "String without any space of length " << str1.length() << ": " << str1 << std::endl;
std::cout << "Length of last word :" << get_length_of_last_word(str1) << std::endl;
std::cout << "String with lot of space in the end, with length of string " << str2.length() << ": " << str2 << std::endl;
std::cout << "Length of last word :" << get_length_of_last_word(str2) << std::endl;
std::cout << "String with lot of space in the middle, with length of string " << str3.length() << ": " << str3 << std::endl;
std::cout << "Length of last word :" << get_length_of_last_word(str3) << std::endl;
return 0;
}