字节面试一道常见的题目
最近看大家面经总有一道常见的题目:
   数组A中给定可以使用的1~9的数,返回由A数组中的元素组成的小于n的最大数。例如A={1, 2, 4, 9},n=2533,返回2499
有两个常见思路:
1.暴力递归回溯,最后判断;
2.贪心思想。
其中第二个思路比较有争议,看讨论大家好像也没有很统一的想法,先贴一下自己的代码
#include <iostream>
#include <stack>
#include <string>
#include <vector>
using namespace std;
int n, x;
//将剩余位全部设置为数组里的最大值
void getmax(stack<int>& sres, int value, int count){
    for(int i = 0; i < count; i++){
        sres.push(value);
    }
}
//二分查找,取得小于等于传入值的最大值
int getindex(vector<int>& vec, int value){
    int left = 0, right = vec.size()-1, target = -1;
    while(left <= right){
        int mid = (right-left) / 2 + left;
        if(vec[mid] > value){
            right = mid-1;
        }else if(vec[mid] < value){
            target = mid;
            left = mid + 1;
        } else
            return mid;
    }
    return target;
}
//主函数
int helper(vector<int>& vec, string s){
    stack<int>sres;
    int ssize = s.size();
    for(int i = 0; i < ssize; i++){
        int countnum = s[i] - '0';
        int index = getindex(vec, countnum);
        //如果当前位置的值比数组的所有位置都要小,栈要进行弹出操作,相当于去回溯
        if(index == -1){
            int tmpcount = 0;
            while(!sres.empty()){
                int top = sres.top();
                sres.pop();
                int tmpindex = getindex(vec, top-1);
                if(tmpindex == -1)
                    continue;
                sres.push(vec[tmpindex]);
                tmpcount++;
                break;
            }
            if(sres.empty())
                getmax(sres, vec[n-1], ssize-1);
            else
                getmax(sres, vec[n-1], ssize - sres.size());
            break;
        }else if(vec[index] == countnum){
            //若是相等,就加进去
            sres.push(countnum);
        }else if(vec[index] < countnum){
            //若是二分查找的值于当前值并不相等,那么把二分查找的值加入,并且后面的值全取数组的最大值
            sres.push(vec[index]);
            getmax(sres, vec[n-1],ssize-i-1);
            break;
        }
    }
    int res = 0, count = 1;
    while(!sres.empty()){
        res += sres.top() * count;
        count *= 10;
        sres.pop();
    }
    return res;
}
int main() {
    cin >> n;
    vector<int> vec(n);
    for(int i = 0; i < n; i++)
        cin >> vec[i];
    cin >> x;
    sort(vec.begin(), vec.end());
    //传入的值是x-1
    std::cout << helper(vec, to_string(x-1)) << std::endl;
    return 0;
}
我觉得这个代码是正确的,希望大家可以指正一下, 我自己编造的例子基本都过了。
#字节跳动##面经##笔经##C/C++# 查看25道真题和解析
查看25道真题和解析