【2018QBXT刷題遊記】

Day2 TEST3

T1 count

【問題描述】
fadbec 很善於數數，他會數將 a 個紅球， b 個黃球， c 個藍球， d
個綠球排成一列，任意相鄰不同色的數目。
現在 R 君不知道 fadbec 數的對不對，想讓你也算一算。
由於數字字較大，所以請輸出除以 $10^9+$

【分析】dp，去掉相鄰相同的排列即可

#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long
const int MOD=1e9+7; 
char c;
int read(){int f=1,s=0;c=getchar();
while(c<'0'||c>'9')if(c=='-')f=-1,c=getchar();
while(c>='0'&&c<='9')s=s*10+c-'0',c=getchar();
return f*s;
}ll dp[35][35][35][35][5];ll getsum(int a,int b,int c,int d,int x){
	if(a+b+c+d==0)return 1LL; 
	ll ret=0;for(int i=1;i<=4;i++){if(x==i)continue;
		ret=ret+dp[a][b][c][d][i];
	}return ret%MOD;
}int main(){
	freopen("count.in","r",stdin);
	freopen("count.out","w",stdout);
	int a,b,c,d,sum;ll ans=0;
	a=read();b=read();c=read();d=read();
	sum=a+b+c+d;
	for(int o=1;o<=sum;o++)
	  for(int i=0;i<=a&&i<=o;i++)
	    for(int j=0;j<=b&&i+j<=o;j++)
	      for(int k=0;k<=c&&i+j+k<=o;k++){
	      	int l=o-i-j-k;
	      	if(i)dp[i][j][k][l][1]=getsum(i-1,j,k,l,1);
	      	if(j)dp[i][j][k][l][2]=getsum(i,j-1,k,l,2);
	      	if(k)dp[i][j][k][l][3]=getsum(i,j,k-1,l,3);
	      	if(l)dp[i][j][k][l][4]=getsum(i,j,k,l-1,4);
	      }ans=getsum(a,b,c,d,233);
	      printf("%lld\n",ans);
	      return 0;
}