2017-03-04 【SCOI2008】着色方案

Description

有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i 种颜色的油漆足够涂ci 个木块。所有油漆刚好足够涂满所有木块，即 c1+c2+…+ck=n。相邻两个木块涂相同色显得很难看，所以你希望统计任意两个相邻木块颜色不同的着色方案。

Input

第一行为一个正整数k，第二行包含k个整数c1, c2, … , ck。

Output

输出一个整数，即方案总数模1,000,000,007的结果。

Sample Input

样例1：
3
1 2 3
样例2：
5
2 2 2 2 2
样例3：
10
1 1 2 2 3 3 4 4 5 5

Sample Output

样例1：
10
样例2：
39480
样例3：
85937576
Hint

数据规模：

50%的数据满足：1 <= k <= 5, 1 <= ci <= 3
100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

Source

SCOI，递推 ，动态规划 ，散列

Time Limit 1000 ms
Memory Limit 128000 KB
Test Cases 10
Difficulty 6
Tags 动态规划 省选/NOI/WC 递推

Solution

最最最最最简单的暴力搜索没有任何优化的那种有50分噢，而且打出来只用10分钟，真是太可爱了，可以用来对拍啥的，嗯，总之就是很好

50分Code

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#define ll long long
using namespace std;
int k,ans,sum;
int c[16][2];//一维记总数，二维记已用数
void dfs(int pos,int colour)
{
  if(pos==sum)
  {
    ans++;
    return;
  }
  for(int i=1;i<=k;i++)
    if(c[i][1]<c[i][0]&&i!=colour)
      c[i][1]++,dfs(pos+1,i),c[i][1]--;
}
int main()
{
  scanf("%d",&k);
  for(int i=1;i<=k;i++)
    scanf("%d",&c[i][0]),sum+=c[i][0];
  for(int i=1;i<=k;i++)
    {
      c[i][1]++;
      dfs(1,i);
      c[i][1]--;
    }
  printf("%d\n",ans);
  return 0;
}

当然，正解还是必须得会的，思路是这样的：
开两个六维数组，一个g[a][b][c][d][e][f]表示还可以用1次的油漆数为a，两次的为b，三次的为c，四次的为d，五次的为e，f表示上一次是用的上次还可用f次的油漆那么这次在剩余f-1次的颜色中，我们只能少拿一次，因为不能拿相同的，这样转移就很容易的表示出来了。

代码是很好理解的呢

#include<cstdio>  
#include<iostream>  
#include<cstring>  
#include<cstdlib>  
#include<cmath>  
#define ll long long  
#define mod 1000000007  
using namespace std;  
int n,x[16];  
ll f[16][16][16][16][16][6];  
//f数组表示当前有能涂1次的油漆a个，能涂2次的b个....前一个颜色为k。  
bool pd[16][16][16][16][16][6];  //判重
ll dp(int a,int b,int c,int d,int e,int k)  //k用来记录上一次用到的颜色
{  
  ll t=0;  //计数
  if(pd[a][b][c][d][e][k]) return f[a][b][c][d][e][k];  
  if(a+b+c+d+e==0) return 1;  //没有油漆了
  if(a) t=t+(a-(k==2))*dp(a-1,b,c,d,e,1);    //因为上一次如果是k，那么k就会减1
  if(b) t=t+(b-(k==3))*dp(a+1,b-1,c,d,e,2);    
  if(c) t=t+(c-(k==4))*dp(a,b+1,c-1,d,e,3);    
  if(d) t=t+(d-(k==5))*dp(a,b,c+1,d-1,e,4);    
  if(e) t=t+e*dp(a,b,c,d+1,e-1,5);    
  pd[a][b][c][d][e][k]=1;//标记
  return f[a][b][c][d][e][k]=(t%mod);  
}  
int main()  
{  
  scanf("%d",&n);  
  for(int i=1;i<=n;i++)  
    {int y;  
      scanf("%d",&y);  
      x[y]++;  
    }  
  printf("%lld",dp(x[1],x[2],x[3],x[4],x[5],0));  
  return 0;  
}