작심삼일

[LeetCode] 389 | Find the Difference | Python 본문

스터디/코테

[LeetCode] 389 | Find the Difference | Python

yun_s 2022. 2. 7. 09:44
728x90
반응형

문제 링크: https://leetcode.com/problems/find-the-difference/


문제

You are given two strings s and t.

String t is generated by random shuffling string s and then add one more letter at a random position.

Return the letter that was added to t.


조건

  • 0 <= s.length <= 1000
  • t.length == s.length + 1
  • s and t consist of lowercase English letters.

내 풀이

s를 random shuffle한 것이 t이므로 둘 다 sorting을 진행한다.

앞에서부터 하나씩 비교하며 둘이 다른 것이 있을 때 그것을 바로 리턴하면 된다.

s가 끝날 때까지 다른 것이 없다면, t의 마지막 글자가 다른 것이므로 그것을 리턴한다.


코드

class Solution:
    def findTheDifference(self, s: str, t: str) -> str:
        s, t = sorted(s), sorted(t)
        for n in range(len(s)):
            if s[n] is not t[n]:
                return t[n]
        
        return t[-1]
728x90
반응형
Comments