UVA 12089 The Luncheon

DP水题，用三维数组记录状态，一维是天数，二维是点到A喜欢菜的天数，三维是点到B喜欢菜的天数，唯一需要注意的是循环时二维加三维的和可能大于天数（一维），不要为了节省时间搞错,

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
double r[105][105][105];
int f[40];
int a[4];
int main()
{
    int n,s,t,d,x,k,i,j,num=1;
    double sum;
    while(scanf("%d",&n),n)
    {
        sum=0;
        memset(f,0,sizeof(f));
        memset(a,0,sizeof(a));
        scanf("%d",&s);
        while(s--)
        {
            scanf("%d",&x);
            f[x]++;
        }
        scanf("%d",&t);
        for(i=0;i<t;i++)
        {
            scanf("%d",&x);
            f[x]+=2;
        }
        scanf("%d",&d);
        for(i=0;i<=d;i++)
        {
            for(j=0;j<=d;j++)
            {
                for(k=0;k<=d;k++)
                    r[i][j][k]=0;
            }
        }
        for(i=1;i<=n;i++)
        a[f[i]]++;
        r[0][0][0]=1;
        for(i=1;i<=d;i++)
        {
            for(j=0;j<=i;j++)
            {
                for(k=0;k<=i;k++)//错误写法：for(k=0;k+j<=i;i++)
                {
                          r[i][j][k]+=r[i-1][j][k]*a[0]/n;
                   if(j>0)r[i][j][k]+=r[i-1][j-1][k]*a[1]/n;
                   if(k>0)r[i][j][k]+=r[i-1][j][k-1]*a[2]/n;
                    if(j>0&&k>0)r[i][j][k]+=r[i-1][j-1][k-1]*a[3]/n;
                }
            }
        }
        for(j=0;j<=d/2;j++)
        {
            sum=sum+r[d][j][j*2];
        }
        printf("Case %d: ",num++);
        printf("%.5f\n",sum);
    }
    return 0;
}