喷泉问题

10 2
3
5 3
3 7
9 4

1. /*
2.  * 喷泉问题
3.  * tanliboy Nov.25 2008
4.  */
5. #include<iostream>
6. #include<vector>
7. #include<cmath>
8. using namespace std;
10. template<class T,class O>
11. int partition(T a,int from,int to,O order)
12. {
13.     int cur = from;
14.     for(int i = from;i<to-1;i++)
15.     {
16.         if(a[i] < a[to-1])
17.         {
18.             swap(a[i],a[cur]);
19.             swap(order[i],order[cur]);
20.             cur++;
21.         }
22.     }
24.     swap(a[cur],a[to-1]);
25.     swap(order[cur],order[to-1]);
26.     return cur;
27. }
29. template<class T,class O>
30. void quickSort(T a,int from,int to,O order)
31. {
32.     if(from>= to)
33.     {
34.         return;
35.     }
37.     int p = partition(a,from,to,order);
38.     quickSort(a,from,p,order);
39.     quickSort(a,p+1,to,order);
40. }
42. int main()
43. {
44.     double l,w;
45.     cin>>l>>w;
47.     int n;
48.     cin>>n;
50.     vector<double> a(n);
51.     vector<double> b(n);
53.     for(int i =0;i<n;i++)
54.     {
55.         double x,r;
56.         cin>>x>>r;
58.         double d=0;
59.         if(r>w/2)
60.         {
61.             d = sqrt(r*r - w*w/4);
62.         }
64.         a[i] = x - d;
65.         b[i] = x + d;
66.     }
68.     vector<int> order(n);
69.     for(int i = 0;i<n;i++)
70.     {
71.         order[i] = i;
72.     }
73.     quickSort(a.begin(),0,n,order.begin());
75.     //calculate
76.     double pos = 0;
77.     for(int i = 0;i<n;)
78.     {
79.         int maxIndex = -1;
80.         double max = 0;
81.         while(a[i]<pos && i<n)
82.         {
83.             double r = b[order[i]];
84.             if(r > max)
85.             {
86.                 max = r;
87.                 maxIndex = i;
88.             }
89.             i++;
90.         }
91.         if(maxIndex == -1)
92.         {
93.             //error: can't cover
94.             cerr<<"error"<<endl;
95.             return -1;
96.         }else
97.         {
98.             cout<<order[maxIndex]<<endl;
99.             pos = max;
100.             if(max >= l)
101.             {
102.                 return 0;
103.             }
104.         }
105.     }
107.     return 0;
108. }