#269471 (C++14) No.121 傾向と対策：門松列（その２）

提出ソース

結果

問題	No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー	tu-sa
提出日時	2018-06-29 09:13:48
言語	C++14 (gcc 13.3.0 + boost 1.87.0)
結果	CE (最新) AC (最初)
実行時間	-
コード長	7,584 bytes
コンパイル時間	1,083 ms
コンパイル使用メモリ	123,172 KB
最終ジャッジ日時	2024-11-14 20:31:08
合計ジャッジ時間	2,035 ms
ジャッジサーバーID （参考情報）	judge4 / judge5

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

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

コンパイルメッセージ

main.cpp: In function 'std::ostream& operator<<(std::ostream&, const iter_pair_t<T>&)':
main.cpp:82:33: error: 'ostream_iterator' was not declared in this scope
   82 |         std::copy(v.beg, v.end, ostream_iterator<typename decltype(v.beg)::reference>(out, " "));
      |                                 ^~~~~~~~~~~~~~~~
main.cpp:29:1: note: 'std::ostream_iterator' is defined in header '<iterator>'; did you forget to '#include <iterator>'?
   28 | #include <array>
  +++ |+#include <iterator>
   29 | using namespace std;
main.cpp:82:85: error: expected '(' before '>' token
   82 |         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 <iomanip>
#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>
#include <array>
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)
typedef unsigned int uint;
typedef long long llong;
typedef unsigned long long ullong;
#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; v.reserve(n); 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; v.reserve(n); 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 << '\n'
#define DEBUG(...) (_Writer{cerr}), __VA_ARGS__; cerr << '\n'

void solve();
int main()
{
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	cout << setprecision(std::numeric_limits<double>::max_digits10);
	solve();

	return 0;
}


/// 2分セグメント木。RMQとかが解ける。
template<class T = int, class TParentFunc = T (*)(const T&, const T&)>
struct SegTree
{
	int size;
	vector<T> tree;
	TParentFunc parent;
	unsigned msb(unsigned v) { REP(i, 5) v |= v >> (1 << i); return (v >> 1) + 1; }

	SegTree(int count, TParentFunc parentBuilder = std::min, T initialFill = INF)
		: size(msb(count - 1) << 1)
		, tree(size * 2)
		, parent(parentBuilder)
	{
		fill(partof(tree, size, size), initialFill);
		rebuild();
	}

	// []を使うと更新されないのでrebuildすること。数が少なければupdateの方がよい。
	T& operator[](int i) { return tree[i + size]; }
	void rebuild() { RREP(i, size) { tree[i] = parent(tree[i * 2], tree[i * 2 + 1]); } }

	void update(int index, T value)
	{
		tree[index + size] = value;
		for (int i = (index + size) / 2; i > 0; i /= 2)
			tree[i] = parent(tree[i * 2], tree[i * 2 + 1]);
	}

	T query(int s, int e, int i, int l, int r, T value)
	{
		if (r <= s || e <= l) return value;
		if (s <= l && r <= e) return parent(value, tree[i]);
		value = query(s, e, i * 2, l, (l + r) / 2, value);
		value = query(s, e, i * 2 + 1, (l + r) / 2, r, value);
		return value;
	}

	T query(int start, int count, T initialValue)
	{
		return query(start, start + count, 1, 0, size, initialValue);
	}
};



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


void solve()
{
	//freopen("input.txt", "r", stdin);

	READ(int, N);
	auto A = read<int>(N);
	auto A_ = A;
	//auto MP = map<int, int>();

	// compress
	int M;
	{
		map<int, int> comp;
		for (auto a : A) { comp[a] = 0; }
		int i = 0; for (auto &p : comp) { p.second = i++; }
		for (auto &a : A) { /* MP[comp[a]] = a; */ a = comp[a];  }
		M = comp.size();
	}
	DEBUG(M);
	vector<llong> lel(N), grl(N);
	vector<llong> ler(N), grr(N);
	vector<llong> l(M), r(M);
	
	{
		SegTree<llong> tree(M, [](const llong &a, const llong &b) {return a + b; }, 0);
		REP(i, N)
		{
			auto a = A[i];
			tree.update(a, tree[a] + 1);
			lel[i] = tree.query(0, a, 0);
			grl[i] = tree.query(a + 1, N - a - 1, 0);
		}
	}
	{
		SegTree<llong> tree(M, [](const llong &a, const llong &b) {return a + b; }, 0);
		RREP(i, N)
		{
			auto a = A[i];
			tree.update(a, tree[a] + 1);
			ler[i] = tree.query(0, a, 0);
			grr[i] = tree.query(a + 1, N - a - 1, 0);
		}
	}

	REP(i,N)
	{
		r[A[i]]++;
	}

	r[A[0]]--;
	l[A[0]]++;

	llong ans = 0;
	llong minus = l[A[0]] * r[A[0]];

	FOR(i, 1, N-1)
	{
		minus -= l[A[i]] * r[A[i]];
		r[A[i]]--;

		ans += lel[i] * ler[i];
		ans += grl[i] * grr[i];

		//int minu = 0;
		//DEBUG("loop: ", A_[i]);
		//REP(j, M)
		//{
		//	if (j != A[i])
		//	{
		//		minu += l[j] * r[j];
		//		if (l[j] * r[j])
		//		{
		//			DEBUG("", MP[j], ":", l[j], "*", r[j], "=", l[j] * r[j]);
		//		}
		//	}
		//	
		//}
		//DEBUG("-->", minu);
		//DEBUG("-->", minus);
		ans -= minus;

		l[A[i]]++;
		minus += l[A[i]] * r[A[i]];
	}

	WRITE(ans);

}

yukicoder

結果

コンパイルメッセージ

ソースコード