結果
| 問題 | No.121 傾向と対策：門松列（その２） |
| ユーザー |
 tu-sa
|
| 提出日時 | 2018-06-29 09:21:50 |
| 言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
CE
(最新)
AC
(最初)
|
| 実行時間 | - |
| コード長 | 7,319 bytes |
| コンパイル時間 | 1,143 ms |
| コンパイル使用メモリ | 120,444 KB |
| 最終ジャッジ日時 | 2024-11-14 20:31:05 |
| 合計ジャッジ時間 | 3,095 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge4 |
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
ただし、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, " "));
| ^
| (
ソースコード
////////////////////////////////////////
/// 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とかが解ける。
struct SegTree
{
typedef int T;
int size;
vector<T> tree;
unsigned msb(unsigned v) { REP(i, 5) v |= v >> (1 << i); return (v >> 1) + 1; }
SegTree(int count, T initialFill = INF)
: size(msb(count - 1) << 1)
, tree(size * 2)
{
fill(partof(tree, size, size), initialFill);
rebuild();
}
// []を使うと更新されないのでrebuildすること。数が少なければupdateの方がよい。
T& operator[](int i) { return tree[i + size]; }
void rebuild() { RREP(i, size) { tree[i] = 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] = 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 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);
DEBUG(N);
// compress
//map<int, int> MP;
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<int> lel(N), grl(N);
vector<int> ler(N), grr(N);
vector<int> l(M), r(M);
{
SegTree tree(M, 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 tree(M, 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 -= (llong)l[A[i]] * r[A[i]];
r[A[i]]--;
ans += (llong)lel[i] * ler[i];
ans += (llong)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 += (llong)l[A[i]] * r[A[i]];
}
WRITE(ans);
}