pku 2978 Colored stones(枚举+DP)

枚举k种颜色的排列顺序，然后DP。
k种颜色共有k!种排列，但用DFS求出这k!种排列需要k^k的时间，记录为order[]。
DP的过程是这样的：DP[i][j]表示在(1-j)这一段，用order[1]到order[i]的颜色表示，至少需要移除多少个石头。于是DP[i][j]=min(DP[i-1][p]+(p+1到j这一段中颜色和order[i]不一样颜色的石头))。时间效率是k*m^2。
于是总的时间效率是k^k*k*m^2

#include <iostream> #define _clr(a,b) memset(a,b,sizeof(a)) template<class T> void get_min(T& a,T b) { if(a>b) a=b;} using namespace std; int m,k,color[105],ans,order[6],cut[6][105],cnt[6][105]; bool visited[6]; void DP() { for(int i=1;i<=m;i++) cnt[1][i]=cut[order[1]][i]; for(int i=2;i<=k;i++) { for(int j=1;j<=m;j++) { cnt[i][j]=cut[order[i]][j]; for(int p=1;p<=j;p++) { get_min(cnt[i][j],cnt[i-1][p]+cut[order[i]][j]-cut[order[i]][p]); } } } get_min(ans,cnt[k][m]); } void DFS(int index) { if(index==k+1) { DP(); return; } for(int i=1;i<=k;i++) { if(visited[i]==false) { visited[i]=true; order[index]=i; DFS(index+1); visited[i]=false; } } } int main() { while(scanf("%d%d",&m,&k)) { if(m==0&&k==0) break; _clr(cut,0); _clr(visited,0); for(int i=1;i<=m;i++) { scanf("%d",&color[i]); for(int j=1;j<=k;j++) { if(color[i]!=j) cut[j][i]=cut[j][i-1]+1; else cut[j][i]=cut[j][i-1]; } } ans=m; DFS(1); printf("%d/n",ans); } return 0; }