UVa Problem Solution: 10041 - Vito's Family

Vito should live on the street which is the median of his relatives.

Code:

1. /*************************************************************************
2.  * Copyright (C) 2008 by liukaipeng                                      *
3.  * liukaipeng at gmail dot com                                           *
4.  *************************************************************************/
6. /* @JUDGE_ID 00000 10041 C++ "Vito's Family" */
8. #include <algorithm>
9. #include <deque>
10. #include <fstream>
11. #include <iostream>
12. #include <list>
13. #include <map>
14. #include <queue>
15. #include <set>
16. #include <stack>
17. #include <string>
18. #include <vector>
20. using namespace std;
22. int min_dist_sum(int *first, int *last)
23. {
24.   int *median = first + (last - first) / 2;
25.   nth_element(first, median, last);
26.   int sum = 0;
27.   for (; first < median; ++first)
28.     sum += *median - *first;
29.   for (; first < last; ++first)
30.     sum += *first - *median;
31.   return sum;
32. }
33.           
34. int main(int argc, char *argv[])
35. {
36. #ifndef ONLINE_JUDGE
37.   filebuf in, out;
38.   cin.rdbuf(in.open((string(argv[0]) + ".in").c_str(), ios_base::in));
39.   cout.rdbuf(out.open((string(argv[0]) + ".out").c_str(), ios_base::out));
40. #endif
42.   int ncases;
43.   cin >> ncases;
44.   while (ncases-- > 0) {
45.     int r;
46.     cin >> r;
47.     int s[500];
48.     for (int i = 0; i < r; ++i)
49.       cin >> s[i];
50.     cout << min_dist_sum(s, s + r) << '/n';
51.   }
53.   return 0;
54. }
55.