diff --git a/programmers/n^2 배열 자르기/solution.java b/programmers/n^2 배열 자르기/solution.java index 8ab528d..eee2eee 100644 --- a/programmers/n^2 배열 자르기/solution.java +++ b/programmers/n^2 배열 자르기/solution.java @@ -1,15 +1,8 @@ // [문제 링크]: https://school.programmers.co.kr/learn/courses/30/lessons/87390 +import java.util.stream.LongStream; class Solution { - public long[] solution(int n, long left, long right) { - long k = left/n, l = left%n, kn = (right)/n, ln = (right)%n; - long[] temp = new long[(int)(right-left+1)]; - for(long i = k; i <= kn; i++){ - for(long j=0; j< n; j++){ - if(i==k&&jln) continue; - else temp[(int)((i-k)*n+j-l)]=(Math.max(i,j)+1); - } - } - return temp; + public int[] solution(int n, long left, long right) { + return LongStream.rangeClosed(left, right).mapToInt(value -> (int) (Math.max(value / n, value % n) + 1)).toArray(); } } \ No newline at end of file