글 읽기 - 예외 케이스에 대한 질문

minshigee 5년 전 0

질문 검색에 나와있는 예외케이스를 모두 넣어봐도 정답이 나옵니다만.... 틀렸다고 나옵니다.

혹시 소스를 한 번 봐주시면 감사합니다. 꾸벅

#include<stdio.h>
#include<stack>

using namespace std;

int N;
int Data[500005];
int D[500005];

void input() {
	scanf("%d", &N);
	for (int i = 0; i < N; i++) {
		scanf("%d", &Data[i]);
	}
}

stack<int> st;

long long solve() {

	long long res = 0;
	st.push(Data[0]);
	for (int i = 1; i < N; i++) {
		if (Data[i] >= st.top()) {
			int cnt = 0;
			while (!st.empty() && Data[i] >= st.top()) {
				if (Data[i] == st.top())
					cnt++;
				st.pop();
				D[i]++;
			}
			if (!st.empty())
				D[i]++;
			for (int i = 0; i < cnt; i++)
				st.push(Data[i]);
		}
		else {
			D[i] = 1;
		}
		st.push(Data[i]);
		res += D[i];
	}
	return res;
}

int main() {
	input();
	long long res = solve();
	printf("%lld\n", res);
	
	return 0;
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

예외 케이스에 대한 질문

minshigee 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말