开始以为是一道组合数学,看了别人的题解发现这是dp 开始也应该想到,必须一步步递推,所以应该是dp
dp[i][j][k][s]i代表第一种水果剩下的数量,j代表第二种水果剩下的数量,k代表第三种水果剩下的数量,s代表这是第几种水果
睡前一题
#include <bits/stdc++.h>
#include <string.h>
#include <math.h>
using namespace std;
const int mod=1000000007;
int dp[55][55][55][3];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n1,n2,n3,d1,d2,d3;
memset(dp,0,sizeof(dp));
cin>>n1>>n2>>n3>>d1>>d2>>d3;
for(int i=n1;i>=0;i--)
for(int j=n2;j>=0;j--)
for(int k=n3;k>=0;k--)
{
for(int s=1;s<=(i>d1?d1:i);s++)
{
if(i==n1&&j==n2&&k==n3)
{
dp[i-s][j][k][0]+=dp[i][j][k][0]+1;
}
else
{
dp[i-s][j][k][0]+=(dp[i][j][k][1]+dp[i][j][k][2])%mod;
dp[i-s][j][k][0]%=mod;
}
}
for(int s=1;s<=(j>d2?d2:j);s++)
{
if(i==n1&&j==n2&&k==n3)
{
dp[i][j-s][k][1]+=dp[i][j][k][1]+1;
}
else
{
dp[i][j-s][k][1]+=(dp[i][j][k][2]+dp[i][j][k][0])%mod;
dp[i][j-s][k][1]%=mod;
}
}
for(int s=1;s<=(k>d3?d3:k);s++)
{
if(i==n1&&j==n2&&k==n3)
{
dp[i][j][k-s][2]+=dp[i][j][k][2]+1;
}
else
{
dp[i][j][k-s][2]+=(dp[i][j][k][1]+dp[i][j][k][0])%mod;
dp[i][j][k-s][2]%=mod;
}
}
}
int ans=0;
ans+=((dp[0][0][0][2]+dp[0][0][0][1])%mod+dp[0][0][0][0])%mod;
cout<<ans<<endl;
}
return 0;
}