编程之美第一篇 01分数规划

01分数规划

01分数规划问题其实就是解决单价之类的问题,假设给你n个物品,让你找出选k个物品的最大单价;例如南阳oj:Yougth的最大化;解决这类问题可以用二分查找,这类问题跟二分极大化最小值,极小化最大值有一些相似的地方,均是从结果出发,来进行二分查找;例如上面南阳那道题,可以转化一下;

由于v/w=单价;所以v=w*单价;即v-w*单价=0;有了这个关系,我们马上可以想到二分来查找这个值;

那么我们可以定义一个count数组来记录v-w*单价的值;由于选k个只需要把count从大到小排下序就可以了;然后就是二分了;这类问题就是01分数规划问题;

代码实现:

#include<stdio.h>
#include<algorithm>
#define MAX(x,y) x>y?x:y
using namespace std;
const int MAXN=10010;
struct Node{int v,w;
};
Node res[MAXN];
double cont[MAXN];
int n,k;
int fun(double mid){
for(int i=0;i<n;i++){ cont[i]=res[i].v-mid*res[i].w; } sort(cont,cont+n);double sum=0; for(int i=n-1;i>=n-k;i--)sum+=cont[i];
//printf("%lf\n",mid); return sum>=0?1:0; } double abs(double x){ return x>0?x:-x; } double search(double max){
double left=0,right=max,mid; while(right-left>1e-10){mid=(left+right)/2; if(fun(mid))left=mid; else right=mid; } return mid; } int main(){ while(~scanf("%d%d",&n,&k)){double max=0; for(int i=0;i<n;i++){ scanf("%d%d",&res[i].w,&res[i].v); max=MAX(max,res[i].v*1.0/res[i].w);//
} printf("%.2f\n",search(max)); } return 0; }

  与这个题相似的还有poj Dropping tests;这个题意是:

给你n个数,让求删除k个数后

的最大值;

跟南阳那个题很相似,只需要找n-k个数即可;

代码实现:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int MAXN=10010;
struct Node{
    int a,b;
};
Node dt[MAXN];
double d[MAXN];
int n,k;
bool fsgh(double R){
    double sum=0;
    for(int i=0;i<n;i++)d[i]=dt[i].a-R*dt[i].b;
    sort(d,d+n);
    for(int i=n-1;i>=n-k;i--)sum+=d[i];
    return sum>0?true:false;
}
double erfen(double l,double r){
    double mid;
    while(r-l>1e-6){
        mid=(l+r)/2;
        if(fsgh(mid))l=mid;
        else r=mid;
    }
    return mid;
}
int main(){
    while(scanf("%d%d",&n,&k),n|k){
        double mx=0;
        k=n-k;
        for(int i=0;i<n;i++)scanf("%d",&dt[i].a);
        for(int i=0;i<n;i++)scanf("%d",&dt[i].b),mx=max(1.0*dt[i].a/dt[i].b,mx);
        printf("%.0f\n",erfen(0,mx)*100);
    }
    return 0;
}

  另外01分数规划还可以与图论结合在一起;例如:

和最小生成树结合在一起让你求一棵最优比率生成树;例如 poj Desert King;

题意:有N个村庄,给出每个村庄的坐标和海拔,benifit为两点之间的距离,cost为两点的高度差,现在要求一棵树使得 cost / benift 最小

咋一看跟01分数规划还真像,但是让求的是一棵树,我们该怎么办?其实就是求一个最小生成树罢了,最小生成树里面的low数组代表的是啥?权值!那直接把low数组里面的值换成cost-benift*R就好了,然后就是一个二分问题了;

代码实现:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#define mem(x,y) memset(x,y,sizeof(x))
using namespace std;
const int INF=0x3f3f3f3f;
typedef long long LL;
const int MAXN=1010;
double vis[MAXN],low[MAXN];
int N;
double R;
struct Node{
    double x,y,h;
};
Node dt[MAXN];
double len[MAXN][MAXN],cost[MAXN][MAXN];
double getl(Node a,Node b){
    double x=b.x-a.x,y=b.y-a.y;
    return sqrt(x*x+y*y);
}

bool prime(){
    double total;
    mem(vis,0);
    for(int i=0;i<N;i++)low[i]=cost[0][i]-R*len[0][i];
    total=0;
    vis[0]=1;//
for(int i=0;i<N;i++){ double temp=INF; int k; for(int j=0;j<N;j++) if(!vis[j]&&low[j]<temp)temp=low[j],k=j; if(temp==INF)break; total+=temp; vis[k]=1; for(int j=0;j<N;j++) if(!vis[j]&&low[j]>cost[k][j]-R*len[k][j])low[j]=cost[k][j]-R*len[k][j]; } //printf("total=%lf R=%lf\n",total,R); if(total>0)return true; else return false; } int main(){ while(scanf("%d",&N),N){ mem(len,INF); mem(cost,INF); double mxl=-INF,mil=INF,mxc=-INF,mic=INF; for(int i=0;i<N;i++) scanf("%lf%lf%lf",&dt[i].x,&dt[i].y,&dt[i].h); for(int i=0;i<N;i++){ for(int j=i+1;j<N;j++){ len[j][i]=len[i][j]=getl(dt[i],dt[j]); cost[j][i]=cost[i][j]=abs(dt[i].h-dt[j].h); mxl=max(mxl,len[i][j]); mxc=max(mxc,cost[i][j]); mil=min(mil,len[i][j]); mic=min(mic,cost[i][j]); } } //printf("%lf %lf %lf %lf\n",mil,mic,mxl,mxc); double l=mic/mxl,r=mxc/mil; // printf("%lf %lf\n",l,r); while(r-l>1e-4){ R=(l+r)/2; if(prime())l=R; else r=R; } printf("%.3f\n",l); } return 0; }

另外01分数规划还可以与最短路结合在一起;求一个最优比率环;例如poj Sightseeing Cows;

题意:这个人带牛旅行,旅行每个城市会有幸福度,通过每个城市会花费时间,让找平均每秒的最大幸福度;

注意这题是让求最大的,好像我们前面讲的都是最小的,那该怎么办呐?很简单以前是hp-R*t;转成R*t-hp不就好了吗?照样转化成最短路问题;但是可能产生负边啊;对了就是负边,二分就是从负边出发的,有负边证明t太大,没有t小,这就是二分的一个条件;由于负边我们可以用bellman,或者邻接表解决;

代码实现:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define mem(x,y) memset(x,y,sizeof(x))
using namespace std;
const int INF=10000000000000;
typedef long long LL;
const int MAXN=1010;
const int MAXM=100010;
/*struct Node{
    int u,v;
    double t;
}; 
Node dt[MAXM];*/
struct Edge{
    int from,to,next,t;
};
Edge edg[MAXM];
int head[MAXM];
int edgnum;
void add(int u,int v,int t){
    Edge E={u,v,head[u],t};
    edg[edgnum]=E;
    head[u]=edgnum++;
}
int L,P;
double hp[MAXN],dis[MAXN];
int usd[MAXN],vis[MAXN];
/*void add(int u,int v,double t){
    Node E={u,v,t};
    dt[edgnum++]=E;
}*/
//double R;
/*bool Bellman(){
    mem(dis,INF);
    mem(usd,0);
    dis[1]=0;
    while(1){
        int temp=0;
        for(int j=0;j<edgnum;j++){
            int u=dt[j].u,v=dt[j].v;
            double t=dt[j].t;
            //dis[v]=min(dis[v],dis[u]+R*t-hp[u]);//应该是R*t-hp[u]; 
            if(dis[v]>dis[u]+R*t-hp[u])usd[v]++,dis[v]=dis[u]+R*t-hp[u],temp=1;
            if(usd[v]>L)return false;
        }
        if(!temp)return true;
    }
}*/
bool SPFA(double R){
    queue<int>dl;
    while(!dl.empty())dl.pop();
    for(int i=1;i<=L;i++){
        dis[i]=INF;
        vis[i]=0;
        usd[i]=0;
    }
    dl.push(1);
    vis[1]=1;
    usd[1]++;
    dis[1]=0;
    while(!dl.empty()){
        int u=dl.front();
        dl.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=edg[i].next){
            int v=edg[i].to,t=edg[i].t;
            if(dis[v]>dis[u]+R*t-hp[u]){
                dis[v]=dis[u]+R*t-hp[u];
                if(!vis[v]){
                    vis[v]=1;
                    usd[v]++;
                    dl.push(v);
                //  printf("%d\n",usd[v]);
                    if(usd[v]>=L)return false;
                }
            }
        }
    }
    return true;
}
int main(){
    int a,b;
    int c;
    while(~scanf("%d%d",&L,&P)){
        edgnum=0;
        double mih=INF,mxh=-INF;
        int mit=INF,mxt=-INF;
        mem(head,-1);
        for(int i=1;i<=L;i++){
            scanf("%lf",hp+i);
            mih=min(mih,hp[i]);
            mxh=max(mxh,hp[i]);
        }
        while(P--){
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            mit=min(mit,c);
            mxt=max(mxt,c);
        }
        double l=mih/mxt,r=mxh/mit;
    //  printf("%f %f\n",l,r);
        double R;
        while(r-l>=0.001){
            R=(l+r)/2;
            if(SPFA(R))r=R;
            else l=R;
        }
        printf("%.2f\n",l);
    }
    return 0;
}

01分数规划是一个基本而且有用的算法,可以与图论连用,也可以与数据结构一起用;

上面提到的题解详情请见:

http://www.cnblogs.com/handsomecui/p/4690691.html

http://www.cnblogs.com/handsomecui/p/4971467.html

http://www.cnblogs.com/handsomecui/p/4972701.html

http://www.cnblogs.com/handsomecui/p/4973041.html

转载请附上地址:

http://www.cnblogs.com/handsomecui

posted @ 2016-01-09 17:46  handsomecui  阅读(2289)  评论(0编辑  收藏  举报