2011 ACM/ICPC 北京赛区 J题 Gem and prince

这道题很暴力，各种搜索都能过关键是怎么搜。

很重要的剪枝是开一个数组存所有颜色的个数，然后每次搜的时候相应减掉消除的个数，然后比较剩余的分数与当前最大得分，小的话直接返回。 不加这个就会TLE。

然后搜索也不要很裸的每个位置都搜，应该是一块一块的搜，否则会TLE；

16180

247

Accepted

2378ms

144kb

G++

3747B

2011-10-26 10:48:38

maddoctor

#include <iostream> #include <algorithm> #include <cstdio> #include <string> #include <cstring> #include <cmath> using namespace std; int n, m, k; int mp[10][10]; bool v2[10][10]; int xx[8] = {0, 0, 1, -1, 1, -1, 1, -1}; int yy[8] = {1, -1, 0, 0, 1, -1, -1, 1}; int cnt, ans, sum; bool in(int x, int y) { if(x <= n && x >= 1 && y <= m && y >= 1) return true; return false; } void dfs(int x, int y, int tt[10][10], int p, bool vv[10][10]) { vv[x][y] = 1; cnt++; for(int i = 0; i < 8; i++) { if(in(x + xx[i], y + yy[i]) && !vv[x + xx[i]][y + yy[i]] && tt[x + xx[i]][y + yy[i]] == p) { dfs(x + xx[i], y + yy[i], tt, p, vv); } } } void dfs2(int x, int y, int tt[10][10], int p) { v2[x][y] = 1; tt[x][y] = 0; for(int i = 0; i < 8; i++) { if(in(x + xx[i], y + yy[i]) && !v2[x + xx[i]][y + yy[i]] && tt[x + xx[i]][y + yy[i]] == p) { dfs2(x + xx[i], y + yy[i], tt, p); } } } void move(int tt[10][10]) { int i, j, a[10], cc; for(i = 1; i <= 8; i++) a[i] = 0; for(i = 1; i <= m; i++) { cc = 0; for(j = 1; j <= n; j++) if(tt[j][i] > 0) { a[++cc] = tt[j][i]; } for(j = 1; j <= n - cc; j++) tt[j][i] = 0; for(j = 1; j <= cc; j++) { tt[j + n - cc][i] = a[j]; } } for(j = 1; j <= m; j++) { cc = 0; for(i = 1; i <= n; i++) if(tt[i][j] > 0) cc++; a[j] = cc; } int tp[10]; for(i = 1; i <= n; i++) { int tc = 0; for(j = 1; j <= m; j++) if(a[j] == 0) { continue; } else { tp[++tc] = tt[i][j]; } for(j = 1; j <= tc; j++) tt[i][j] = tp[j]; for(j = tc + 1; j <= m; j++) tt[i][j] = 0; } } void click(int tt[10][10], int s, int *a, int x, int y) { int i, j; int ta[7], tmp[10][10]; bool vv[10][10]; int ts = s; for(i = 1; i <= k; i++) { if(a[i] >= 3) ts += a[i] * a[i]; } if(ts < ans) return; for(int ii = 1; ii <= n; ii++) { for(int jj = 1; jj <= m; jj++) { vv[ii][jj] = 0; v2[ii][jj] = 0; tmp[ii][jj] = tt[ii][jj]; } } if(x != 0 && y != 0) dfs2(x, y, tmp, tmp[x][y]); move(tmp); for(i = 1; i <= n; i++) { for(j = 1; j <= m; j++) { if(!vv[i][j] && tmp[i][j] > 0) { for(int ii = 1; ii <= k; ii++) ta[ii] = a[ii]; cnt = 0; dfs(i, j, tmp, tmp[i][j], vv); if(cnt >= 3) { ta[tmp[i][j]] -= cnt; click(tmp, s + cnt * cnt, ta, i, j); } } } } ans = max(ans, s); } int main() { //freopen("d:/data.in","r",stdin); //freopen("d:/data.out","w",stdout); int i, j, a[7]; while(scanf("%d%d%d", &n, &m, &k) != EOF) { memset(a, 0, sizeof(a)); memset(v2, 0, sizeof(v2)); for(i = 1; i <= n; i++) { for(j = 1; j <= m; j++) { scanf("%d", &mp[i][j]); if(mp[i][j] > 0) a[mp[i][j]]++; } } ans = 0; click(mp, 0, a, 0 , 0); printf("%d\n", ans); } return 0; }