【公司真题--字节跳动】字节跳动2019春招研发部分编程题汇总

1.万万没想到之聪明的编辑

直接暴力了！

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <unordered_set>
#include <unordered_map>
using namespace std;

int main() {
    int n; cin>>n;
    while(n--){
        string str;
        cin>>str;
        //AAA->AA
        for(int i=2;i<str.length();i++){
            if(str[i]==str[i-1]&&str[i-1]==str[i-2]){
                str.erase(i,1);
                i--;
            }
        }

        //AABB->AAB
        for(int i=3;i<str.length();i++){
            if(str[i]==str[i-1]&&str[i-2]==str[i-3]&&str[i]!=str[i-2]){
                str.erase(i,1);
                i--;
            }
        }
        cout<<str<<endl;
    }

    return 0;
}

2.万万没想到之抓捕孔连顺

解法一：回溯(超时)

我想的是就从集合中选三个数，并且符合一定的条件，那就回溯呗，但其实用回溯没必要，因为如果确定了一个区间，从这个区间取三个数的取法可以直接一个数学公式搞定，而这样回溯求组合数虽然没毛病，但很明显有点小题大做了，最终过了两个样例，其他的超时！

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <unordered_set>
#include <unordered_map>
using namespace std;

int Count=0;
void backtracking(vector<int>& nums,vector<int> path,int startIndex,int d){
    if(path.size()==3){
        Count++;
        return;
    }
    for(int i=startIndex;i<nums.size();i++){
        path.push_back(nums[i]);
        if(nums[i]-path[0]<=d) backtracking(nums,path,i+1,d);
        path.pop_back();
    }
}

int main() {
    //
    int n,d;
    cin>>n>>d;
    vector<int> nums(n);
    for(int i=0;i<n;i++){
        int a; cin>>a;
        nums[i]=a;
    }
    //
    backtracking(nums,vector<int>(0),0,d);
    cout<<Count%99997867<<endl;

    return 0;
}

解法二：确定区间后数学组合问题直接求解

1.首先不能位置重复，那么固定right求组和，因为每次的right都不一样，所以位置一定不会一样， 每次right+1都通过while循环动态维护first的位置，一旦确定，则可以用数学方法确定俩者之间所有的排列组合
2.上述数学思路：举例：（1 2 3 4 5）–6
6的位置即right位置，已确定，而包括first（1）的值有5个，在他们5个中找2个与last排列组合成3个数，C(5,2)=(5!)/((5-2)!2!),由此推导，C(last-first,2)=((last-first)!)/((last-first-2)!*2!)=(last-first)(last-first-1)/2。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

//求n个数任取两个的取法，也即Cn2
long long combinatorialNum_2(long long n) {
    return n*(n-1)/2;
}

int main() {
    int n,d; cin>>n>>d;
    long long count = 0;
    vector<int> nums(n);
    for(int i=0;i<n;i++){
        cin>>nums[i];
    }
    //初始化就是left=0,right=2，因为选三个人小于三个就没结果直接略过
    for(int left=0,right=2;right<n;right++){
        //如果不符合条件left++，直到符合条件咱们再用数学公式算结果
        while(nums[right]-nums[left]>d){
            left++;
        }
        //不断地固定right，一个right对应一些组合最后加起来就好
        //如果left--right这个区间大于三个数了那才可以开始算
        //但算的是这个区间的个数减一取两个的取法，因为right是固定的，咱们是固定了right求组合数的
        if(right-left+1>=3) count+= combinatorialNum_2(right-left);
    }
    cout<<count%99997867<<endl;
    return 0;
}

3.雀魂启动

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

bool isHu(vector<int> nums){//其实是一个dfs的过程
    if (nums.empty())    return true;//递归出口
    //记录与首元素相同的数字的个数，因为数组有序，并且最多出现4次，所以不用遍历所有区间
    int cnt = count(nums.begin(), nums.begin()+4, nums[0]);

    //注意：下面的几个if不能加上else奥，因为每种情况都需要考虑
    //取了雀头之后的size肯定能被3整除（因为除了雀头都是三个一组取的）
    if (nums.size() % 3 != 0 && cnt >= 2){//当前两个元素相等时，尝试以他两作为雀头
        if ( isHu(vector<int>(nums.begin() + 2, nums.end())) )   return true;
    }

    if (cnt >= 3){//尝试取一个刻子（三个相同的数字）
        if ( isHu(vector<int>(nums.begin() + 3, nums.end())) )    return true;
    }

    //首元素以顺子的形式出现
    //先判断首元素+1和+2的元素是否存在，只有同时存在才有可能构成顺子
    if(count(nums.begin(), nums.end(), nums[0]+1) > 0 && count(nums.begin(), nums.end(), nums[0] + 2) > 0){
        //把该顺子从数组中去掉，继续判断剩下的顺子
        vector<int> nums_new(nums.begin() + 1, nums.end());//去掉一个首元素（顺子的第一个）
        nums_new.erase(find(nums_new.begin(), nums_new.end(), nums[0] + 1));//去掉一个首元素+1（顺子的第二个）
        nums_new.erase(find(nums_new.begin(), nums_new.end(), nums[0] + 2));//去掉一个首元素+2（顺子的第三个）
        if (isHu(nums_new)) return true;
    }
    return false;
}

int main(){
    vector<int> nums(13), count(10,  0);//已有的13张牌，以及计数器，记录每张牌出现的次数
    vector<int> res;
    for (int i = 0; i < 13; ++i){
        cin >> nums[i];
        ++count[nums[i]];
    }    
    sort(nums.begin(), nums.end());//先排序
    for (int i = 1; i <= 9; ++i){//遍历下一张可能的牌
        if (count[i] == 4)    continue;//最多就4张，真的不能再多了
        auto it = lower_bound(nums.begin(), nums.end(), i);
        nums.insert(it, i);//插入i

        if(isHu(nums) )     res.push_back(i);   //判断是否ok

        it = lower_bound(nums.begin(), nums.end(), i);
        nums.erase(it);//恢复原始nums数组
    }

    if (res.empty()) cout << 0 << endl;
    else{
        for (const auto &v : res)   cout << v <<" ";
    }
    return 0;
}