百度2016实习 前端试题中的编程题1
输出乘法表中第k小(倒数第k大)的数的值
原始来源: http://codeforces.com/contest/448/problem/D?locale=en (Codeforces Round #256 (Div. 2) - Multiplication Table)
题目大意:
n*m的一个乘法表,选出第k小的数【相等的数字不去重】 (1 ≤ n, m ≤ 5·10^5; 1 ≤ k ≤ n·m)
输入:
2 3 4
输出:
3
样例输入: 2 3 4
相应的乘法表为:
1 2 3
2 3 4
非减序列是:1, 2, 2, 3, 3, 4。
第4个数字是3,所以输出3。
分析:
看到题目,注意到横向递增,纵向也递增,可以想到两种办法:
1. 暴力法(使用STL算法库中的sort函数);
2. 二分搜索法,由于每行分别是非递减排序的,故可以考虑对每行分别使用此法,总的时间复杂度为O(n*2*log n).
据说有第3种方法,从n*m开始搜索,后来发现状态实在太多,时间复杂度是O(n*m)。
1. 暴力法 代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
int m, n, k;
vector<int> vec(1,0);
vector<int>::iterator iter = vec.begin();
while(cin>>m>>n>>k && m>=1 && n>=1 && k>=1 && k<=m*n)
{
for(int i=1;i<=m;i++)
{
for(int j=1;j<=n;j++)
{
vec.push_back(i*j);
}
}
sort(vec.begin(),vec.end());
cout<<*(vec.begin()+k);
vec.erase(vec.begin()+1,vec.end()); // 清除vec.at(0)之后的数,从下标为1的地方开始存放乘法表中的数
}
return 0;
}
二分搜索法:
1. 假设第k小的数的值为kval,在"排序"后的新位置为kth,当然排序并没有真的进行,1<= kval <= m*n,题意即为输出kval;
2. 取m*n与1的中间值,记为midVal,统计当前已扫描的范围中比midVal小的数的个数sum(对每行小于midVal的数的个数分别进行统计,然后求和),将sum与kth比较,当sum<kth时,二分区间的左值leftVal增加1;当sum>=kth时,二分区间的右值rightVal减少1;
3. 递归的进行步骤2,直到while循环结束,此时leftVal即为所求的值。
AC代码:
#include <iostream>
using namespace std;
typedef long long int64; // int64(LL): 64位整型数, 即long long的别称
int64 m,n,k;
int64 cal(int64 x) // 计算比x小的数的个数
{
int64 sum=0;
for(int i=1;i<=n;i++)
{
sum += (m <= x/i)?m:(x/i); // min(m,x/i) 是每一行中比x小的数的个数(若i*m<=x,则会有m个数满足要求,否则会有x/i个数满足),也可以用if,else的写法 或 ? : 表达式的写法.
}
return sum;
}
int64 binarySearch(int64 leftVal, int64 rightVal, int64 kth)
{
int64 midVal;
while(leftVal <= rightVal)
{
midVal=(leftVal+rightVal)/2;
if(cal(midVal) < kth) leftVal=midVal+1; // 要找的数在后面,区间下限继续增大
else rightVal=midVal-1; // 要找的数在前面,区间上限继续减小
}
return leftVal;
}
int main()
{
while(cin>>m>>n>>k && m>=1 && n>=1 && k>=1 && k<=m*n)
{
cout<<binarySearch(1,n*m,k);
cout<<endl;
}
return 0;
}
二分查找法的实现和应用汇总 - Ider - 博客园 http://www.cnblogs.com/ider/archive/2012/04/01/binary_search.html
其中提到很重要的一点,最终需要应用二分搜索的部分实质上都需转化为数组来解决,即使用链表,链表的node中也需要有数组元素作为成员...