結果

問題 No.121 傾向と対策：門松列（その２）
ユーザー 夜
提出日時 2021-02-25 23:19:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,562 ms / 5,000 ms
コード長 1,820 bytes
コンパイル時間 1,174 ms
コンパイル使用メモリ 109,784 KB
実行使用メモリ 140,020 KB
最終ジャッジ日時 2024-10-01 14:28:06
合計ジャッジ時間 8,567 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cmath>
#include <stdio.h>
#include <queue>
#include <deque>
#include <cstdio>
#include <set>
#include <map>
#include <bitset>
#include <stack>
#include <cctype>
using namespace std;
long long n;
vector<long long> BIT(1000010);
void add(long long i, long long x) {
    while (i <= n) {
        BIT[i] += x;
        i += i & -i;
    }
}
long long csum(long long i) {
    long long count = 0;
    while (i > 0) {
        count += BIT[i];
        i -= i & -i;
    }
    return count;
}
long long sum(long long l, long long r) {
    return csum(r) - csum(l - 1);
}
vector<pair<int, int>> vec;
int a[1000010];
map<long long,long long> m;
long long l[1000010] = {}, r[1000010] = {}, co[1000010] = {}, co1[1000010] = {};
long long la[1000010], ra[1000010], li[1000010], ri[1000010];
long long z = 0;
int main() {
    cin >> n;
    for (int i = 0; i < n; i++) {
        int a;
        cin >> a;
        vec.emplace_back(make_pair(a, i));
    }
    sort(vec.begin(), vec.end());
    int now = 1;
    for (int i = 0; i < n; i++) {
        if (i != 0 && vec[i].first != vec[i - 1].first)now++;
        a[vec[i].second] = now;
    }
    add(a[0], 1);
    for (int i = 1; i < n; i++) {
        la[i] = sum(a[i] + 1, n);
        li[i] = sum(1, a[i] - 1);
        add(a[i], 1);
    }
    for (int i = 1; i <= n; i++) {
        add(i, -BIT[i]);
    }
    add(a[n - 1], 1);
    for (int i = n-2; i >= 0; i--) {
        ra[i] = sum(a[i] + 1, n);
        ri[i] = sum(1, a[i] - 1);
        add(a[i], 1);
    }
    for (int i = 0; i < n; i++) {
        l[i + 1] = -1;
    }
    long long ans = 0;
    for (int i = 0; i < n; i++) {
        ans += li[i] * ri[i];
        ans += la[i] * ra[i];
    }
    for (int i = n - 1; i >= 0; i--) {
        m[a[i]]++;
        co1[i] = m[a[i]];
    }
    for (int i = 0; i < n; i++) {
        if (l[a[i]] != -1) {
            ans -= (i - l[a[i]] - 1) * co[a[i]] * co1[i];
        }
        co[a[i]]++;
        l[a[i]] = i;
    }
    cout << ans << endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0