UVA - 10913 Walking on a Grid

簡單dp。由於只能橫着走和向下走以及每個格子只能走一次。可以枚舉i,j 這個位置時,下一行應該在哪個位置

核心部分冗餘度很高,但想不出0冗餘又不會增加複雜度的辦法。。。乾脆就直接把代碼複製一邊然後改個循環

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 76
#define MAXK 6
using namespace std;
int n,k;
int s[MAXN][MAXN],dp[MAXN][MAXN][MAXK],vis[MAXN][MAXN][MAXK];

void init(){
	memset(vis,0,sizeof(vis));
	int count=0,all=0;
	for(int i=n;i>=1;i--){
		if(s[n][i]<0)
			count++;
		if(count>k)
			break;
		all+=s[n][i];
		dp[n][i][count]=all;
		vis[n][i][count]=1;
	}
}

int main(){
	int cases=1;
	while(cin>>n>>k&&n+k){
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				cin>>s[i][j];
		init();
		for(int i=n-1;i>=1;i--){
			int count,all;
			for(int j=1;j<=n;j++){
				count=0,all=0;
				for(int t=j;t>=1;t--){
					if(s[i][t]<0)count++;
					if(count>k)break;
					all+=s[i][t];
					for(int p=0;p+count<=k;p++){
						if(vis[i+1][t][p]){
							if(vis[i][j][p+count])
								dp[i][j][p+count]=max(dp[i][j][p+count],dp[i+1][t][p]+all);
							else{
								dp[i][j][p+count]=dp[i+1][t][p]+all;
								vis[i][j][p+count]=1;
							}
						}
					}
				}
				count=0,all=0;
				for(int t=j;t<=n;t++){
					if(s[i][t]<0)count++;
					if(count>k)break;
					all+=s[i][t];
					for(int p=0;p+count<=k;p++){
						if(vis[i+1][t][p]){
							if(vis[i][j][p+count])
								dp[i][j][p+count]=max(dp[i][j][p+count],dp[i+1][t][p]+all);
							else{
								dp[i][j][p+count]=dp[i+1][t][p]+all;
								vis[i][j][p+count]=1;
							}
						}
					}
				}
			}
		}
		int flag=0,ans;
		for(int i=0;i<=k;i++){
			if(vis[1][1][i]){
				if(flag==0)
					ans=dp[1][1][i];
				else
					ans=max(ans,dp[1][1][i]);
				flag=1;
			}
		}
		cout<<"Case "<<cases++<<": ";
		if(flag)
			cout<<ans<<endl;
		else
			cout<<"impossible"<<endl;
	}
	return 0;
}


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