[leetcode] Easy-58번. Search Insert Position 문제풀이

[leetcode] Easy-58번. Search Insert Position 문제풀이

728x90

문제

https://leetcode.com/problems/search-insert-position/

풀이 코드

package SITE03_leetcode.easy; import java.util.Objects; /** * https://leetcode.com/problems/search-insert-position/ */ public class E004_leetCode58_LengthofLastWord { public static void main(String[] args) { E004_leetCode58_LengthofLastWord solution = new E004_leetCode58_LengthofLastWord(); System.out.println(solution.lengthOfLastWord(" fly me to the moon ")); } public int lengthOfLastWord(String s) { int result = 0; boolean isInit = false; for (int i = s.length() - 1; i >= 0; i--) { if (!Objects.equals(String.valueOf(s.charAt(i)), " ")) { isInit = true; result++; } else { if (isInit) { break; } } } return result; } }

첫번째로 공백을 만났을 경우를 flag 변수를 통해 처리했다. 더 좋은 방법이 있을거같아, 확인해보았다.

더 좋은 방법

package SITE03_leetcode.easy; import java.util.Objects; /** * https://leetcode.com/problems/search-insert-position/ */ public class E004_leetCode58_LengthofLastWord { public static void main(String[] args) { E004_leetCode58_LengthofLastWord solution = new E004_leetCode58_LengthofLastWord(); System.out.println(solution.lengthOfLastWord2(" fly me to the moon ")); } public int lengthOfLastWord(String s) { if (null == s || s.trim().length() == 0) return 0; //" fly me to the moon " s = s.trim(); // "fly me to the moon" - 양 옆의 공백 제거 // 마지막 공백 기준으로 자르기 String lastWord = s.substring(s.lastIndexOf(" ") + 1); return lastWord.length(); } }

주석을 보고 코드를 이해하자. 문자열의 substring 메서드를 사용하여 간단하게 풀이할 수 있다. 참고로 앞뒤 공백 제거는 trim() 메서드 이외에도 JAVA11에 등장한 strip() 메서드도 사용할 수 있다.

from http://devfunny.tistory.com/658 by ccl(A) rewrite - 2021-12-15 00:01:37