poj 3111 K Best 最大化平均值 二分思想

題目鏈接：

http://poj.org/problem?id=3111

思路：

• 挑戰程序競賽書上講的很好，下面的解釋也基本來源於此書
• 設定條件C(x):=可以選擇使得單位重量的價值不小於x
• 如何判定C(x)是否可行
• 假設選了某個物品的集合是S，那麽單位重量的價值是：\[ \sum\limits_{i \in S} {v_i } /\sum\limits_{i \in S} {w_i } \]
• 因此就變成了判斷是否存在S滿足下面的條件：\[ \sum\limits_{i \in S} {v_i } /\sum\limits_{i \in S} {w_i } \ge x \]
• 把這個不等式變形就得到：\[ \sum\limits_{i \in S} {(v_i - x + w_i )} \ge 0 \]
• 因此可以對\((v_i-x+w_i)\)的值進行排序貪心地進行選取，因此就變成了,滿足條件的前k大項。

• 代碼：

  #include <iostream>
  #include <algorithm>
  #include <stdio.h>
  using namespace std;
  const int maxn = 100005;
  const int INF = 1000006;
  struct node {
  int v,w,id;
  double y;
  bool
   
   operator < (const node& x) const {
      return y>x.y;
  }
  } jew[maxn];
  int n,k;
  bool check(double x) {
  for(int i=1;i<=n;++i) jew[i].y=jew[i].v-jew[i].w*x;
  sort(jew+1,jew+1+n);
  double sum=0;
  for(int i=1;i<=k;++i) sum+=jew[i].y;
  return sum>=0;
  }
  int main() {
  scanf("%d %d",&n,&k);
  for
   
  (int i=1;i<=n;++i) {
      scanf("%d %d",&jew[i].v,&jew[i].w);
      jew[i].id=i;
  }
  double l=0,r=INF;
  for(int i=0;i<100;++i) {
      double mid = (l+r)/2;
      if(check(mid)) l=mid;
      else r=mid;
  }
  for(int i=1;i<=k;++i) {
      if(i==1) printf("%d",jew[i].id);
      else printf(" %d",jew[i].id);
  }
  printf("\n");
  return 0;
  }

poj 3111 K Best 最大化平均值 二分思想