字节跳动2018算法工程师校招编程题题解（第二批）

编程题1

题目大意

为了不断优化推荐效果，今日头条每天要存储和处理海量数据。假设有这样一种场景：我们对用户按照它们的注册时间先后来标号，对于一类文章，每个用户都有不同的喜好值，我们会想知道某一段时间内注册的用户（标号相连的一批用户）中，有多少用户对这类文章喜好值为k。因为一些特殊的原因，不会出现一个查询的用户区间完全覆盖另一个查询的用户区间(不存在L1<=L2<=R2<=R1)。

思路

多次区间查询区间内等于x的个数
没有更新，不适用线段树或者树状数组。单纯遍历是O(n)，先排序预处理然后二分查找是O(logn)，但是排序会破坏区间。所以我们考虑一个区间内的数部分遍历部分二分查找，就是分块的思想，两端遍历，中间每个排序好的块二分查找个数。

AC代码

#include<bits/stdc++.h> 
#define ll long long
using namespace std;
vector<int>a[1000];
int b[300005];
int n,k,q;
int getp(int x){
    return x/k;
}
int query(int l,int r,int c){
    int res=0;
    int m=getp(l);
    int p=getp(r);
    for(int i=l;i<(m+1)*k;i++){
        if(b[i]==c)res++;
    }
    for(int i=m+1;i<p;i++){
        res+=upper_bound(a[i].begin(),a[i].end(),c)-lower_bound(a[i].begin(),a[i].end(),c);
    }
    for(int i=p*k;i<=r;i++){
        if(b[i]==c)res++;
    }
    return res;
}
int main(){
    scanf("%d",&n);
    k=sqrt(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&b[i]);
        a[getp(i)].push_back(b[i]);
    }
    for(int i=0;i<=n/k;i++){
        sort(a[i].begin(),a[i].end());
    }
    scanf("%d",&q);
    int l,r,c;
    while(q--){
        scanf("%d%d%d",&l,&r,&c);
        printf("%d\n",query(l,r,c));
         
    }
     
}

编程题2

题目大意

作为一个手串艺人，有金主向你订购了一条包含n个杂色串珠的手串——每个串珠要么无色，要么涂了若干种颜色。为了使手串的色彩看起来不那么单调，金主要求，手串上的任意一种颜色（不包含无色），在任意连续的m个串珠里至多出现一次（注意这里手串是一个环形）。手串上的颜色一共有c种。现在按顺时针序告诉你n个串珠的手串上，每个串珠用所包含的颜色分别有哪些。请你判断该手串上有多少种颜色不符合要求。即询问有多少种颜色在任意连续m个串珠中出现了至少两次。

思路

比第一题简单得多，滚动区间即可

AC代码

#include<bits/stdc++.h> 
#define ll long long
using namespace std;
vector<int>a[100005];
int now[55];
int ans[55];
int n,m,c;
int main(){
    scanf("%d%d%d",&n,&m,&c);
    int x,y;
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        while(x--){
            scanf("%d",&y);
            a[i].push_back(y);
        }
    }
    for(int i=1;i<=m;i++){
        for(int j=0;j<a[i].size();j++){
            now[a[i][j]]++;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=c;j++){
            if(now[j]>1)ans[j]=1;
        }
        for(int j=0;j<a[i].size();j++){
            now[a[i][j]]--;
        }
        if(i+m>n)x=(i+m)%n;
        else x=i+m;
        for(int j=0;j<a[x].size();j++){
            now[a[x][j]]++;
        }
    }
    int res=0;
    for(int i=1;i<=c;i++){
        if(ans[i]==1)res++;
    }
    printf("%d",res);
    return 0;
}

编程题3

题目大意

字符串S由小写字母构成，长度为n。定义一种操作，每次都可以挑选字符串中任意的两个相邻字母进行交换。询问在至多交换m次之后，字符串中最多有多少个连续的位置上的字母相同？

思路

贪心或者dp，我写的是贪心。
首先理解交换字母，当我们以某个字母为目标时，我们尽量将所有字母向中间靠拢可以使得使用交换次数最小，具体是哪个字母我们需要遍历，每种字母的每个位置作为中间聚集点计算一遍答案求最长距离。
在计算时用到贪心思路，为了使得连续字母数最多，肯定是先将距离聚集点最近的点先交换，所以用两个动态指针向两边扩展，比较两边的距离，选短的进行交换，如此循环即可。
复杂度在O(n²)

AC代码

#include<stdio.h> 
#include<string.h>
#include<vector> 
#define ll long long
using namespace std;
char s[10005];
int m,n,k,l,r,num,x,left,right,leftp,rightp,ans;
vector<int>pose[30];
int main(){
	scanf("%s%d",&s,&m);
	n=strlen(s);
	for(int i=0;i<n;i++){
		pose[s[i]-'a'].push_back(i);
	}
	ans=0;
	for(int i=0;i<=25;i++){
		k=pose[i].size();
		for(int j=0;j<k;j++){
			l=pose[i][j],r=pose[i][j],num=1,x=m;
			left=j-1,right=j+1;
			leftp,rightp;
			while(left>=0&&right<k){
				leftp=l-pose[i][left]-1;
				rightp=pose[i][right]-r-1;
				if(leftp>x&&rightp>x)break;
				else if(leftp<rightp){
					x-=leftp;
					left--;
					l--;
					num++;
				}
				else{
					x-=rightp;
					right++;
					r++;
					num++;
				}
			}
			while(left>=0){
				leftp=l-pose[i][left]-1;
				if(leftp>x)break;
				x-=leftp;
				left--;
				l--;
				num++;
			}
			while(right<k){
				rightp=pose[i][right]-r-1;
				if(rightp>x)break;
				x-=rightp;
				right++;
				r++;
				num++;
			}
			ans=ans>num?ans:num;
		}
	}
	printf("%d",ans);
	return 0;
}