noip2016提高组换教室（数学期望+Floyd）

最新推荐文章于 2022-01-14 21:50:32 发布

细雨欣然

最新推荐文章于 2022-01-14 21:50:32 发布

阅读量354

点赞数 1

分类专栏：数学期望图论

本文链接：https://blog.csdn.net/qq_35546274/article/details/62870560

版权

图论同时被 2 个专栏收录

32 篇文章 0 订阅

订阅专栏

数学期望

5 篇文章 0 订阅

订阅专栏

这篇博客回顾了NOIP2016提高组的一道题目，主要涉及数学期望的计算和Floyd算法的应用。作者提到，虽然在考试期间未学习期望，但后来发现这个问题其实并不复杂。文章详细解释了如何利用期望设f(i,j,0)和f(i,j,1)来解决问题，并指出第i-1次申请与第i次申请的无关性，从而得出转移方程。" 112616414,10535097,深入解析K8s Pod：定义、静态Pod与共享Volume,"['Kubernetes', '容器技术', 'Pod管理', '存储卷', '集群管理']

摘要由CSDN通过智能技术生成

终于还是面对这道题了，考试的时候没学期望，现在看来其实挺简单的。当时跑了个暴力，然后不知道加了些什么优化，居然跑了80分出来。

Floyd部分就不说了，只是算一下每2个教室间的距离而已，重点在期望。

期望我们设f(i,j,0)为前i个教室申请了j次第i个教室没申请的情况的数学期望。
相对的f(i,j,1)为前i个教室申请了j次第i个教室申请了的情况的数学期望。

下面就是转移了，我们可以证明到前面的情况出来第i-1次是否申请都与i无关，所以转移方程就出来了。

详细的解释我在代码里注释了，代码如下：

#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=2005;

int a[maxn][2],n,m,v,e;
double d[305][305],c[maxn];
double f[maxn][maxn][2];
double inf;

void dp()
{
    for(int i=1;i<=n;i++)
    for(int j=0;j<=m;j++)
    f[i][j][0]=f[i][j][1]=inf;
    f[0][0][0]=0;
    double x,y,ans=inf;
    for(int i=1;i<=n;i++)
    for(int j=0;j<=m&&j<=i;j++)//x表示上一次没申请的情况，y表示申请了的情况。（主要前面的if）
    {
        x=inf,y=inf;
        x=f[i-1][j][0]+d[a[i-1][0]][a[i][0]];
        if(j) y=f[i-1][j][1]+c[i-1]*d[a[i-1][1]][a[i][0]]+(1-c[i-1])*d[a[i-1][0]][a[i][0]];//上一次申请了所以有失败和成功2种
        f[i][j][0]=min(x,y);
        x=inf,y=inf;
        if(j) x=f[i-1][j-1][0]+c[i]*d[a[i-1][0]][a[i][1]]+(1-c[i])*d[a[i-1][0]][a[i][0]];
        if(j>1) y=f[i-1][j-1][1]+(c[i]*d[a[i-1][0]][a[i][1]]+(1-c[i])*d[a[i-1][0]][a[i][0]])*(1-c[i-1])+(c[i]*d[a[i-1][1]][a[i][1]]+(1-c[i])*d[a[i-1][1]][a[i][0]])*c[i-1];
//由于2次都申请了，所以一共4种情况，不要漏了
        if(j) f[i][j][1]=min(x,y);
        if(i==n) ans=min(ans,min(f[i][j][1],f[i][j][0]));
    }
    printf("%.2lf\n",ans);
}
void init()
{
    scanf("%d%d%d%d",&n,&m,&v,&e);
    for(int i=1;i<=n;i++) scanf("%d",&a[i][0]);
    for(int i=1;i<=n;i++) scanf("%d",&a[i][1]);
    for(int i=1;i<=n;i++) scanf("%lf",&c[i]);
    inf=1000.0*100000.0;
    for(int i=1;i<=v;i++)
    for(int j=1;j<=v;j++)
    d[i][j]=inf;
    for(int i=1;i<=e;i++)
    {
        int x,y;
        double z;
        scanf("%d%d%lf",&x,&y,&z);
        d[x][y]=min(d[x][y],z);
        d[y][x]=min(d[y][x],z);
    }
    for(int i=1;i<=v;i++)
    d[i][i]=0,d[0][i]=0;
    a[0][1]=a[0][0]=0;
    for(int z=1;z<=v;z++)//Floyd
    for(int x=1;x<=v;x++)if(d[x][z]!=inf)
    for(int y=1;y<=v;y++)if(d[z][y]!=inf)
    d[x][y]=min(d[x][y],d[x][z]+d[z][y]);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("classroom.out","w",stdout);
    init();
    dp();
    return 0;
}