ZOJ 3605 Find the Marble

转载自 http://blog.sina.com.cn/s/blog_5123df350100zp0s.html
照着题解写一次，就当自己懂了。
zoj 和virtual oj 都挂了。好像。。

n个罐子，有一个石头放在第s号罐子里，然后交换这些罐子m次，只能记住k次，每次交换被忘掉的概率相同，那个猜的人（按概率最大的猜）最可能猜哪号罐子里有石头。用dp[i][j][k]表示交换了i次，忘掉了j次，石头在k号罐子里的概率。知道dp[i-1][j][t]开始推，如果它忘掉了这次dp[i][j+1][t]+=dp[i-1][j][t]*p2。如果没忘掉这次，如果这次交换(x,y)有x==t，则dp[i][j][y]+=dp[i-1][j][t]*p1,同理如果y==t则dp[i][j][y]+=dp[i-1][j][t]*p1;否则的话dp[i][j][t]=dp[i-1][j][t]*p1。其中p1是没忘记的概率,p2是忘记的概率p1=k/m , p2=1-p1。 初始dp[0][0][s]=1。最终求argmax{dp[m][m-k][x]}。

#include<stdio.h>
#include<algorithm>
#include<string.h>

struct  sp
{
    int x,y;
}p[100];
double  dp[100][100][100];
int main()
{
//  freopen("F:\\123.txt","r",stdin);
        int t,m,n,k,s;
    scanf("%d",&t);
    for(int ik=1;ik<=t;ik++)
    {
       scanf("%d%d%d%d",&n,&m,&k,&s);
       for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&p[i].x ,&p[i].y );

        }
     memset(dp,0,sizeof(dp));
     double p1=(double)k/m,p2=(double)1-p1;

      // printf("%f %f\n",p1,p2);
       dp[0][0][s]=1;

       for(int i=1;i<=m;i++)
      {
         for(int j=0;j<=m-k &&j<=i;j++)
         {
            for(int u=1;u<=n;u++)
            {
               dp[i][j+1][u]+=dp[i-1][j][u] *p2;
         // printf("see %d %d\n",p[i].x ,u);
                if(p[i].x ==u )
                {
                   dp[i][j][p[i].y ]+=dp[i-1][j][u]*p1;
        //    printf("look %d %f\n",p[i].y ,dp[i][j][p[i].y ]);
                }else if(p[i].y ==u)
                 {
                    dp[i][j][p[i].x ]+=dp[i-1][j][u]*p1;
                 }
                 else
                 {
                     dp[i][j][u]+=dp[i-1][j][u];
        /* 题解一开始漏了说这种转移情况，然后我用codeblock演算过程，发现当    p[i].x ==u 和    p[i].y ==u 都不成立的时候，应该
        有       dp[i][j][u]+=dp[i-1][j][u]; 情况，即是看到了，但是交换的两个罐子都不含有硬币。。。即当前状态，需要继承i-1时的状态*/
                 }
            }
         }
      }
        int maxi=1;
        for(int i=1;i<=n;i++)
        {
        //  printf("%d %f\n",i,dp[m][m-k][i]);
            if(dp[m][m-k][i]>dp[m][m-k][maxi])
            {
                maxi=i;
            }
          }
          printf("%d\n",maxi);
    }
    return 0;
}