CodeForces 295 C.Greg and Friends（bfs+dp）

Description

有一个载重为$k$千克的船，现在有$n$个人要过河，只有一条船，人的体重要么$50$千克，要么$100$千克，问有多少种方案可以使得这些人全部到河对岸且船来回次数最少

Input

第一行两个整数$n,k$表示人数和载重量，之后输入$n$个整数表示每个人的体重，要么$50$要么$100$$(1\le n\le 50,1\le k\le 5000)$

Output

如果有方案可以使得所有人可以过河则输出船来回的最少次数以及方案数，结果模$10^9+7$，若无解则来回次数输出为$-1$，方案数输出$0$

Sample Input

1 50
50

Sample Output

1

1

Solution

以$dp[i][j][k]$表示在河这边($k=0$)或者河对岸$(k=1)$，体重为$50$千克的人有$i$个，体重为$100$千克的人有$j$个的方案数，$dis[i][j][k]$为达到这个状态的最少步数，令$n_0,n_1$分别为体重为$50$和$100$的人数，初始$dis[n_0][n_1][0]=0,dp[n_0][n_1][0]=1$，用队列存下当前的状态$BFS$转移，每次会从$i$个体重为$50$的人中选$x$个，从$j$个体重为$100$的人中选$y$个让这些人去对岸，只要满足$x+y>0,50x+100y\le k$即可，令$ii=n_0-i+x,jj=n_1-j+y$，则这次坐船过去，状态$(i,j,k)$变成$(ii,jj,1-k)$，如果这个状态没有出现过，那么$dis[i][j][k]+1$即为到达这个状态的最少步数，该状态入队转移，如果这个状态之前出现过，且$dis[ii][jj][1-k]=dis[i][j][k]+1$，说明当前这步转移也是最优方案，要将方案数$dp[i][j][k]\cdot C_i^x\cdot C_j^y$加到$dp[ii][jj][1-k]$中，最后$dis[n_0][n_1][1]$即为将所有人转移到河对岸的最少步数，$dp[n_0][n_1][1]$即为方案数

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=55;
#define mod 1000000007
int n,k,C[maxn][maxn],dis[maxn][maxn][2],dp[maxn][maxn][2];
struct node
{
    int x,y,z;
};
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int main()
{
    C[0][0]=1;
    for(int i=1;i<=50;i++)
    {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)add(C[i][j],C[i-1][j-1]+C[i-1][j]);
    }
    scanf("%d%d",&n,&k);
    int n0=0,n1=0;
    while(n--)
    {
        int temp;
        scanf("%d",&temp);
        if(temp==50)n0++;
        else n1++;
    }
    memset(dis,-1,sizeof(dis));
    dis[n0][n1][0]=0,dp[n0][n1][0]=1;
    queue<node>que;
    que.push((node){n0,n1,0});
    while(!que.empty())
    {
        node now=que.front();
        que.pop();
        int x=now.x,y=now.y,z=now.z;
        for(int i=0;i<=x;i++)
        {
            if(i*50>k)break;
            for(int j=0;j<=y;j++)
            {
                if(i+j==0)continue;
                if(i*50+j*100>k)break;
                int xx=n0-x+i,yy=n1-y+j,zz=1-z;
                if(dis[xx][yy][zz]==-1)
                {
                    dis[xx][yy][zz]=dis[x][y][z]+1;
                    dp[xx][yy][zz]=(ll)dp[x][y][z]*C[x][i]%mod*C[y][j]%mod;
                    que.push((node){xx,yy,zz});
                }
                else if(dis[xx][yy][zz]==dis[x][y][z]+1)
                    add(dp[xx][yy][zz],(ll)dp[x][y][z]*C[x][i]%mod*C[y][j]%mod);
            }
        }
    }
    printf("%d\n%d\n",dis[n0][n1][1],dp[n0][n1][1]);
    return 0;
}