From a28b8d9d26fca289f8d0327e2b5d12ff9e99f516 Mon Sep 17 00:00:00 2001 From: rl544 <111494264+rl544@users.noreply.github.com> Date: Fri, 30 Dec 2022 16:29:18 +0900 Subject: [PATCH] =?UTF-8?q?updated=20at=202022.=2012.=2030.=20=EC=98=A4?= =?UTF-8?q?=ED=9B=84=204:29:19?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- programmers/n^2 배열 자르기/solution.java | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) 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