poj Dropping tests 01分数规划---Dinkelbach算法

分类： Algorithm-动态规划 2013-07-23 23:57  345人阅读  评论(0)  收藏  举报

01分数规划 Dinkelbach算法

果然比二分要快将近一倍。63MS。二分94MS。

[cpp]  view plain copy

1. #include <iostream>  
2. #include <algorithm>   
3. #include <cstdio>  
4. #include <ctime>  
5. #include <cstdlib>  
6. #include <cmath>  
7. using namespace std;  
8. const int maxn=1005;  
9. double a[maxn],b[maxn];  
10. const double eps=1e-6;  
11. int n,k;  
12. struct dnode{  
13.     double x;  
14.     int index;  
15. }d[maxn];  
16. bool cmp(dnode a,dnode b){  
17.     return a.x>b.x;  
18. }  
19. int main()  
20. {  
21.     while(scanf("%d%d", &n, &k) != EOF)  
22.     {  
23.         if(n+k==0)break;  
24.         for(int i=0;i<n;i++)  
25.              scanf("%lf", &a[i]);  
26.         for(int i=0;i<n;i++)  
27.             scanf("%lf", &b[i]);   
28.         double l=0.5,ans;  
29.         do{  
30.             ans=l;  
31.             for(int i=0;i<n;i++)  
32.                 {d[i].x=a[i]-ans*b[i];  
33.                  d[i].index=i;  
34.                 }  
35.             sort(d,d+n,cmp);  
36.             double p=0,q=0;  
37.             for(int i=0;i<n-k;i++)  
38.             {  
39.                 p+=a[d[i].index];  
40.                 q+=b[d[i].index];  
41.             }     
42.             l=p/q;  
43.         }while(fabs(ans-l)>=eps);  
44.           
45.         printf("%.0f\n", l * 100);    
46.       
47.     }  
48.     return 0;  
49. }