#113440 (C++14) No.416 旅行会社

提出ソース

結果

問題	No.416 旅行会社
ユーザー	satanic
提出日時	2016-08-27 18:16:44
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	314 ms / 4,000 ms
コード長	5,129 bytes
コンパイル時間	1,975 ms
コンパイル使用メモリ	137,136 KB
実行使用メモリ	16,128 KB
最終ジャッジ日時	2024-12-14 20:02:47
合計ジャッジ時間	6,045 ms
ジャッジサーバーID （参考情報）	judge2 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	84 ms 10,880 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,248 KB
testcase_03	AC	2 ms 5,248 KB
testcase_04	AC	2 ms 5,248 KB
testcase_05	AC	2 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	2 ms 5,248 KB
testcase_08	AC	4 ms 5,248 KB
testcase_09	AC	13 ms 5,248 KB
testcase_10	AC	104 ms 11,648 KB
testcase_11	AC	103 ms 11,748 KB
testcase_12	AC	101 ms 11,880 KB
testcase_13	AC	83 ms 10,880 KB
testcase_14	AC	297 ms 15,616 KB
testcase_15	AC	308 ms 15,492 KB
testcase_16	AC	307 ms 16,128 KB
testcase_17	AC	314 ms 15,616 KB
testcase_18	AC	312 ms 15,616 KB
testcase_19	AC	202 ms 15,872 KB
testcase_20	AC	213 ms 16,000 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <algorithm>
#include <bitset>
#include <cassert>
#include <complex>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <typeinfo>
#include <utility>
#include <vector>
#include <array>
#include <chrono>
#include <random>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#define INIT std::ios::sync_with_stdio(false);std::cin.tie(0);
#define VAR(type, ...)type __VA_ARGS__;Scan(__VA_ARGS__);
template<typename T> void Scan(T& t) { std::cin >> t; }
template<typename First, typename...Rest>void Scan(First& first, Rest&...rest) { std::cin >> first; Scan(rest...); }
#define OUT(d) std::cout<<d;
#define FOUT(n, d) std::cout<<std::fixed<<std::setprecision(n)<<d;
#define SOUT(n, c, d) std::cout<<std::setw(n)<<std::setfill(c)<<d;
#define SP std::cout<<" ";
#define TAB std::cout<<"\t";
#define BR std::cout<<"\n";
#define FLUSH std::cout<<std::flush;
#define VEC(type, c, n) std::vector<type> c(n);for(auto& i:c)std::cin>>i;
#define MAT(type, c, m, n) std::vector<std::vector<type>> c(m, std::vector<type>(n));for(auto& r:c)for(auto& i:r)std::cin>>i;
#define ALL(a) (a).begin(),(a).end()
#define FOR(i, a, b) for(int i=(a);i<(b);++i)
#define RFOR(i, a, b) for(int i=(b)-1;i>=(a);--i)
#define REP(i, n) for(int i=0;i<int(n);++i)
#define RREP(i, n) for(int i=(n)-1;i>=0;--i)
#define PAIR std::pair<int, int>
#define IN(a, x, b) (a<=x && x<b)
#define SHOW(d) {std::cout << #d << "\t:" << d << "\n";}
#define SHOWVECTOR(v) {std::cout << #v << "\t:";for(const auto& i : v){std::cout << i << " ";}std::cout << "\n";}
#define SHOWVECTOR2(v) {std::cout << #v << "\t:\n";for(const auto& i : v){for(const auto& j : i){std::cout << j << " ";}std::cout << "\n";}}
#define SHOWPAIRVECTOR2(v) {std::cout << #v << "\t:\n";for(const auto& i : v){for(const auto& j : i){std::cout<<'('<<j.first<<", "<<j.second<<") ";}std::cout << "\n";}}
#define SHOWPAIRVECTOR(v) {for(const auto& i:v){std::cout<<'('<<i.first<<", "<<i.second<<") ";}std::cout<<"\n";}
#define CHECKTIME(state) {auto start=std::chrono::system_clock::now();state();auto end=std::chrono::system_clock::now();auto res=std::chrono::duration_cast<std::chrono::nanoseconds>(end-start).count();std::cerr<<"[Time:"<<res<<"ns  ("<<res/(1.0e9)<<"s)]\n";}
#define SHOWQUEUE(a) {std::queue<decltype(a.front())> tmp(a);std::cout << #a << "\t:";for(int i=0; i<static_cast<int>(a.size()); ++i){std::cout << tmp.front() << "\n";tmp.pop();}std::cout << "\n";}

//#define int ll
using ll = long long;
using ull = unsigned long long;
constexpr int INFINT = 1 << 30;
constexpr ll INFLL = 1LL << 60;
constexpr double EPS = 0.0000000001;
constexpr int MOD = 1000000007;

int ans[100001];

//Quick Find Weighted
class QuickFindWeighted {
private:
	std::vector<int> group;
	std::vector<std::vector<int>> items;
public:
	QuickFindWeighted(int size) {
		group.resize(size);
		items.resize(size);
		for (int i = 0; i < size; ++i) {
			group[i] = i;
			items[i].assign({ i });
		}
	}

	// i1の所属するグループとi2の所属するグループを統合する
	// qはクエリ番号
	void merge(int i1, int i2, int q) {
		if (group[i1] == group[i2]) return;
		if (items[group[i1]].size() < items[group[i2]].size()) {
			std::swap(i1, i2);
		}
		int g1 = group[i1], g2 = group[i2];
		if (group[0] == g1) {
			for (int i : items[g2]) ans[i] = q;
		}
		else if (group[0] == g2) {
			for (int i : items[g1]) ans[i] = q;
		}
		for (int i : items[g2]) group[i] = g1;
		items[g1].insert(items[g1].end(), items[g2].begin(), items[g2].end());
		items[g2].clear();
		items[g2].shrink_to_fit();
	}

	// i1とi2が同じグループに所属しているか
	bool same(int i1, int i2) {
		return group[i1] == group[i2];
	}
};

/*
最後に残る橋を調べて、つなげる
(そのときつながっている島は-1)

橋破壊クエリを逆から読む
島pがクエリi番目で1とつながったらpはi
(つなげるときには"データ構造をマージする一般的なテク"を用いる)
(↓参考↓)
(https://topcoder.g.hatena.ne.jp/iwiwi/20131226/1388062106)

クエリを全て読んだ後につながっていないものを調べる
(そのときつながっていない島は0)
*/

signed main() {
	INIT;
	VAR(int, n, m, q);
	std::set<PAIR> edge;
	std::vector<PAIR> bridge(m), broken(q);
	// 元ある橋
	REP(i, m) {
		VAR(int, a, b);
		--a; --b;
		bridge[i].first = a;
		bridge[i].second = b;
		edge.insert(bridge[i]);
	}
	// 壊される橋
	REP(i, q) {
		VAR(int, c, d);
		--c; --d;
		broken[i].first = c;
		broken[i].second = d;
		edge.erase(broken[i]);
	}
	QuickFindWeighted qf(n);
	// 最終的に残る橋でつなげる
	for (auto it = edge.begin(); it != edge.end(); ++it) {
		qf.merge(it->first, it->second, -1);
	}
	
	// 橋破壊クエリを逆から読む
	RREP(i, q) {
		qf.merge(broken[i].first, broken[i].second, i+1);
	}
	
	// 結果
	FOR(i, 1, n) {
		OUT(ans[i])BR;
	}

	return 0;
}

yukicoder

結果

テストケース

ソースコード