Capitalize in python - HackerRank solution

Solution for hackerRank problem Find a string in python

Problem :

You are asked to ensure that the first and last names of people begin with a capital letter in their passports. For example, alison heck should be capitalised correctly as Alison Heck.

Given a full name, your task is to capitalize the name appropriately.

Input Format

A single line of input containing the full name,S.

Constraints

0 < len(S) < 1000

The string consists of alphanumeric characters and spaces.

Note: in a word only the first character is capitalized. Example 12abc when capitalized remains 12abc.

Output Format

Print the capitalized string,S

Sample Input

chris alan

Sample Output

Chris Alan.

Procedure

1. Start.

2. Loop through each element of the string.

    => if the character is in lowercase and if the previous element is a space and if the character is not a number

        => convert the lower case to upper case

    => else continue

3. Print the transformed string.

4. End.

Code

#!/bin/python3

import math

import os

import random

import re

import sys

# Complete the solve function below.

def solve(s):

    s = list(s)

    for i in range(len(s)):

        if(i==0)and(s[i].isalpha())and(s[i].islower()):

            s[i] = chr(ord(s[i])-32)

        if(s[i]==" ")and(s[i+1]!=" ")and(s[i+1].islower()):

            if(s[i+1].isalpha()):

                s[i+1] = chr(ord(s[i+1])-32)

    Capital = ""

    return Capital.join(s)

    

if __name__ == '__main__':

    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    s = input()

    result = solve(s)

    fptr.write(result + '\n')

    fptr.close()