2018QBXT刷题游记（7）

【2018QBXT刷题游记】

Day2 TEST3

T1 count

【问题描述】
fadbec 很善于数数，他会数将 a 个红球， b 个黄球， c 个蓝球， d
个绿球排成一列，任意相邻不同色的数目。
现在 R 君不知道 fadbec 数的对不对，想让你也算一算。
由于数字字较大，所以请输出除以 $10^9+7$ 的余数。
【输入格式】
一行四个正整数 a,b,c,d。
【输出格式】
输出包含一个整数，表示答案。
【样例输入 1】
1 1 1 2
【样例输出 1】
36
【数据规模及约定】
对于前 30% 的数据， 1 ≤ a; b; c; d ≤ 3。
对于前 100% 的数据， 1 ≤ a; b; c; d ≤ 30。

【分析】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;
}