ZOJ 3711 Give Me Your Hand

Give Me Your Hand

---

Time Limit: 2 Seconds      Memory Limit: 131072 KB      Special Judge

---

BellyWhite and MightyHorse are two students of Marjar University. They are best friends and also roommates living in the same dorm. There are K students living in that dorm, including BellyWhite and MightyHorse.

BellyWhite posted a strange microblog in 2012-9-3:

 

 

This is quite strange. It means that BellyWhite will chop some "hands" into pieces if he's been found playing games or some other activities which is not relevant to his research. This logic is insane, isn't it?

We could call those things which are not relevant to BellyWhite's research "bad things". Now we know that BellyWhite will start doing bad things when he's in the dorm for T minutes, and he will stop doing those things when he's being warned by MightyHorse or leaving the dorm. If he's been warned to stop doing bad things and then he stays in the dorm for another T minutes, he will start doing bad things again.

MightyHorse noticed the strange microblog BellyWhite posted, but he's a good roommate, so he must took the risk of losing his "hands" to warn BellyWhite that he was doing something harmful to his research progress or even his PhD degree. Fortunately, MightyHorse has M plastic toy "hands", so BellyWhite will only chop those toy hands into pieces when he's being warned.

Here comes the problem. We only know that no one is in the dorm initially, and we heard N door open and close sounds, which means there are N people entered or exited the dorm. We logged exact time of all door sounds, but we don't know who made the sound, that means all K students living in the dorm have the same possibility to enter / exit. We'd like to know the expected number of toy hands MightyHorse will have after 24 hours (1440 minutes).

Please note that using toy hand to stop BellyWhite from doing bad things take no time, which means that even at the moment MightyHorse or BellyWhite enter / exit the dorm, a toy hand will be used to stop the bad thing. But if that there's no toy hand left, MightyHorse will not able to stop BellyWhite from doing bad things.

Input

There are multiple test cases. The first line of input is an integer Casenum indicating the number of test cases.

For each test case, the first line contains 4 integers T (1 ≤ T ≤ 100), N (1 ≤ N ≤ 100), M (1 ≤ M ≤ 100) and K (2 ≤ K ≤ 8) which are defined in descriptions above.

The following line contains N integers. For every integer ti(0 ≤ ti ≤ 1440, 0 ≤ i < N), it means that a student entered or exited the dorm at time ti.

We guarantee that all those N ti will be given in strictly increasing order.

Output

For each test case, output the expected number of toy hands that MightyHorse still have at time 1440, rounded to 6 decimal points.

Sample Input

260 2 10 2200 260100 2 8 51340 1341

Sample Output

5.0000007.960000

---

Author: FAN, Yuzhe
Contest: The 10th Zhejiang Provincial Collegiate Programming Contest 

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

double dp[110][4][110][110];
///dp[第几次][位置关系][被警告一次后再屋内待了多久][剩下几只手]=概率
int t[110];

int ca,T,N,M,K;

int main()
{
    scanf("%d",&ca);
    while(ca--)
    {
        scanf("%d%d%d%d",&T,&N,&M,&K);
        memset(dp,0,sizeof(dp));
        dp[0][0][0][M]=1.0;t[0]=0;
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&t[i]);
            int ti=t[i]-t[i-1];
            ///state 0: b  X   m  X
            for(int j=0;j<=M;j++)
            {
                dp[i][0][0][j]+=dp[i-1][0][0][j]*(K-2.)*1./K;
                dp[i][1][0][j]+=dp[i-1][0][0][j]*1./K;
                dp[i][2][0][j]+=dp[i-1][0][0][j]*1./K;
            }
            ///state 1: b X m Y
            for(int j=0;j<=M;j++)
            {
                dp[i][0][0][j]+=dp[i-1][1][0][j]*1./K;
                dp[i][1][0][j]+=dp[i-1][1][0][j]*(K-2.)*1./K;
                dp[i][3][0][j]+=dp[i-1][1][0][j]*1./K;
            }
            ///state 2 b Y  m X
            for(int k=0;k<=T;k++)
            {
                for(int j=0;j<=M;j++)
                {
                    dp[i][0][0][j]+=dp[i-1][2][k][j]*1./K;
                    dp[i][2][min(T,k+ti)][j]+=dp[i-1][2][k][j]*(K-2.)*1./K;
                    dp[i][3][(k+ti)>=T?0:(k+ti)][max(0,j-((k+ti)>=T))]+=dp[i-1][2][k][j]*1./K;
                }
            }
            ///state 3 b Y m Y
            for(int k=0;k<=T;k++)
            {
                for(int j=0;j<=M;j++)
                {
                    dp[i][1][0][max(0,j-(k+ti)/T)]+=dp[i-1][3][k][j]*1./K;
                    dp[i][2][(k+ti)%T][max(0,j-(k+ti)/T)]+=dp[i-1][3][k][j]*1./K;
                    dp[i][3][(k+ti)%T][max(0,j-(k+ti)/T)]+=dp[i-1][3][k][j]*(K-2.)/K;
                }
            }
        }
        double ans=0.;
        for(int i=0;i<=T;i++)
        {
            for(int j=0;j<=2;j++)
            {
                for(int k=0;k<=M;k++)
                {
                    ans+=dp[N][j][i][k]*k;
                }
            }
        }
        int ti=1440-t[N];
        for(int i=0;i<=T;i++)
        {
            for(int j=0;j<=M;j++)
            {
                ans+=dp[N][3][i][j]*max(0,j-(i+ti)/T);
            }
        }
        printf("%.6lf\n",ans);
    }
    return 0;
}