[SCOI2008]着色方案(DP)

题目链接

思想

显然我们后面的决策是跟前一步相关的，因此我们可以考虑DP，可以用一个15维的数组来进行转移，但是这样显然回mle，所以我们考虑如何压缩状态，由于1<=Ci<=51 <= C_i <= 51<=Ci<=5，所以我们可以有dp数组:
dp[a1][a2][a3][a4][a5][last]dp[a_1][a_2][a_3][a_4][a_5][last]dp[a1][a2][a3][a4][a5][last]，a1a_1a1表示可以涂1块木块的有多少种颜色，以此类推，lastlastlast表示上一次用的是可以涂lastlastlast个木块的颜色。

接下来就是考虑dp方程的转移了。
举个例子：
假设上一次用的颜色是可以涂5个块的，那么下一步的状态转移就会变成：
sum=a1∗dp[a1−1][a2][a3][a4][a5][1]+a2∗dp[a1+1][a2−1][a3][a4][a5][2]+a3∗dp[a1][a2+1][a3−1][a4][a5][3]+(a4−1)∗dp[a1][a2][a3+1][a4−1][a5][4]+a5∗dp[a1][a2][a3][a4+1][a5−1][5]sum = a_1 * dp[a_1 - 1][a2][a3][a_4][a_5][1] + a_2 * dp[a_1 + 1][a_2 - 1][a_3][a_4][a_5][2] + a_3 * dp[a_1][a_2 + 1][a_3 - 1][a_4][a_5][3] + (a4 - 1) * dp[a_1][a_2][a_3 + 1][a_4 - 1][a_5][4] + a5 * dp[a_1][a_2][a_3][a_4 + 1][a_5 - 1][5]sum=a1∗dp[a1−1][a2][a3][a4][a5][1]+a2∗dp[a1+1][a2−1][a3][a4][a5][2]+a3∗dp[a1][a2+1][a3−1][a4][a5][3]+(a4−1)∗dp[a1][a2][a3+1][a4−1][a5][4]+a5∗dp[a1][a2][a3][a4+1][a5−1][5]

之所以a4−1a_4 - 1a4−1是因为，上一步选的是5，所以转移过来的时候a4+1a_4 + 1a4+1，这里面有一个是跟上一个块同颜色的，所以需要减去，其他情况同理。

考虑到数据比较小，并且这个dp方程有点难转移，因此我们可以考虑用记忆化搜索来进行dp转移。

代码

/*Author : lifehappy
*/
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define endl '\n'using namespace std;typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;const double pi = acos(-1.0);
const double eps = 1e-7;
const int inf = 0x3f3f3f3f;inline ll read() {ll f = 1, x = 0;char c = getchar();while(c < '0' || c > '9') {if(c == '-')    f = -1;c = getchar();}while(c >= '0' && c <= '9') {x = (x << 1) + (x << 3) + (c ^ 48);c = getchar();}return f * x;
}void print(ll x) {if(x < 10) {putchar(x + 48);return ;}print(x / 10);putchar(x % 10 + 48);
}const int mod = 1e9 + 7;ll dp[20][20][20][20][20][10];
int n, a[10];ll dfs(int a1, int a2, int a3, int a4, int a5, int last) {if(dp[a1][a2][a3][a4][a5][last]) return dp[a1][a2][a3][a4][a5][last];ll ans = 0;if(a1) ans = (ans + 1ll * (a1 - (last == 2)) * dfs(a1 - 1, a2, a3, a4, a5, 1)) % mod;if(a2) ans = (ans + 1ll * (a2 - (last == 3)) * dfs(a1 + 1, a2 - 1, a3, a4, a5, 2)) % mod;if(a3) ans = (ans + 1ll * (a3 - (last == 4)) * dfs(a1, a2 + 1, a3 - 1, a4, a5, 3)) % mod;if(a4) ans = (ans + 1ll * (a4 - (last == 5)) * dfs(a1, a2, a3 + 1, a4 - 1, a5, 4)) % mod;if(a5) ans = (ans + 1ll * a5 * dfs(a1, a2, a3, a4 + 1, a5 - 1, 5)) % mod;return dp[a1][a2][a3][a4][a5][last] = ans;
}int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);n = read();for(int i = 1; i <= n; i++) {int x = read();a[x]++;}for(int i = 1; i <= 5; i++) dp[0][0][0][0][0][i] = 1;print(dfs(a[1], a[2], a[3], a[4], a[5], 0));return 0;
}

[SCOI2008]着色方案(DP)相关推荐

bzoj1079: [SCOI2008]着色方案(DP)
1079: [SCOI2008]着色方案题目:传送门题解: DP刚神多年前讲过的一道神题. 二话不说,上来就是一个六维数组:F[i][a][b][c][d][e]//表示上一次涂的颜色是还剩下i次 ...
[BZOJ1079][SCOI2008]着色方案 dp
1079: [SCOI2008]着色方案 Time Limit: 10 Sec Memory Limit: 162 MB Submit: 2368 Solved: 1428 [Submit][St ...
bzoj1079: [SCOI2008]着色方案
1079: [SCOI2008]着色方案 Time Limit: 10 Sec Memory Limit: 162 MB Description 有n个木块排成一行,从左到右依次编号为1~n.你有k ...
bzoj 1079 [SCOI2008]着色方案
bzoj 1079 [SCOI2008]着色方案 Description 有n个木块排成一行,从左到右依次编号为1~n.你有k种颜色的油漆,其中第i种颜色的油漆足够涂ci个木块.所有油漆刚好足够涂满所 ...
BZOJ 1079: [SCOI2008]着色方案记忆化搜索
1079: [SCOI2008]着色方案 Time Limit: 20 Sec Memory Limit: 256 MB 题目连接 http://www.lydsy.com/JudgeOnline/p ...
BZOJ1079 [SCOI2008]着色方案记忆化搜索
1079: [SCOI2008]着色方案 Time Limit: 10 Sec Memory Limit: 162 MB Submit: 2826 Solved: 1682 [Submit][St ...
bzoj1079 [SCOI2008]着色方案
Description 有n个木块排成一行,从左到右依次编号为1~n.你有k种颜色的油漆,其中第i种颜色的油漆足够涂ci个木块.所有油漆刚好足够涂满所有木块,即c1+c2+...+ck=n.相邻两个木 ...
[SCOI2008]着色方案
这题也是看了别人题解也才会做的. 题解: 本题用的记忆优化搜索,题目说了给的颜色可以恰好图够所有木块. 题目给了能涂几块木块的颜色一共有几种.因为我们不可以连续图,所以我们把他分开来涂色. 用dp[a ...
bzoj 1079: [SCOI2008]着色方案
思路:按个数分类dp 1 #include<bits/stdc++.h> 2 #define LL long long 3 #define fi first 4 #define se se ...

[SCOI2008]着色方案(DP)

题目链接

思想

代码

[SCOI2008]着色方案(DP)相关推荐

最新文章

热门文章