UVA1456|LA4731------Cellular Network------带贪心的概率型DP

本文出自：http://blog.csdn.net/dr5459

题目地址：http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4202

题目意思：

手机在蜂窝网络中的定位是一个基本问题。假设蜂窝网络已经得知手机处于c1, c2,…,cn这些区域中的一个，最简单的方法是同时在这些区域中寻找手机。但这样做很浪费带宽。由于蜂窝网络中可以得知手机在这不同区域中的概率，因此一个折中的方法就是把这些区域分成w组，然后依次访问。比如，已知手机可能位于5个区域中，概率分别为0.3、0.05、0.1、0.3和0.25，w=2，则一种方法是先同时访问{c1,c2,c3}，再同时访问{c4,c5}，访问区域数的数学期望为3*(0.3+0.05+0.1)+(3+2)*(0.3+0.25)=4.1。另一种方法是先同时访问{c1,c4}，再访问{c2,c3,c5}，访问区域数的数学期望为2×(0.3+0.3)+(3+2)×(0.05+0.1+0.25)=3.2。要你求出最小的期望

解题思路：

首先是贪心的思路

从题目可以看出，要使期望最小，应该是概率大的放在前面

所以要按概率从大到小排个序

然后就是划分的问题了

我们定义f[i][j]为前i个分成j组的情况，注意i>=j

然后状态转移就是：f[i][j] = min(f[k-1][j-1]+i*(sum[i]-sum[k-1])) 注意到这里k>=j && k<=i

在写的时候初始化一下f[i][1]就好了

后面就是傻逼式的递推了

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn = 110;

double dp[maxn][maxn];
double sum[maxn];
int u[maxn];
double p[maxn];
int n,w;

bool cmp(int a,int b)
{
    return a>b;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&w);
        int tot = 0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&u[i]);
            tot+=u[i];
        }

        sort(u+1,u+1+n,cmp);

        sum[0] = 0;
        for(int i=1;i<=n;i++)
        {
            p[i] = u[i]*1.0/tot;
            sum[i] = sum[i-1]+p[i];
            //printf("i %d ")
        }

        for(int i=1;i<=n;i++)
        {
            dp[i][1] = i*sum[i];
        }


        double INF = 0x3f3f3f3f;
        for(int i=2;i<=n;i++)
        {
            for(int j=2;j<=w;j++)
            {
                dp[i][j] = INF;
                for(int k=j;k<=i;k++)
                {
                    dp[i][j] = min(dp[i][j],dp[k-1][j-1]+i*(sum[i]-sum[k-1]));
                }
            }
        }

        printf("%.4f\n",dp[n][w]);

    }
    return 0;
}