AtCodedrのABC180E問題のbitDP
ABC180のEが、初めてbitDPでした。
解説見て自分で実装してみましたが、WAになってしまいます。
dpのところのfor文の順番を変えると通りました。
しかし、なぜ順番を変えて通ったのかよくわかりません。
該当のコード
以下がWAになったコードです。
c++
1#include<bits/stdc++.h> 2using namespace std; 3 4#define INF 1001001001 5 6int main(){ 7 8 int n; 9 cin >> n; 10 vector<int> x(n),y(n),z(n); 11 for(int i=0;i<n;i++) cin >> x[i] >> y[i] >> z[i]; 12 int max_bit=(1<<n); 13 14 //距離計算 15 vector<vector<int>> dist(n,vector<int>(n)); 16 for(int i=0;i<n;i++){ 17 for(int j=0;j<n;j++){ 18 if(i==j)continue; 19 int cost=abs(x[j]-x[i]); 20 cost+=abs(y[j]-y[i]); 21 cost+=max(0,z[j]-z[i]); 22 dist[i][j]=cost; 23 } 24 } 25 26 //dp[i][S]=iにいて(S)まわったときの、minコスト 27 vector<vector<int>> dp(n,vector<int>(max_bit,INF)); 28 for(int i=1;i<n;i++){ 29 dp[i][1<<i]=dist[0][i]; 30 } 31 32 for(int i=0;i<n;i++){ 33 for(int s=0;s<max_bit;s++){ 34 for(int j=0;j<n;j++){ 35 if(~s>>i&1)break;//sにiが含まれていなかったらcontinue; 36 if(s>>j&1)continue;//sにiが含まれてたらcontinue; 37 dp[j][s|(1<<j)]=min(dp[i][s]+dist[i][j],dp[j][s|(1<<j)]); 38 } 39 } 40 } 41 42 43 cout << dp[0][max_bit-1] << endl; 44} 45
dpの遷移のところを以下のように変更すると通りました。
c++
1 for(int s=0;s<max_bit;s++){ 2 for(int i=0;i<n;i++){ 3 for(int j=0;j<n;j++){ 4 if(~s>>i&1)continue;//sにiが含まれていなかったらcontinue; 5 if(s>>j&1)continue;//sにiが含まれてたらcontinue; 6 dp[j][s|(1<<j)]=min(dp[i][s]+dist[i][j],dp[j][s|(1<<j)]); 7 } 8 } 9 }
試したこと
以下を追加して遷移を見てみました。
cout << "s:" << bitset<4>(s) << ",i:" << i << ",j:" << j << endl;
なんとなく、通った点の集合Sを元にfor文を回さないといけないような気はしますが、何故なのかは明確にわかりません。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/11/13 13:30