Submission #4641587

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod=1000000007;
int main(){
  ll n,x,y,z;cin>>n>>x>>y>>z;
  ll dp[n][501],dpx[n][501],dpy[n][501],dpz[n][501],dpl[n][501];
  for(ll i=0;i<n;i++){
    for(ll j=0;j<=400;j++){
      dp[i][j]=0;
      dpx[i][j]=0;
      dpy[i][j]=0;
      dpz[i][j]=0;
      dpl[i][j]=0;
    }
  }
  for(ll i=1;i<=10;i++){
    dp[0][i]=1;
    dpx[0][i]=1;
  }
  for(ll i=0;i<n-1;i++){
    for(ll j=1;j<=400;j++){
      for(ll k=1;k<=10;k++){
        dp[i+1][j+k]+=dp[i][j];
        dp[i+1][j+k]%=mod;
        dpx[i+1][j+k]+=dpx[i][j];
        dpx[i+1][k]+=dp[i][j];
        dpx[i+1][j+k]%=mod;
        dpx[i+1][k]%=mod;
        dpy[i+1][j+k]+=dpy[i][j];
        dpy[i+1][j+k]%=mod;
        if(j==x){
          dpy[i+1][k]+=dpx[i][j];
          dpy[i+1][k]%=mod;
        }
        dpz[i+1][j+k]+=dpz[i][j];
        dpz[i+1][j+k]%=mod;
        if(j==y){
          dpz[i+1][k]+=dpy[i][j];
          dpz[i+1][k]%=mod;
        }
        dpl[i+1][j+k]+=dpl[i][j];
        dpl[i+1][j+k]%=mod;
        if(j==z){
          dpl[i+1][k]+=dpz[i][j];
          dpl[i+1][k]%=mod;
        }
      }
    }
  }
  ll ans=dpz[n-1][z];
  for(ll i=1;i<=400;i++){
    ans+=dpl[n-1][i];
    ans%=mod;
  }
  cout<<ans<<endl;
  for(ll i=0;i<n;i++){
    for(ll j=0;j<=10;j++){
      //cout<<dpl[i][j]<<" ";
    }
   // cout<<endl;
  }
}
        
        

Submission Info

Judge Result