[c++] 백준 - 10773: 제로

10773번: 제로 (acmicpc.net)

10773번: 제로

 

배열과 if문을 사용하며 푼 코드

#include <iostream>
using namespace std;

int main() {

	int K, tmp, index = -1, sum = 0; cin >> K;
	int* num = new int[K]; 
	for (int i = 0; i < K; i++) {
		cin >> tmp;

		if (tmp == 0) {
			num[index] == 0;
			index--;
		}
		else {
			index++;
			num[index] = tmp;
		}
	}

	for (int i = 0; i <= index; i++) sum += num[i];
	cout << sum;

	return 0;
}

 

 

 

깨달음을 얻고 스택을 이용한 코드 

#include <iostream>
using namespace std;

int top = -1;
int Num[100000];

void push(int e) {
	Num[++top] = e;
}
int pop() {
	return Num[top--];
}

int main() {

	int K, tmp, sum = 0; cin >> K;

	for (int i = 0; i < K; i++) {
		cin >> tmp;
		
		if (tmp == 0) pop();
		else push(tmp);
	}

	for (int i = 0; i <= top; i++) sum += Num[i];
	cout << sum;

	return 0;
}

 

스택을 동적할당으로 생성할순없을까..?? 왜 자꾸오류가뜨지