UVa Problem Solution: 846 - Steps

Let f(n) denote the furthest distance we can go with n steps, then we can derive this formula easily:
   f(n) = ((n+1)/2)^2 if n is odd, (1, 2, 3, 4, 3, ,2, 1)
   f(n) = (n/2)^2 + n/2 if n is even. (1, 2, 3, 3, 2, 1)
Give the distance of f(n), we can find n by looking around the square root of f(n).

Code:

1. /***************************************************************************
2.  *   Copyright (C) 2008 by Liu Kaipeng                                     *
3.  *   LiuKaipeng at gmail dot com                                           *
4.  ***************************************************************************/
6. /* @JUDGE_ID 00000 846 C++ "Steps" */
8. #include <algorithm>
9. #include <cmath>
10. #include <cstdio>
11. #include <cstring>
12. #include <deque>
13. #include <fstream>
14. #include <iostream>
15. #include <list>
16. #include <map>
17. #include <queue>
18. #include <set>
19. #include <stack>
20. #include <string>
21. #include <vector>
23. using namespace std;
24.      
25. /*
26.  * Let f(n) denote the furthest distance we can go with n steps, then we can
27.  * derive this formula easily:
28.  *   f(n) = ((n+1)/2)^2 if n is odd, (1, 2, 3, 4, 3, ,2, 1) 
29.  *   f(n) = (n/2)^2 + n/2 if n is even. (1, 2, 3, 3, 2, 1)
30.  * Give the distance of f(n), we can find n by looking around the square root 
31.  * of f(n).
32.  */
33. int steps(int dist)
34. {
35.   if (dist == 0) return 0;
36.   int step = sqrt(dist);
37.   if (step * step == dist) step = step * 2 - 1;
38.   else if (step * step + step < dist) step = step * 2 + 1;
39.   else step = step * 2;
40.   return step;
41. }
42.           
43. int main(int argc, char *argv[])
44. {
45. #ifndef ONLINE_JUDGE
46.   freopen((string(argv[0]) + ".in").c_str(), "r", stdin);
47.   freopen((string(argv[0]) + ".out").c_str(), "w", stdout);
48. #endif
50.   int ncases;
51.   cin >> ncases;
52.   for (int x, y; ncases > 0 && cin >> x >> y; --ncases)
53.     cout << steps(y - x) << '/n';
55.   return 0;
56. }
57.