【LeetCode剑指offer 01】数组中重复的数字、两个栈实现队列

数组中重复的数字

数组中重复的数字

找出数组中重复的数字。

在一个长度为 n 的数组 nums 里的所有数字都在 0~n-1 的范围内。数组中某些数字是重复的,但不知道有几个数字重复了,也不知道每个数字重复了几次。请找出数组中任意一个重复的数字。

示例 1:

输入:
[2, 3, 1, 0, 2, 5, 3]
输出:2 或 3

限制:

2 <= n <= 100000

思路

很容易想到使用哈希表来记录元素出现次数

且因为题目规定了n的范围,所以我们使用的哈希表类型应该是map

代码

第一版

直接定义一个unordered_map

然后遍历数组,在哈希表对应位置计数,最后再遍历哈希表,返回第一个遇到的出现次数大于等于2的元素

class Solution {
public:
    int findRepeatNumber(vector<int>& nums) {
        //定义一个map
        unordered_map<int, int> hash;
        for(int i = 0; i < nums.size(); ++i){
            hash[nums[i]]++;
        }
        //在哈希表中找出现次数大于等于2的元素
        for(unordered_map<int, int>::iterator it = hash.begin(); it != hash.end(); ++it){
            if(it->second >= 2) return it->first;
        }
        return -1;
    }
};

这么做其实有点冗余,其实完全可以只用一个循环解决问题的

意思就是,我们其实可以不用记录元素出现的具体次数,只要在遍历时又碰到了即可直接返回

于是有了第二版

第二版
class Solution {
public:
    int findRepeatNumber(vector<int>& nums) {
        //定义一个map
        unordered_map<int, int> hash;
        //遍历数组,将元素加入哈希表,同时判断某元素是否重复出现
        for(int num : nums){
            if(hash[num]) return num;//当前遍历值出现在哈希表,为重复值,返回
            hash[num]++;//未出现的,则在哈希表中记录
        }
        return -1;
    }
};

用两个栈实现队列

用两个栈实现队列

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

示例 1:

输入:
["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
[[],[3],[],[],[]]
输出:[null,null,3,-1,-1]

示例 2:

输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]

提示:

1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

思路

一拿到题肯定会下意识联想到 232. 用栈实现队列

这题之前刷代码随想录的时候写过,于是很自然的照着以前的思路写

事实上,这两题的核心思路是一样的,即使用两个栈(stack_in和stack_out),分别模拟队列的进和出

模拟加入队列功能的栈stack_in就直接使用push功能就行;

模拟从队列拿数据时,要先判断 stack_out 是否为空,不为空可以直接取该数据作为当前队列的头部数据

如果为空,就把 stack_in 的所有数据弹出,然后压入 stack_out ,然后取 stack_out 栈顶数据作为当前队列的头部数据,最后弹出栈顶数据

代码与坑

如果真是这么顺利就ac了我也不会特地写个记录

最开始我是完全按照之前的思路去写的代码,如下:

class CQueue {
public:
    //定义两个栈
    stack<int>stack_in;
    stack<int>stack_out;

    CQueue() {

    }
    
    void appendTail(int value) {
        stack_in.push(value);
    }
    
    int deleteHead() {
        if (stack_in.empty()) return -1;
        if(!stack_out.empty()) {
            return stack_out.top();
        }else{
            while(!stack_in.empty()){
                stack_out.push(stack_in.top());//问题位置
                stack_in.pop();
            }
            int res = stack_out.top();
            return res;
        }        
    }
};

看起来很好,但是第一个用例都过不去

后面经过定位发现问题出现在交换两个栈元素的过程中

如果像上面那样写,可能出现莫名其妙的错误(原因TBD)

所以后面我用一个变量来接收弹出数据,然后再入栈(老老实实的写)

至少不会错了,以下是ac代码

class CQueue {
public:
    //定义两个栈
    stack<int>stack_in;
    stack<int>stack_out;

    CQueue() {

    }
    
    void appendTail(int value) {
        stack_in.push(value);
    }
    
    int deleteHead() {
        int res;//用于保存弹出栈的数据
        if(!stack_out.empty()){//当stack_out不为空,先将该数获取为res,然后弹出
            res = stack_out.top();
            stack_out.pop();
            return res;
        }else if(!stack_in.empty()){//当stack_in不为空,循环将所有数加入stack_out中
            while(!stack_in.empty()){
                res = stack_in.top();
                stack_in.pop();
                stack_out.push(res);
                // stack_out.push(stack_in.top());//这种写法会出错
                // stack_in.pop();      
            }
            // res = stack_out.top();
            stack_out.pop();//两个栈交换完数据后,直接将stack_out的栈顶弹出
            return res;//当前res记录的是最后一个压入stack_out的数据,即stack_out的栈顶数据,返回即可
        }else{//栈为空时,返回-1
            return -1;
        } 
    }
};

以下写法在 232. 用栈实现队列仍使用

stack_out.push(stack_in.top());//这种写法会出错
stack_in.pop(); 

为了保证统一,均修改为本题中的这种写法

为什么会出错?

代码1
int res;
res = stack_in.top();
stack_in.pop();
stack_out.push(res);

代码2
stack_out.push(stack_in.top());
stack_in.pop();  

这两种写法有何区别?为什么代码2在某些情况下会出错?

这两种写法的功能是相同的,都是将 stack_in 的栈顶元素弹出并压入 stack_out 中。但是,它们之间存在一些细微的区别。

主要是顺序不同

代码1先将栈顶元素存储到 res 中,再将 res 压入 stack_out 栈中,而代码2直接将 stack_in 栈顶元素压入 stack_out 栈中。

如果 stack_in 为空栈,那么代码1和代码2都会出错,因为都会试图访问空栈的栈顶元素。

但是,如果 stack_out 栈非空,那么代码2在将 stack_in 栈顶元素压入 stack_out 栈之前,没有先检查 stack_out 栈是否为空,因此可能会导致将元素插入到错误的位置

因此,如果 stack_out 栈非空,最好使用代码1的写法。事实上,如果没超时,都无脑用第一种方式就行。

posted @ 2023-04-05 22:54  dayceng  阅读(23)  评论(0编辑  收藏  举报