結果

問題 No.1577 織姫と彦星2
ユーザー kagasantwi
提出日時 2021-06-19 15:58:22
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 172 ms / 2,000 ms
コード長 2,411 bytes
コンパイル時間 835 ms
コンパイル使用メモリ 87,620 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-22 22:07:13
合計ジャッジ時間 5,179 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>
#include <map>
using namespace std;

/**
 * 方針
 * 石をノードとした無向グラフとして考える。
 * ハミング距離が1以下の石の間に重み1のエッジを張る。
 * start, end間の最短距離を求める。
 * ノード数に対してエッジが少なめなことを利用する。
 */

int main(){
    
    // ローカルではテストケースを1回の実行で全部試す
    for (int N; cin >> N;) {
        // 石はstart, end, NのN+2種類
        vector<int>stone(N + 2);
        // startの石から何種類使ってその石に辿り着けたか。辿り着けない場合は-1。
        vector<int>score(N + 2, -1);
        
        // 石の性質を入力
        cin >> stone[0] >> stone[1];
        for(int i = 2; i < N + 2; i++)cin >> stone[i];
        
        // 性質の場所を記録しておく
        map<int, int>stonePosition;
        for (int i = 0; i < N + 2; i++) {
            stonePosition[stone[i]] = i;
        } 

        // 幅優先探索用キュー
        queue<int>Q;
        // 幅優先探索の初期状態
        score[0] = 0;
        Q.push(0);
        
        // 幅優先探索
        while(!Q.empty()){
            int index = Q.front();
            Q.pop();
            
            // 未利用の石を調べる
            for (int bit = 1; bit <= 1000000000; bit <<= 1) {
                // 1bitだけ異なる石の性質
                int target = stone[index] ^ bit;
                // そんな石があるかどうか
                if (stonePosition.find(target) == stonePosition.end())continue;
                int toIndex = stonePosition[target];
                
                // 繋げられない場合はcontinue
                if (__builtin_popcount(stone[index] ^ stone[toIndex]) > 1) continue;
                // 利用済みの場合はcontinue
                if (score[toIndex] >= 0) continue;
                
                // 利用する石のscoreを更新してキューに追加
                score[toIndex] = score[index] + 1;
                Q.push(toIndex);
            }
        }
        
        // 出力
        if (score[1] >= 0) {
            // end側の石は答えに含めない
            cout << score[1] - 1 << endl;    
        } else {
            cout << -1 << endl;
        }
    }
    
    return 0;
}
0