UVA - 10604 Chemical Reaction

本身記憶化搜索沒什麼難的,但被自己坑了幾次。

首先是dfs裏面少了恢復minus[i]的語句,當時根本沒意識到這裏的錯誤,結果debug了半天才找到。

然後是注意處理只有一個管子的情況,這時的熱量應該是0.

最坑的就是i與j反應和j與i反應結果可能會不同,看了stainger的題解才知道

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define MAXN 7
#define MAXK 11
using namespace std;

int dp[MAXK][MAXK][MAXK][MAXK][MAXK][MAXK];
int n,k,reaction[MAXN][MAXN][2];
int amount[7];

void init(){
	memset(dp,0x7f,sizeof(dp));
	dp[0][0][0][0][0][1]=\
	dp[0][0][0][0][1][0]=dp[0][0][0][1][0][0]=\
	dp[0][0][1][0][0][0]=dp[0][1][0][0][0][0]=\
	dp[1][0][0][0][0][0]=0;
	memset(amount,0,sizeof(amount));
}

void dfs(int a,int b,int c,int d,int e,int f){
	if(dp[a][b][c][d][e][f]!=0x7f7f7f7f)
		return ;
	int help[7]={0,a,b,c,d,e,f};
	int minus[7]={0,0,0,0,0,0,0};
	for(int i=1;i<=n;i++){
		if(help[i]==0){
			continue;
		}
		minus[i]=1;
		//cout<<minus[i]<<' ';
		for(int j=1;j<=n;j++){
			if(help[i]==1&&i==j)
				continue;
			if(help[j]==0)
				continue;
			minus[j]+=1;
			minus[reaction[i][j][0]]-=1;
			dfs(a-minus[1],b-minus[2],c-minus[3],d-minus[4],e-minus[5],f-minus[6]);
			dp[a][b][c][d][e][f]=min(dp[a][b][c][d][e][f],dp[a-minus[1]][b-minus[2]][c-minus[3]][d-minus[4]][e-minus[5]][f-minus[6]]+reaction[i][j][1]);
			minus[j]=0;minus[reaction[i][j][0]]=0;
			minus[i]=1;//少了這句
		}
		minus[i]=0;
	}
}

int main(){
	int T,temp;
	char ch[100];
	cin>>T;
	while(T--){
		init();
		cin>>n;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				cin>>reaction[i][j][0]>>reaction[i][j][1];
			}
		}
		cin>>k;
		for(int i=0;i<k;i++){
			cin>>temp;
			amount[temp]++;
		}
		scanf("%s",ch);
		dfs(amount[1],amount[2],amount[3],amount[4],amount[5],amount[6]);
		int ans=dp[amount[1]][amount[2]][amount[3]][amount[4]][amount[5]][amount[6]];
		if(ans==0x7f7f7f7f)
			ans=0;
		cout<<ans<<endl;
	}
	return 0;
}


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