Python - 문자열 문제 1

Q . 문자열 리스트 중에서 첫 자와 끝자가 같은 문자열이 몇 개 있는지 알아보는 프로그램을 작성해 보시오. 단 문자열은 2개 이상의 문자로 이루어져 있을 때만 검사하고 문자가 하나인 경우는 검사하지 않는다. 문자열 리스트 검사하는 함수는 checkWord로 만든다.

 

예시) 

list=[‘hello’, ‘k’, ‘abc’, ‘12341’]   ->  1개
list=[‘hello’, ‘abc’, ‘121’, ‘aba’]   ->  2개

---

def checkWord(listVal):
    count = 0

    for i in listVal:

        if(len(i) > 1) :
            tmp1 =''
            tmp2 =''

            for j in range(len(i)):

                if(j == 0):
                    tmp1 = i[0]

                if(j == len(i)-1):
                    tmp2 = i[j]

                if(tmp1 == tmp2):
                    count+=1

    print(count, '개')

list=['hello','k','abc','12341']

checkWord(list)