[Leet Code] 58. Length of Last Word - Python

Level : Easy
Topics : String

 

https://leetcode.com/problems/length-of-last-word/description/

Code

class Solution:
    def lengthOfLastWord(self, s: str) -> int:
        # 1. str의 뒤 공백을 없애줍니다
        # 2. str을 뒤에서 부터 돌면서 ' '이 나온다면 break
        # 3. return cnt
        
        s = list(s)
        for i in range(len(s)-1, -1, -1):
            if s[i] == ' ':
                s.pop(i)
            else:
                break
        # last word 개수 세기
        cnt = 0
        for i in range(len(s)-1, -1, -1):
            if s[i] == ' ':
                return cnt
            else:
                cnt += 1
        return cnt
        
	# Time Complexity : O(n)
        # Space Complexity : O(1)