인프런 커뮤니티 질문&답변

강의내용의 코드가 헷갈려서 아래내용대로 수정해보았는데 이렇게 해도 될까요 ?

public static void main(String[] args) {

Scanner sc = new Scanner(System.in);

int N = sc.nextInt(); // 수열의 개수

int A[] = new int[N]; // 수열을 저장할 배열

// 데이터 입력

for (int i = 0; i < N; i++) {

A[i] = sc.nextInt();

}

Stack<Integer> stack = new Stack<>();

StringBuffer bf = new StringBuffer(); // 연산 출력 저장

int num = 1;

for (int i = 0; i < N; i++) {

int su = A[i];

while (su >= num) { // 현재 수가 스택의 수와 같거나 큰 경우

stack.push(num++);

bf.append("+\n");

}

if (stack.isEmpty() || stack.peek() != su) {

System.out.println("NO");

return;

}

stack.pop();

bf.append("-\n");

}

System.out.println(bf.toString()); // 결과 출력

}

}