etc./Leetcode

Leetcode - 28. Implement strStr() - Easy

솔웅 2022. 8. 18. 08:45
반응형

이번 문제는 아주 쉬운 문제 입니다.

입력값은 두 스트링 입니다.

건초더미 에서 바늘 찾기를 연상 케 하는 이름인데요. Haystack 과 needle 입니다.

- Needle이 haystack에 포함돼 있으면 몇번째에 needle이 시작하는지 그 값을 리턴합니다.

- 포함돼 있지 않으면 -1을 리턴하구요.

- needle 이 empty면 0을 리턴합니다.

 

 

이렇게 패턴은 아주 간단합니다.

 

코드는 심플하게 String 관련된 메소드 들 중 isEmpty(), contains() 그리고 split() 을 사용했습니다.

 

class Solution {
    public int strStr(String haystack, String needle) {
        String[] haySplit = haystack.split(needle);
        
        if(!haystack.contains(needle)) {
            return -1;
        } else if (haystack.equals(needle) || haySplit.length ==0 || needle.isEmpty()) {
            return 0;
        } else {
            return haySplit[0].length(); 
        }
    }
}

 

다른 사람들이 한 것을 보니까 막 for 문을 돌리고 그러던데... 이렇게 하는게 아주 간단하지 않을까요?

반응형