aboutsummaryrefslogtreecommitdiff
path: root/capitalise_the_title.py
blob: 3ba759592a126faf228b2121ca4ee131a1abea04 (plain)
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
# 2129. Capitalize the Title

"""
given a string 'title' consisting of one or more words separated by a single
space. where each word consists of english letters. capitalise the string by
changing the capitalisation of each word such that if the length of the word
is 1 or 2 letters, change all letters to lowercase. otherwise,change the
first letter to uppercase and the remaining letters to lowercase. return the
capitalised title.
"""


class Solution(object):
    def capitalizeTitle(self, title):
        """
        :type title: str
        :rtype: str
        """
        return " ".join(
            [word.lower() if len(word) < 3 else word.title() for word in title.split()]
        )


if __name__ == "__main__":
    obj = Solution()
    print(obj.capitalizeTitle(title="capiTalIze tHe titLe"))
    print(obj.capitalizeTitle(title="First leTTeR of EACH Word"))
    print(obj.capitalizeTitle(title="i lOve leetcode"))