链家网-后台开发工程师笔试题-第三题

序言（废话）

今天做链家网的笔试题， 选择题考的好杂好杂啊， php, java, C++ 几乎是等比重的考察，完全没有侧重，完全没有侧重。 幸亏都是基本语法， 大部分只要用心猜（反正编程语言都差不多了，毕竟咱是学C++的， 什么邪恶的写法没见过~~~~）。

然后是笔试题， 其实笔试题三道都挺简单的， 只不过第三道题会卡时间。

题目描述

小恪要给学校里的人分组， 每个人的编号是连续的正整数（从1开始）。 然后根据某些同学的编号，查询它在哪个组里。

输入描述

第一行输入正整数 n， 代表总共有几数, 然后下一行输入 n 个正整数，表示每一个组有几个成员。
第三行输入正整数m, 代表有 m 次的询问。 然后下一行有 m 个正整数, 表示询问编号为 Mi的成员，在哪一组？

5
2 7 3 4 9
3
1 25 11

题目输出

1
5
3

数据范围

1<= N <= 10^5, 1<= 每个组的成员数量 <= 10^6, 1<= M <= 10^5.

解题代码

方法一： 直接做

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <algorithm>
#include <stdexcept>
using namespace std;

const int maxn = 100000 + 5; 
long long arr[maxn]; 
long long brr[maxn]; 

int main()
{
	int n; 
	cin >> n; 
	arr[0] = 0; 
	for(int i = 1; i <= n; i++)
		cin >> arr[i]; 
	for(int i = 2; i <= n; i++)
		arr[i] = arr[i] + arr[i-1]; 
	
	int m; 
	cin >> m; 
	for(int i = 1; i <= m; i++)
		cin >> brr[i]; 
	for(int i = 1; i <= m; i++)
	{
		for(int j = 1; j <= n; j++)
		{
			if(arr[j] >= brr[i])
			{
				cout << j << endl; 
				break; 
			}
		}
	}		
	
	return 0; 
}

在规定的时间内，通过了 82%样例。 也就是说稍微有些超时~~~

方法二： 用二分查找代替顺序查找

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>
using namespace std;

const int maxn = 100000 + 5; 
long long arr[maxn]; 
long long brr[maxn]; 

int bSearch(long long arr[], int left, int right, long long val)
{
	while(left < right)
	{
		int mid = left + (right - left)/2;
		if(arr[mid] < val)
			left = mid+1; 
		else if(arr[mid] > val)
			right = mid; 
		else
			return mid; 
	} 
	
	return left; 
}

int main()
{
	int n; 
	cin >> n; 
	arr[0] = 0; 
	for(int i = 1; i <= n; i++)
		cin >> arr[i]; 
	for(int i = 2; i <= n; i++)
		arr[i] = arr[i] + arr[i-1]; 
	
	int m; 
	cin >> m; 
	for(int i = 1; i <= m; i++)
		cin >> brr[i]; 
	for(int i = 1; i <= m; i++)
	{
		cout << bSearch(arr, 1, n+1, brr[i]) << endl; 
	}		
	
	return 0; 
}

然而结果仍然是 82%， 好坑啊， 数据好没区分度啊！！！, 本来以为这种程度的优化绝壁就能过了， 于是就没打算写下面的第三种方法，结果。。。。悲剧了！！！！

方法三： 空间换时间， 排序+hash

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <algorithm>
using namespace std;

const int maxn = 100000 + 5; 
long long arr[maxn]; 
long long brr[maxn]; 
long long crr[maxn]; 

int main()
{
	int n; 
	cin >> n; 
	arr[0] = 0; 
	for(int i = 1; i <= n; i++)
		cin >> arr[i]; 
	for(int i = 2; i <= n; i++)
		arr[i] = arr[i] + arr[i-1]; 
	
	int m; 
	cin >> m; 
	for(int i = 1; i <= m; i++)
	{
		cin >> brr[i]; 
		crr[i] = brr[i]; 
	}
		
	unordered_map<long long, long long> mm; 
	sort(crr+1, crr+m+1);
	int pos = 1;  
	int i = 1; 
	while(i <= n)
	{
		if(pos > m)
			break; 
			
		while(arr[i] < crr[pos])
			i++; 
		mm[crr[pos]] = i; 
		pos++; 	
	}		
	
	for(int i = 1; i <= m; i++)
		cout << mm[brr[i]] << endl; 
		
	return 0; 
}

毫无疑问， 这个提交绝不会超时，因为时间复杂度是O(NlogN)啊， 并且没有常数因子， 线段树也只能是这样的复杂度了吖。 但是， 但是， 链家网用的笔试OJ好差啊， 提交一个题目要十分钟 ~ ~ 。


听说由于这个原因， 最后延时半小时 ~ ~ ~， 可是我已经提前交卷了， ~ ~ ~， 感觉好坑啊。 （当然主要还是怪自己没有再等等 ~ ~ ~）。 写篇博客，总结一下教训，不喜勿喷~~。