結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー uenoku
提出日時 2017-02-21 14:14:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3,864 ms / 5,000 ms
コード長 2,757 bytes
コンパイル時間 4,042 ms
コンパイル使用メモリ 112,524 KB
実行使用メモリ 128,120 KB
最終ジャッジ日時 2025-03-31 18:08:23
合計ジャッジ時間 11,987 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include "math.h"
#include <algorithm>
#include <complex>
#include <cstdio>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <vector>
#define ifor(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 < (n); i++)
#define rrep(i, n) for (int i = (n)-1; i >= 0; i--)
using namespace std;
typedef long double ld;
typedef long long int lli;
typedef complex<double> P;
const double eps = 1e-11;
int vex[4] = {1, 0, -1, 0};
int vey[4] = {0, 1, 0, -1};
typedef vector<double> Vec;
typedef vector<int> vec;
typedef vector<Vec> MAT;
typedef vector<vec> mat;
lli MOD = 1000000007;
template <class T>
class bit
{
public:
    vector<T> dat;
    int n;

    bit(int n) : n(n) { dat.assign(n, 0); }
    bit(int n, T ini) : n(n) { dat.assign(n, ini); }
    // sum [0,i)
    T sum(int i)
    {
        int ret = 0;
        for (--i; i >= 0; i = (i & (i + 1)) - 1)
            ret += dat[i];
        return ret;
    }
    // sum [i,j)
    T sum(int i, int j) { return sum(j) - sum(i); }
    // add x to i
    void add(int i, T x)
    {
        for (; i < n; i |= i + 1)
            dat[i] += x;
    }
};
lli n;
lli solve(vector<lli> a, lli x)
{
    rep(i, n)
    {
        a[i] *= x;
    }
    bit<lli> b(n + 20);
    map<lli, lli> loop;
    vector<pair<lli, lli>> d;
    rep(i, n)
    {
        d.push_back(make_pair(a[i], i));
        loop[a[i]]++;
    }
    sort(d.begin(), d.end());
    //a_jが中心でa_i,a_jとなるものを考える.
    lli ans = 0;
    rep(j, n)
    {
        //cout << "LOOP" << j << endl;
        lli num = d[j].first;
        //cout << num << endl;
        vector<lli> ins;
        rep(l, loop[num])
        {
            lli in = d[j].second;
            lli left = b.sum(0, in);
            lli right = b.sum(in, n + 10);
            ans += left * right;
            ins.push_back(in);
            j++;
        }
        j--;
        for (auto k : ins) {
            b.add(k, 1);
        }
    }
    // ここまででa_i == a_jも含めたものができた.
    return ans;
}
int main()
{
    cin >> n;
    vector<lli> a(n);
    rep(i, n) cin >> a[i];

    //座標圧縮
    vector<lli> b(a), left(n, 0), right(n, 0);
    sort(b.begin(), b.end());
    b.erase(unique(b.begin(), b.end()), b.end());
    rep(i, n)
    {
        a[i] = lower_bound(b.begin(), b.end(), a[i]) - b.begin();
        right[a[i]]++;
    }
    lli ans = solve(a, 1);
    ans += solve(a, -1);
    lli sum = 0;
    rep(i, n)
    {
    }
    rep(i, n)
    {
        sum -= left[a[i]] * right[a[i]];
        ans -= sum;
        left[a[i]]++;
        right[a[i]]--;
        sum += left[a[i]] * right[a[i]];
    }
    cout << ans << endl;
}
0