Careercup - Microsoft面试题 - 5943729928011776

2014-05-10 21:56

题目链接

原题：

Suppose you get number of unique users every second from bing
For eg, ,,,,,etc
You need to write a web service method , such that it takes the input n, which return lowest n unique number from the list of unique numbers. For eg, if n is  then you need to return ,,

题目：从n个数里找出m个最小的数。

解法1：题目虽然没有要求结果是否有序，也没有说算法是否是在线的。那么肯定是有两种解法的。第一种就是O(n * log(m))的最大堆法。能提供有序的结果。如果要求在线算法的话，堆仍能满足要求。

代码：

 // http://www.careercup.com/question?id=5943729928011776
 #include <iostream>
 #include <queue>
 #include <vector>
 using namespace std;

 template <class T>
 struct myless {
     bool operator () (const T &x, const T &y) {
         return x < y;
     };
 };

 int main()
 {
     int val;
     int n, k;
     int i;
     // max heap
     priority_queue<int, vector<int>, myless<int> > q;
     vector<int> v;

     while (cin >> n >> k && (n >  && k > )) {
         k = k < n ? k : n;
         for (i = ; i < k; ++i) {
             cin >> val;
             q.push(val);
         }

         for (i = k; i < n; ++i) {
             cin >> val;
             if (q.top() > val) {
                 q.pop();
                 q.push(val);
             }
         }
         while (!q.empty()) {
             v.push_back(q.top());
             q.pop();
         }
         reverse(v.begin(), v.end());

         cout << '{';
         for (i = ; i < k; ++i) {
             i ? (cout << ' '),  : ;
             cout << v[i];
         }
         cout << '}' << endl;

         v.clear();
     }

     return ;
 }

解法2：快速选择算法可以在O(n)时间找出第k大的数，不过写法复杂，并且运行的实际效率由于常系数很大，递归不稳定等缺点，使得这算法很多时候只是看上去很美。利用这个算法得到第m大的数之后，再扫描一次整个数组就能得到最小的m个数了。这个算法不能在线进行，因为快速选择算法的思想源自快速排序，是不稳定的内存排序算法。

代码：

 // http://www.careercup.com/question?id=5943729928011776
 #include <algorithm>
 #include <iostream>
 #include <vector>
 using namespace std;

 const int CUT_OFF = ;

 int medianThree(vector<int> &v, int ll, int rr)
 {
     int mm = (ll + rr) / ;

     if (v[ll] > v[mm]) {
         swap(v[ll], v[mm]);
     }
     if (v[ll] > v[rr]) {
         swap(v[ll], v[rr]);
     }
     if (v[mm] > v[rr]) {
         swap(v[mm], v[rr]);
     }
     swap(v[mm], v[rr - ]);
     return v[rr - ];
 }

 void quickSelect(vector<int> &v, int ll, int rr, int k)
 {
     // reference from "Data Structure and Algorithm Analysis in C" by Mark Allen Weiss.
     int pivot;
     int i, j;

     if (ll + CUT_OFF <=    rr) {
         pivot = medianThree(v, ll, rr);
         i = ll;
         j = rr - ;

         while (true) {
             while (v[++i] < pivot);
             while (v[--j] > pivot);
             if (i > j) {
                 break;
             }
             swap(v[i], v[j]);
         }
         swap(v[i], v[rr - ]);

         if (k < i) {
             return quickSelect(v, ll, i - , k);
         } else if (k > i) {
             return quickSelect(v, i + , rr, k);
         }
     } else {
         for (i = ll; i <= rr; ++i) {
             for (j = i + ; j <= rr; ++j) {
                 if (v[i] > v[j]) {
                     swap(v[i], v[j]);
                 }
             }
         }
     }
 }

 int main()
 {
     vector<int> v;
     vector<int> res;
     int n, k;
     int i;
     int k_small, count;

     while (cin >> n >> k && (n >  && k > )) {
         v.resize(n);
         for (i = ; i < n; ++i) {
             cin >> v[i];
         }

         // find the kth smallest number
         // this will change the order of elements
         quickSelect(v, , n - , k - );
         k_small = v[k - ];
         count = k;
         for (i = ; i < n; ++i) {
             if (v[i] < k_small) {
                 --count;
             }
         }
         for (i = ; i < n; ++i) {
             if (v[i] < k_small) {
                 res.push_back(v[i]);
             } else if (v[i] == k_small && count > ) {
                 res.push_back(v[i]);
                 --count;
             }
         }

         cout << '{';
         for (i = ; i < k; ++i) {
             i ? (cout << ' '),  : ;
             cout << res[i];
         }
         cout << '}' << endl;

         v.clear();
         res.clear();
     }

     return ;
 }