首页 > 代码库 > 【滚动数组】【状压dp】Gym - 100956F - Colored Path
【滚动数组】【状压dp】Gym - 100956F - Colored Path
f(i,j,S)表示到(i,j),且经由的路径上的颜色集合为S的价值的最小值,从上方和左方转移过来即可。
要注意,内存不足,需要滚动数组优化,即使用了map,还是需要。
路径输出的时候,可以再跑一遍dp,这样就不用再开一个大数组了。
我的写法比较菜。卡了常数
#include<cstdio> #include<algorithm> #include<map> using namespace std; int w[401][401],c[401][401]; typedef pair<int,int> Point; map<int,int>f[401][401]; typedef pair<int,Point> data; typedef map<int,int>::iterator ITER; map<int,data>g[401][401]; int n,K,W; int calc(int x) { int res=0; while(x) { res+=(x&1); x>>=1; } return res; } Point path[1001]; int e; int main() { scanf("%d%d%d",&n,&K,&W); for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) scanf("%d",&w[i][j]); for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) scanf("%d",&c[i][j]); if(w[1][1]>W) { puts("-1"); return 0; } f[1][1][1<<(c[1][1]-1)]=w[1][1]; for(int i=1;i<=n;++i){ for(int j=1;j<=n;++j) { if(i>1 && j>1) { for(ITER it=f[i-1][j].begin();it!=f[i-1][j].end();++it) { int nS=(*it).first|(1<<(c[i][j]-1)),tmp=(*it).second+w[i][j]; if(f[i][j].find(nS)==f[i][j].end()) { if(tmp<=W) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i-1,j)); } } else if(f[i][j][nS]>tmp) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i-1,j)); } } for(ITER it=f[i][j-1].begin();it!=f[i][j-1].end();++it) { int nS=(*it).first|(1<<(c[i][j]-1)),tmp=(*it).second+w[i][j]; if(f[i][j].find(nS)==f[i][j].end()) { if(tmp<=W) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i,j-1)); } } else if(f[i][j][nS]>tmp) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i,j-1)); } } } else if(i==1 && j>1) { for(ITER it=f[i][j-1].begin();it!=f[i][j-1].end();++it) { int nS=(*it).first|(1<<(c[i][j]-1)),tmp=(*it).second+w[i][j]; if(f[i][j].find(nS)==f[i][j].end()) { if(tmp<=W) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i,j-1)); } } else if(f[i][j][nS]>tmp) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i,j-1)); } } } else if(i>1 && j==1) { for(ITER it=f[i-1][j].begin();it!=f[i-1][j].end();++it) { int nS=(*it).first|(1<<(c[i][j]-1)),tmp=(*it).second+w[i][j]; if(f[i][j].find(nS)==f[i][j].end()) { if(tmp<=W) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i-1,j)); } } else if(f[i][j][nS]>tmp) { f[i][j][nS]=tmp; g[i][j][nS]=data((*it).first,Point(i-1,j)); } } } } if(i>=3) { for(int j=1;j<=n;++j) { f[i-2][j].clear(); // g[i-2][j]. } } } int ans=2147483647; int anS; for(ITER it=f[n][n].begin();it!=f[n][n].end();++it) { int tmp=calc((*it).first); if(tmp<ans) { ans=tmp; anS=(*it).first; } } if(ans==2147483647) puts("-1"); else { printf("%d\n",ans); data U=data(anS,Point(n,n)); path[++e]=Point(n,n); while(U.second!=Point(1,1)) { U=g[U.second.first][U.second.second][U.first]; path[++e]=U.second; } for(int i=e;i>=1;--i) printf("%d %d%c",path[i].first,path[i].second,i==1 ? ‘\n‘ : ‘ ‘); } return 0; }
【滚动数组】【状压dp】Gym - 100956F - Colored Path
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。