hdu 4418 Time travel

         题意: 开始从x点出发,方向为d,走到数轴的一端会原路折返,总共有n个点,你要去的点为y,输入p[1]-p[m]表示你走了i个点停下来的概率,如果停下来的不是要达到的点就重新开始计算,求走到y号点的期望。

       (下面是一种略有问题的思路)

        我们设dp[i][0/1]表示从i出发,在向左/右的时候到达终点的期望,显然dp[y][0]=dp[y][1]=0。

        转移也是传统的期望的转移,为了方便写拿dp[i]来表示,dp[i]=sigma(dp[i+k]+k)*p[k],这个显然是分左右分情况讨论的。由于涉及到很多变量,我们并不到找到一个合理的枚举顺序,那么我们把方程打开,就有了 

          -(1*p[1]+2*p[2]+...)=(-dp[i]+dp[i+1]*p[1]+dp[i+2]*p[2]+...)

        显然是一个可以高斯消元的式子。但是方程一个元素是自由元或者一个元素并不会直接导致所有元素无解,在解方程的时候我们对自由元和无解元素进行特判一下就好了。

        当然题解也是基于我这个dp的方法,不过是把数轴看成了1,2,3,4,3',2',1这样的走动,这样非常好处理出来可行的状态,高斯消元的时候也可以减少元素来降低复杂度,所以我就T了。

         下附TLE代码。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#pragma GCC optimize("Ofast")
#define maxn 205
using namespace std;
const double eps=1e-8;
int read()  
{  
     char c;int sum=0,f=1;c=getchar();  
     while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}  
     while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}  
     return sum*f;  
}
int n,m,y,x,d;
double p[maxn];
double a[maxn][maxn]; // 0向左 1 向右
int getpos(int i,int f)
{
    return f*n+i;
}
double res=0.0;
void init()
{
    res=0.0;
    for(int i=1;i<=m;i++)
        res+=1.0*i*p[i];
}
void solve(int now)
{
    a[now][now]=-1.0;
//    double res=0.0;
//    for(int i=1;i<=m;i++)
//        res+=1.0*i*p[i];

    a[now][n+n+1]=-res;

    if(now<=n)
    {
        int f=-1,temp=now;
        for(int i=1;i<=m;i++)
        {
            if((temp==n && f==1 )|| (temp==1 && f==-1))
                f=-f;
            temp+=f;
            if(f==1)
            {
                a[now][temp+n]+=p[i];
            }
            else
            {
                a[now][temp]+=p[i];
            }
        }
    }
    else
    {
        int f=1,temp=now-n;
        for(int i=1;i<=m;i++)
        {
            if((temp==n && f==1 ) || (temp==1 && f==-1))
                f=-f;
            temp+=f;
            if(f==1)
            {
                a[now][temp+n]+=p[i];
            }
            else
            {
                a[now][temp]+=p[i];
            }
        }
    }
}
double myabs(double now)
{
    return now>0.0 ? now : -now;
}
int gauss()
{
    for(int i=1;i<=n;i++)
    {
        int row=i;
        for(int j=i+1;j<=n;j++)
            if(myabs(a[j][i])>myabs(a[row][i]))
                row=j;
        if(row!=i)
        {
            for(int j=i+1;j<=n+1;j++)
                swap(a[row][j],a[i][j]);
        }
        if(myabs(a[i][i])<eps) continue;
        for(int j=i+1;j<=n;j++)
        {
            double temp=a[j][i]/a[i][i];
            for(int k=i+1;k<=n+1;k++)
            {
                a[j][k]-=temp*a[i][k];
            }
        }
    }
    for(int i=n;i>=1;i--)
    {
        if(myabs(a[i][i])<eps) continue;
        double temp=a[i][n+1]=a[i][n+1]/a[i][i];
        for(int j=i-1;j>=1;j--)
        {
            a[j][n+1]-=a[j][i]*temp;
        }
    }
    return true;
}
int main()
{
    int _;
    //scanf("%d",&_);
    _=read();
    while(_--)
    {
        //memset(a,0,sizeof(a));
        //scanf("%d%d%d%d%d",&n,&m,&y,&x,&d);
        n=read(); m=read(); y=read(); x=read(); d=read();
        for(int i=0;i<=n+n+1;i++)
            for(int j=0;j<=n+n+1;j++)
                a[i][j]=0;
        if(n==1)
        {
            printf("0\n");
            continue;
        }
        for(int i=1;i<=m;i++)
            p[i]=read(),p[i]/=100.0;
        y++; x++;
        init();
//        cerr<<res<<endl;
        for(int i=1;i<=n+n;i++)
        {
            if(i==y || i==y+n)
            {
                a[i][i]=1.0;
                a[i][n+n+1]=0.0;
            }
            else
            {
                solve(i);
            }
        }
        n=n+n;
//        for(int i=1;i<=n;i++)
//        {
//            for(int j=1;j<=n+1;j++)
//                printf("%3.2lf ",a[i][j]);//cout<<a[i][j]<<" ";
//            cout<<endl;
//        }
        if(gauss())
        {
            if(d!=-1)
            {
                d^=1;
                if(d==1)
                    printf("%.2lf\n",a[x+n/2][n+1]);
                else
                    printf("%.2lf\n",a[x][n+1]);
            }
            else
                printf("%.2lf\n",a[x][n+1]);
        }
        else
        {
            printf("impossible!\n");
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值