HDU 1029 Ignatius and the Princess IV （数据结构）

http://acm.hdu.edu.cn/showproblem.php?pid=1029

这道题最容易想到的写法就是用一个数组来记录每个数字出现的次数。但是，如果数的范围大的话就不太友好了，容易超时。

然后一开始做的时候也是很纠结。然后就了解到了这个写法~ 真的很厉害！（题解思路来自灯神~）

这里给出的是n个数字，而且n是奇数。这里不用太纠结。。。反正最多的那个数最后一定能留下来。不用排序。

先初始化一个栈。

读取整数的时候判断：1、栈是否为空。　　2、栈顶元素是否等于当前读入的元素 - > (1)相等 (2)不相等

若栈为空，则将当前的数字放进栈内。

若相等，则进栈。否则出栈。

最后输出栈顶元素就行啦~

代码：

 

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>


#define N 100010

using namespace std;

typedef long long int ll;

stack<int> a;
int main()
{
    int n;
    while(cin>>n){
        int num;
        for(int i=0; i<n; i++){
            cin>>num;
            if(a.empty()){
                a.push(num);
            }
            else if(a.top()==num){
                a.push(num);
            }
            else{
                a.pop();
            }
        }
        cout<<a.top()<<'\n';
        while(!a.empty()) a.pop();
    }
    return 0;
}

 

最后不要忘了清空栈，因为是多实例~

更加优化的写法 (空间复杂度O ( 1 ) ) :

不设置栈，而是用一个变量来储存我们需要的那个值。

然后设置一个计数器，判断栈是否为空。

 

#include <bits/stdc++.h>

#define N 100010

using namespace std;

typedef long long int ll;

//stack<int> a;
int main()
{
    int n;
    while(cin>>n){
        int num, cnt, cnd;
        cnt=0;
        for(int i=0; i<n; i++){
            cin>>num;
            if(cnt==0){
                cnd=num;
                cnt++;
            }
            else if(cnd==num){
                cnt++;
            }
            else cnt--;
        }
        cout<<cnd<<endl;
    }
    return 0;
}