题目意思不说了,模拟都不怎么会描述,
本题目,直接dfs暴力哪一个点该修改然后搜索即可,
剪枝两个 : 一,对于联通的点,只需尝试其中一个(避免重复搜索)。 二,记录当前各种颜色剩余值,若当前所得值+最大期望可得<=当前最优值,剪枝(最优性剪枝)。
这份代码神奇之处是 G++能过,C++不能过。
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
#include<string>
#include<vector>
#include <iostream>
#include<assert.h>
#define pb push_back
#define rep1(i,j,k) for(int i=(j);i<=(int)k;i++)
#define per1(i,j,k) for(int i=(j);i>=(int)k;i--)
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define lowbit(x) ((x)&(-(x)))
#define fi first
#define se second
#define pii pair<int,int>
#define VI vector<int>
#define mem(v,c) memset(v,c,sizeof(v))
using namespace std;
typedef long long ll;
const int N = 9;
struct node{int v[N][N]; node(){ mem(v,0);}};
int ma[N][N],n,m,K;
const int dx[]={-1,0,1,0,-1,-1,1,1};
const int dy[]={0,-1,0,1,1,-1,1,-1};
inline int judge(int i,int j){ return i>=1&&i<=n&&j>=1&&j<=m;}
void show(int ma[N][N]){rep1(i,1,n) rep1(j,1,m){cout<<ma[i][j]<<" "; if(j==m) cout<<endl;} }
int cnt__(int x,int y,node& V){
V.v[x][y]=1;
int cnt=1;
rep(d,8){
int nx=dx[d]+x,ny=dy[d]+y;
if(judge(nx,ny) && ma[nx][ny]==ma[x][y] && !V.v[nx][ny])
cnt+=cnt__(nx,ny,V);
}
return cnt;
}
void col(int x,int y,int c){
ma[x][y]=0;
rep(d,8){
int nx=dx[d]+x,ny=dy[d]+y;
if(judge(nx,ny) && ma[nx][ny]==c){
col(nx,ny,c);
}
}
}
inline void standard(){
for(int j=1;j<=m;j++){
int p = -1;
for(int i=n;i>=1;i--){
if(!ma[i][j]){
if(p == -1) p = i-1;
for(;p>0&&(!ma[p][j]);p--);
if(!p) break;
swap(ma[i][j],ma[p][j]);
}
if(!p) break;
}
}
int j=1,tj = 1;
for(;tj<=m;){
int ok = 0;
for(int i=1;i<=n;i++) if(ma[i][tj]){ ok=1; break;}
if(ok){
rep1(i,1,n) ma[i][j]=ma[i][tj]; j++;
}
tj++;
}
for(;j<=m;j++) rep1(i,1,n) ma[i][j]=0;
}
int ans,cou[N];
inline int cal(){
int sum = 0;
for(int i=1;i<=K;i++) if(cou[i]>2) sum+=cou[i]*cou[i];
return sum;
}
void dfs(int sum){
if(sum + cal() < ans) return ;
node V;
int tma[N][N],ok=0;
rep1(i,1,n) rep1(j,1,m){
if(ma[i][j] && !V.v[i][j]){
int cnt = cnt__(i,j,V);
if(cnt < 3) continue;
int ti = ma[i][j];
memcpy(tma,ma,sizeof(ma)); cou[ti]-=cnt;
col(i,j,ma[i][j]); standard();
dfs(cnt*cnt+sum);
memcpy(ma,tma,sizeof(ma)); cou[ti]+=cnt;
}
}
ans=max(ans,sum);
}
int main()
{
while(scanf("%d %d %d",&n,&m,&K)==3){
rep1(i,1,n) rep1(j,1,m) scanf("%d",&ma[i][j]);
ans = 0;
mem(cou,0);
rep1(j,1,n) rep1(k,1,m) cou[ma[j][k]]++;
dfs(0);
printf("%d\n",ans);
}
return 0;
}