PG Battle 2018 企業 ましゅまろ4:旅(難易度5)
問題
町が 個あり、 本の双方向に移動可能な道で結ばれています。町には から までの番号が、道には から までの番号が付いています。 番目の道は、町 と町 の間を結んでいて、通ると の幸福を得ることができます。町 と町 を を満たすように選び、同じ町を二回以上訪れないように町 から町 まで移動するとき、得られる幸福の総和の最大値を求めてください。
制約
考え方
巡回セールスマン問題に似た要領で解くことができる。つまり
:= 訪問頂点を として、現在 にいるときの最大値
とする。その時の遷移は、頂点 は訪問済で、頂点 は未訪問であるようなときに、
訪問後の値 = 訪問前の値+ から への辺の値
となるので、訪問後の値の最大値を集合の小さい状態から徐々に求めていけば良い。
public void solve(int testNumber, MyInput in, PrintWriter out) { int n = in.nextInt(), m = in.nextInt(); int[][] mat = new int[n][n]; fill(mat, -1); for (int i = 0; i < m; i++) { int a = in.nextInt()-1, b = in.nextInt()-1, c = in.nextInt(); mat[a][b] = c; mat[b][a] = c; } long[][] dp = new long[1 << n][n]; for (int bit = 0; bit < 1 << n; bit++) { for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { if ((bit >> i & 1) == 1 && (bit >> j & 1) == 0 && mat[i][j] != -1) { dp[bit | 1 << j][j] = max(dp[bit | 1 << j][j], dp[bit][i] + mat[i][j]); } } } } long ans = 0; for (int i = 0; i < 1 << n; i++) { for (int j = 0; j < n; j++) { ans = max(ans, dp[i][j]); } } out.println(ans); }
どこに着目して考察するべきだったか
頂点を移動する様子を考えるとDAGになっていることがわかるのでDPで解ける。
何がバグっていたか
得た知見
DAGはDP。巡回セールスマン問題の遷移で、現在 にいることを忘れない。
類題
補足
本問題は公開してよいとアナウンスがありましたので、記載しています。