[검 지 offer] min 함 수 를 포함 하 는 창고 (최소 창고)

1378 단어 검지 제공
제목:
스 택 의 데이터 구 조 를 정의 합 니 다. 이 형식 에서 스 택 에 포 함 된 최소 요 소 를 얻 을 수 있 는 min 함수 (시간 복잡 도 는 O (1) 를 실현 하 십시오.
메모: 테스트 중 스 택 이 비어 있 을 때 스 택 에 pop () 또는 min () 또는 top () 방법 을 호출 하지 않도록 합 니 다.
생각:
스 택 2 개 사용 하기 
창고 스 택  push 모든 값 (한 값 이면 push 들 어가 기)  다른 창고 stacklittle  push 비 원 stacklittle 창고 꼭대기 원소 작은 값
스 택 때문에. pop 이후 최소 값 이 변 할 수 있 습 니 다. 
stack pop 의 요소 와 스 택 stacklittle 스 택 상단 요 소 는 이 스 택 의 최소 값 에 영향 을 주지 않 습 니 다.  그래서 pop 원래 스 택 지붕 의 값 만 스 택 의 최소 값 에 영향 을 주지 않 습 니 다.
그러나. stack 창고 꼭대기 원소 와 stacklittle 스 택 꼭대기 요소 가 같 을 때 두 스 택 은 동시에 pop 이 필요 합 니 다. 그렇지 않 으 면 스 택 의 최소 값 이 맞지 않 습 니 다. min () 함수 가 스 택 꼭대기 요 소 를 되 돌려 주기 때 문 입 니 다.
코드:
class Solution {
  private:
	stack _stack;
	stack _stack_little;
public:
	void push(int value) {
		if (_stack.empty()) {
			_stack.push(value);
		}
		if (_stack_little.empty()) {
			_stack_little.push(value);
		}
		else if (value <= _stack_little.top()) {
			_stack_little.push(value);
		}
			_stack.push(value);
	}
	void pop() {
		if (_stack.empty()) {
			return;
		}
		if (_stack.top() == _stack_little.top()) {
			_stack.pop();
			_stack_little.pop();
		}
		else {
			_stack.pop();
		}
	}
	int top() {
		if (!_stack.empty()) {
				return _stack.top();
		}
	}
	int min() {
		if (!_stack_little.empty()) {
			return _stack_little.top();
		}
	}
};

좋은 웹페이지 즐겨찾기