結果

問題 No.482 あなたの名は
ユーザー umaumax
提出日時 2017-05-08 08:34:37
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 932 bytes
コンパイル時間 914 ms
コンパイル使用メモリ 83,736 KB
実行使用メモリ 35,584 KB
最終ジャッジ日時 2024-09-14 16:58:11
合計ジャッジ時間 9,173 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <map>
#include <vector>

using namespace std;

int main(int argc, const char* argv[])
{
	int N;
	int64_t K;
	cin >> N >> K;
	int D[N];
	for (auto&& d : D) cin >> d;

	//	NOTE 循環するグループに分ける
	map<int, int> m;
	map<int, vector<int>> gvm;
	for (int i = 1; i <= N; i++) m[i] = i;
	for (int i = 1; i <= N; i++) gvm[i] = vector<int>{i};
	for (int i = 1; i <= N; i++) {
		int d = D[i - 1];
		//	NOTE 異なるグループに所属する場合は併合する
		if (m[i] != m[d]) {
			vector<int>& a = gvm[m[i]];
			vector<int>& b = gvm[m[d]];
			a.insert(a.end(), b.begin(), b.end());
			gvm.erase(m[d]);
			m[d] = m[i];
		}
	}

	int64_t cnt = 0;
	for (auto&& tmp : gvm) cnt += max((int)(tmp.second.size() - 1), 0);

	//	NOTE 2,3回の倍数の交換は組み合わせで可能
	bool ret = cnt == K || cnt + 2 <= K;
	cout << (ret ? "YES" : "NO") << endl;
	return 0;
}
0