Topcoder-10524 BrickPuzzle

圖很小,考慮狀壓dp。每一個位置只與前面一個位置的dp值有關,因此可以使用滾動數組優化空間。因爲每一個方塊的放置最多影響到下一行往右兩個的位置,所以mask大概可以開到25位,空間是比較吃緊的,所以使用哈希表進行存儲達到空間上的優化。

狀壓dp一向是挺難寫的,對代碼能力要求較高,具體見代碼吧。。

#include<bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define ll long long
#define pq priority_queue
#define mp make_pair
#define pii pair<int,int>
#define mod 111911
#define debug(x) cerr<<#x<<"="<<x<<'\n'
const int INF=1000;

int lowbit(int x) {return x&(-x);}
int n,m,cnt=0;
struct F{
	int tot,ans;
	int head[mod];
	vector <int> val,dp,nxt;
	void init() {
		tot=0;
		ans=1e9+11;
		val.clear();
		dp.clear();
		nxt.clear();
		memset(head,-1,sizeof(head));
	}
	void update(int mask,int key) {
		ans=min(ans,key);
		int cur=mask%mod;
		for (int i=head[cur];i!=-1;i=nxt[i]) {
			if (val[i]==mask) {
				dp[i]=min(key,dp[i]);
				return;
			}
		}
		nxt.pb(head[cur]);
		dp.pb(key);
		val.pb(mask);
		head[cur]=tot++;
		return;
	}
}f[2];
//1<<25

class BrickPuzzle{
	public:
		int leastShapes(vector<string> board) {
			n=board.size(),m=board[0].size();
			int cur=0,lst=1;
			f[cur].init();f[lst].init();
			f[cur].update(0,0);
			int ans=1e9+11;
			for (int i=0;i<n;i++)
				for (int j=0;j<m;j++) {
					swap(cur,lst);
					f[cur].init();
					for (int k=0;k<f[lst].tot;k++) {
						int mask=f[lst].val[k];
						if (mask&1) {
							f[cur].update(mask>>1,f[lst].dp[k]);
							continue;
						}
						if (board[i][j]=='X') {
							f[cur].update(mask>>1,f[lst].dp[k]);
						}
						if (i+1<n&&j+1<m&&!((mask>>1)&1)&&!((mask>>(m+1))&1)&&!((mask>>m)&1)) {
							int nmask=mask>>1;
							nmask|=1;
							nmask=nmask|(1<<(m-1));
							nmask=nmask|(1<<m);
							f[cur].update(nmask,f[lst].dp[k]+1);
						}
						if (j+3<m&&!((mask>>1)&1)&&!((mask>>2)&1)&&!((mask>>3)&1)) {
							int nmask=mask>>1;
							nmask|=1;
							nmask|=2;
							nmask|=4;
							f[cur].update(nmask,f[lst].dp[k]+1);
						}
						if (i+1<n&&j+2<m&&!((mask>>1)&1)&&!((mask>>(m+1))&1)&&!((mask>>(m+2))&1)) {
							int nmask=mask>>1;
							nmask|=1;
							nmask|=(1<<m);
							nmask|=(1<<m+1);
							f[cur].update(nmask,f[lst].dp[k]+1);
						}
						if (i+1<n&&j>0&&j+1<m&&!((mask>>1)&1)&&!((mask>>m-1)&1)&&!((mask>>m)&1)) {
							int nmask=mask>>1;
							nmask|=1;
							nmask|=(1<<m-1);
							nmask|=(1<<m-2);
							f[cur].update(nmask,f[lst].dp[k]+1);
						}
					}
				}
			if (f[cur].ans==1e9+11) return -1;
			return f[cur].ans;
		}
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章