StepByStep #짧은노래

[hwangnk1004]

• 문제 http://59.23.150.58/30stair/snotes/snotes.php?pname=snotes -농부 존은 그의 소들에게 노래를 가르키고 있다. 노래는 N(1 <= N <= 10,000 ) 개의 악보(notes)로 이루어져 있고 , 각 i 번째 악보는 Bi( 1 <= Bi <= 120 ) 비트동안 지속된다. (노래는 1,200,000 보다는 길지 않다) 소들은 시각 0 에 노래를 시작한다. 그러므로 첫 번째 악보에서는 시각 0 에서 B1 - 1 동안 두 번째 악보에서는 시각 B1 에서 B1 + B2 - 1 .. 소들은 노래가 너무 길고 지겨워 노래에 흥미가 떨어졌다. 그래서 소들에게 집중을 시키려고 한다. 시각 Ti ( 0 <= Ti < 노래의 총 시간) 의 비트 동안 , 어떤 악보가 연주되는지를 그들에게 Q ( 1 <= Q <= 50,000) 개의 질문을 던진다. 소들은 당신의 도움이 필요하다. 예를 들어 다음은 1 악보에서 2 시간이 , 2 악보에서는 1 시간이 , 3 악보에서는 3 시간이 소요되는 경우이다. 악보 1 1 2 3 3 3 +---+---+---+---+---+---+ 시각 0 1 2 3 4 5 이 경우 5 개 질의에 대한 답이다. 질의 답 2 2 3 3 4 3 0 1 1 1

• 답

import java.util.Scanner;

public class Main { public static void main(String args[]) {

    Scanner scanner = new Scanner(System.in);

    int n = scanner.nextInt();
    int q = scanner.nextInt();

    int arr[] = new int[120*n];

    int acc = 0;
    int k = 1;

    for (int i = 1; i <= n; i++) {
        int count = scanner.nextInt();
        int x = 0;

        while (true) {
            if (count == x++) {
                k++;
                break;
            }

            arr[acc++] = k;

        }
    }

    for (int i = 0; i < q; i++) {
        System.out.println(arr[scanner.nextInt()]);
    }

}

}