10-kangrae-jo

[kangrae-jo]

🔗 문제 링크

[구간 합 구하기 4]

✔️ 소요된 시간

9m

✨ 수도 코드

구간합 문제를 살펴보던중에 실버3 인데 정답률이 30%대인 문제를 발견해서
들어가서 풀어봤습니다.

처음에는 그냥 입력받고 나서, 구간이 주어지면 구간만큼 누적합을 하는 문제로 파악을 했습니다.
그런데.... 그렇게 구현을 하니까 시간 초과가 뜨는 것입니다.
(문제를 딱 보면 시간초과겠구나 라는 걸 알게되는 경지가 되고싶습니다.....)

시간초과를 해결할 방법은 바로 떠올랐습니다.

처음 숫자 배열 입력을 그냥 받지말고, 1번째부터 i번째 까지의 합을 i번째 자리에 넣게끔 수정했습니다.
(계산의 편의를 위해 index를 1부터 count했습니당)

그러면 i번째 부터 j번째 까지의 누적합은,
num[j] - num [i-1]
로 표현될 수 있습니다!

📚 새롭게 알게된 내용

입력을 받을때부터 우리가 계산하기 유리한 형태로 바꾸는 방식을 경험해 봤습니다!

[g0rnn]

저는 이 문제 처음보고 2차원 배열을 만들어야하나 생각했습니다. i와 j를 입력받고 먼저 계산한 후 그 값을 2차원 배열 [i][j] 번째에 저장하려고 했었습니다. 근데 이게 되나.? 메모리는 될거 같은데 강래 코드도 아마 2차원으로 했겠지? 하면서 pr을 보는데 완전 간단하게 구현되었더군요.... ㄷㄷ

그래서 코드는 거의 같습니다. 입력받는 부분이 아주 약간 다른것 빼고는 다 똑같네요

code

    for (int i=1; i<=n; i++) {
        cin >> k;
        arr[i] = k + arr[i-1];
    }

[g0rnn]

여기에 0을 넣어도 작동이 되는군요

[kangrae-jo]

저때 운좋게 정답 풀이가 빠르게 떠올랐던 것 같네요

그리고 저는 저기에 NULL도 쓰고 0도 쓰고 구분없이 그냥 손 가는대로 쓰는데,
c에서 NULL은 0으로 컴파일 되는 걸로 알고 있어요(또는 정수 0아스키가 NULL이라서 그럴까...요?)
어쩄든 그래서 잘 작동하나봐요 ㅋㅋㅋㅋ

[kokeunho]

저도 우선 풀어봤는데 시간 초과가 떠서
강래님 수도코드를 보고 풀었습니다.
숙련도가 어느정도 쌓이면 이런 시간 제한이나 메모리 제한 문제를 많이 풀어봐야겠습니다. (저도 딱 보고 제한 조건에 맞춰서 풀이를 낼 경지에 도달하고 싶네요..)

code

   import java.util.*;

   public class Main {
       public static void main(String[] args) {
           Scanner sc = new Scanner(System.in);
           int n = sc.nextInt();
           int m = sc.nextInt();
           int[] nums = new int[n+1];
           int[] result = new int[m];

           nums[0] = 0;

           for(int i = 1; i <= n; i++) {
               nums[i] = nums[i-1] + sc.nextInt();
           }

           for(int i = 0; i < m; i++) {
               int s = sc.nextInt();
               int e = sc.nextInt();

               result[i] = nums[e] - nums[s-1];
           }

           for(int i = 0; i < m; i++) {
               System.out.println(result[i]);
           }
       }
   }

[wnsmir]

아무생각없이

N, M = map(int, input().split())

sum = 0
num = list(map(int, input().split()))

for i in range(M):
    start, end = map(int, input().split())
    for j in range(end-start+1):
        atom = num[start -1+j]
        sum += atom
    print(sum)
    sum = 0

이렇게 풀었는데
바로 시간초과뜨더라구요 ㅎ,,

근데 제가푼걸 보니 N값을 사용도 안했습니다 ㅋㅋㅋ

뭔가 잘못됨을 직감하고 시간복잡도를 어떻게 줄일지 고민해보았습니다.

pr 5678문제로 동적배열을 풀었었는데
그때 미리 동적배열에 값들을 계산해나가며 다 저장해놓으면
굳이 반복문을 돌며 구간합을 더하고있을 필요가없고 찾기만 하면 됩니다.

동적배열 즉 리스트를 미리 만들어놔야하니 N값이 이제야 쓰이더군요 ..

N, M = map(int, input().split())

# 입력 배열 및 누적 합 배열 생성
nums = list(map(int, input().split()))
prefix = [0] * (N + 1)  # 동적 배열(리스트) 초기화

# 누적 합 배열 채우기
for i in range(1, N + 1):
    prefix[i] = prefix[i - 1] + nums[i - 1]

# 각 구간 합 계산
for _ in range(M):
    start, end = map(int, input().split())
    print(prefix[end] - prefix[start - 1])