山东省第七届ACM省赛 Feed the monkey

开始以为是一道组合数学,看了别人的题解发现这是dp   开始也应该想到,必须一步步递推,所以应该是dp

dp[i][j][k][s]i代表第一种水果剩下的数量,j代表第二种水果剩下的数量,k代表第三种水果剩下的数量,s代表这是第几种水果

睡前一题

#include 
#include 
#include 
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<


你可能感兴趣的:(第七届蓝桥杯大赛预赛)