#267102 (C++14) No.19 ステージの選択

提出ソース

結果

問題	No.19 ステージの選択
ユーザー	tu-sa
提出日時	2018-06-19 02:38:17
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	CE (最新) AC (最初)
実行時間	-
コード長	6,435 bytes
コンパイル時間	1,059 ms
コンパイル使用メモリ	120,876 KB
最終ジャッジ日時	2024-11-14 20:28:17
合計ジャッジ時間	1,659 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

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

コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ

main.cpp: In function 'std::ostream& operator<<(std::ostream&, const iter_pair_t<T>&)':
main.cpp:77:33: error: 'ostream_iterator' was not declared in this scope
   77 |         std::copy(v.beg, v.end, ostream_iterator<typename decltype(v.beg)::reference>(out, " "));
      |                                 ^~~~~~~~~~~~~~~~
main.cpp:27:1: note: 'std::ostream_iterator' is defined in header '<iterator>'; did you forget to '#include <iterator>'?
   26 | #include <mutex>
  +++ |+#include <iterator>
   27 | using namespace std;
main.cpp:77:85: error: expected '(' before '>' token
   77 |         std::copy(v.beg, v.end, ostream_iterator<typename decltype(v.beg)::reference>(out, " "));
      |                                                                                     ^
      |                                                                                     (

ソースコード

raw source code

////////////////////////////////////////
///  tu3 pro-con template            ///
////////////////////////////////////////
#include <cassert>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <sstream>
#include <algorithm>
#include <numeric>
#include <functional>
#include <vector>
#include <queue>
#include <string>
#include <complex>
#include <stack>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <bitset>
#include <regex>
#include <type_traits>
#include <mutex>
using namespace std;

//// MACRO ////
#define countof(a) (sizeof(a)/sizeof(a[0]))

#define REP(i,n) for (int i = 0; i < (n); i++)
#define RREP(i,n) for (int i = (n)-1; i >= 0; i--)
#define FOR(i,s,n) for (int i = (s); i < (n); i++)
#define RFOR(i,s,n) for (int i = (n)-1; i >= (s); i--)
#define pos(c,i) c.being() + (i)
#define allof(c) c.begin(), c.end()
#define aallof(a) a, countof(a)
#define partof(c,i,n) c.begin() + (i), c.begin() + (i) + (n)
#define apartof(a,i,n) a + (i), a + (i) + (n)
#define long long long

#define EPS 1e-9
#define INF (1L << 28)
#define LINF (1LL << 60)

#define PREDICATE(t,a,exp) [&](const t & a) -> bool { return exp; }
#define COMPARISON_T(t) bool(*)(const t &, const t &)
#define COMPARISON(t,a,b,exp) [&](const t & a, const t & b) -> bool { return exp; }
#define CONVERTER(TSrc,t,TDest,exp) [&](const TSrc &t)->TDest { return exp; }

inline int sign_of(double x) { return (abs(x) < EPS ? 0 : x > 0 ? 1 : -1); }
inline bool inRange(int val, int min, int max) { return val >= min && val < max; }
inline bool inRange(double val, double min, double max) { return val - min > -EPS && val - max < EPS; }
inline bool inRange(int x, int y, int W, int H) { return x >= 0 && x < W && y >= 0 && y < H; } // W,H含まない

template <class T> struct vevector : public vector<vector<T>> { vevector(int n = 0, int m = 0, const T &initial = T()) : vector<vector<T>>(n, vector<T>(m, initial)) { } };
template <class T> struct vevevector : public vector<vevector<T>> { vevevector(int n = 0, int m = 0, int l = 0, const T &initial = T()) : vector<vevector<T>>(n, vevector<T>(m, l, initial)) { } };
template <class T> struct vevevevector : public vector<vevevector<T>> { vevevevector(int n = 0, int m = 0, int l = 0, int k = 0, const T &initial = T()) : vector<vevevector<T>>(n, vevevector<T>(m, l, k, initial)) { } };

//// i/o helper ////

namespace std {
template <class T1, class T2> inline istream & operator >> (istream & in, pair<T1, T2> &p) { in >> p.first >> p.second; return in; }
template <class T1, class T2> inline ostream & operator << (ostream &out, const pair<T1, T2> &p) { out << p.first << " " << p.second; return out; }
}
template <class T> T read() { T t; cin >> t; return t; }
template <class T> vector<T> read(int n) { vector<T> v; REP(i, n) { v.push_back(read<T>()); } return v; }
template <class T> vevector<T> read(int n, int m) { vevector<T> v; REP(i, n) v.push_back(read<T>(m)); return v; }
template <class T> vector<T> readjag() { return read<T>(read<int>()); }
template <class T> vevector<T> readjag(int n) { vevector<T> v; REP(i, n) v.push_back(readjag<T>()); return v; }

template <class T> struct iter_pair_t { T beg, end; };
template <class T> iter_pair_t<T> iter_pair(T beg, T end) { return iter_pair_t<T>{beg, end}; }
template <class T> ostream & operator << (ostream &out, const iter_pair_t<T> &v)
{
	std::copy(v.beg, v.end, ostream_iterator<typename decltype(v.beg)::reference>(out, " "));
	return out;
}
template <class T1> ostream & operator << (ostream &out, const vector<T1> &v) { return out << iter_pair(begin(v), end(v)); }
template <class T1> ostream & operator << (ostream &out, const set<T1> &v) { return out << iter_pair(begin(v), end(v)); }
template <class T1, class T2> ostream & operator << (ostream &out, const map<T1,T2> &v) { return out << iter_pair(begin(v), end(v)); }

struct _Reader { istream &cin;template <class T> _Reader operator ,(T &rhs) { cin >> rhs; return *this; } };
struct _Writer { ostream &cout; bool f{false}; template <class T> _Writer operator ,(const T &rhs) { cout << (f ? " " : "") << rhs; f = true; return *this; } };
#define READ(t,...) t __VA_ARGS__; (_Reader{cin}), __VA_ARGS__
#define WRITE(...) (_Writer{cout}), __VA_ARGS__; cout << endl
#define DEBUG(...) (_Writer{cerr}), __VA_ARGS__; cerr << endl

void solve();
int main()
{
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	solve();
	return 0;
}

// warshall_floyd // O(n^3)
template <class COST>
void warshall_floyd(vector<vector<COST>> &cost)
{
	size_t n = cost.size();
	REP(k, n) REP(i, n) REP(j, n) cost[i][j] = min(cost[i][j], cost[i][k] + cost[k][j]);
}

// 強連結成分 O(n^2)
// 巡回グラフを頂点集合の非巡回グラウ府に変換できる。
// memo: もっと速いアルゴリズムもある。
template <class COST>
vevector<int> strong_connection_components(const vevector<COST> &graph)
{
	vevector<int> ren;
	int N = graph.size();
	REP(i, N)
	{
		bool found = false;
		REP(j, ren.size())
		{
			int r = ren[j][0];
			if (graph[i][r] < INF && graph[r][i] < INF)
			{
				ren[j].push_back(i);
				found = true;
				break;
			}
		}
		if (!found)
		{
			ren.push_back({ i });
		}
	}

	return ren;
}

// 選択ソート O(n^2)
template <class Iter, class Pred>
void selection_sort(Iter first, Iter last, Pred pred = less<typename Iter::value_type>() )
{
	for (Iter i = first; i != last; ++i)
	{
		Iter minI = i;
		for (Iter j = i; j != last; ++j)
		{
			if (pred(*j, *minI))
			{
				minI = j;
			}
		}
		swap(*i, *minI);
	}
}

////////////////////
/// template end ///
////////////////////

void solve()
{
	READ(int, N);
	auto S = read<pair<int, int>>(N);
	
	vevector<int> route = vevector<int>(N, N, INF);
	REP(i,N)
	{
		route[S[i].second - 1][i] = 0;
	}
	warshall_floyd(route);
	vevector<int> ren = strong_connection_components(route);
	selection_sort(allof(ren), COMPARISON(vector<int>, a, b, route[a[0]][b[0]] == 0));

	int score = 0;
	REP(i, ren.size())
	{
		bool allHangaku = false;
		FOR(j, 0, i)
		{
			allHangaku |= route[ren[j][0]][ren[i][0]] == 0;
		}

		auto it = min_element(allof(ren[i]), COMPARISON(int, a, b, S[a].first < S[b].first));
		swap(*ren[i].begin(), *it);

		if (allHangaku)
		{
			score += S[ren[i][0]].first;
		}
		else
		{
			score += S[ren[i][0]].first * 2;
		}

		FOR(j, 1, ren[i].size())
		{
			score += S[ren[i][j]].first;
		}
	}

	printf("%3.1f\n", score / 2.0);

}

yukicoder

結果

コンパイルメッセージ

ソースコード