結果

問題 No.404 部分門松列
ユーザー bal4u
提出日時 2019-08-23 17:37:49
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 176 ms / 2,000 ms
コード長 2,971 bytes
コンパイル時間 356 ms
コンパイル使用メモリ 33,536 KB
実行使用メモリ 8,856 KB
最終ジャッジ日時 2024-10-13 14:32:08
合計ジャッジ時間 5,980 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:10:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
   10 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:18:24: note: in expansion of macro 'gc'
   18 |         int n = 0, c = gc();
      |                        ^~
main.c: In function 'out':
main.c:11:15: warning: implicit declaration of function 'putchar_unlocked' [-Wimplicit-function-declaration]
   11 | #define pc(c) putchar_unlocked(c)
      |               ^~~~~~~~~~~~~~~~
main.c:26:17: note: in expansion of macro 'pc'
   26 |         if (!n) pc('0');
      |                 ^~

ソースコード

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

// yukicoder: No.404 部分門松列
// 2019.8.23 bal4u
#include <stdio.h>
#include <stdlib.h>
typedef long long ll;
#if 1
#define gc() getchar_unlocked()
#define pc(c) putchar_unlocked(c)
#else
#define gc() getchar()
#define pc(c) putchar(c)
#endif
int in() { // 整数の入力（負数対応）
    int n = 0, c = gc();
    do n = 10*n + (c & 0xf), c = gc(); while (c >= '0');
    return n;
}
void out(ll n) { // 非負整数の表示（出力）
    int i; char b[30];
    if (!n) pc('0');
    else {
        i = 0; while (n) b[i++] = n % 10 + '0', n /= 10;
        while (i--) pc(b[i]);
    }
}
#define MAX 200005
//// bit木 1-indexedに注意
int bitf[MAX], bits[MAX], bitn;
void add(int *bit, int i) { while (i <= bitn) bit[i]++, i += i & -i; }
void sub(int *bit, int i) { while (i <= bitn) bit[i]--, i += i & -i; }
int sum(int *bit, int i) { int s = 0; while (i > 0) s += bit[i], i -= i & -i; return s; }
//// 本問題関連
#define MAX 200005
typedef struct { int a, id; } T;
T t[MAX]; int N;
int A[MAX];
int vf[MAX], vs[MAX];
ll  cnt[MAX], cum[MAX];
// a[i]>=key という条件を満たす最小のi
int lower_bound(int x) {
    int m, l = 1, r = N+1;
    while (l+1 < r) {
        m = (l+r) >> 1;
        if (t[m].a < x) l = m; else r = m;
    }
    return t[l].a < x? r: l;
}
// a[i]>key という条件を満たす最小のi
int upper_bound(int x) {
    int m, l = 1, r = N+1;
    while (l+1 < r) {
        m = (l+r) >> 1;
        if (t[m].a <= x) l = m; else r = m;
    }
    return t[l].a <= x? r: l;
}
int cmp(const void *u, const void *v) { return ((T *)u)->a - ((T *)v)->a; }
void compact(int *a, int n) { // データ圧縮。圧縮後のデータ順序は圧縮前と同様
    int i, v;
    for (i = 1; i <= n; i++) t[i].a = a[i], t[i].id = i;
    qsort(t+1, n, sizeof(T), cmp);
    v = 1; a[t[1].id] = v;
    for (i = 2; i <= n; i++) {
        if (t[i].a != t[i-1].a) v++;
        a[t[i].id] = v;
    }
}
int main()
{
    int i, x, Q; ll same;
    
    N = in(); for (i = 1; i <= N; i++) A[i] = in();
    compact(A, N);  // 1-indexed
    bitn = N+1;  // bit木の初期化
    
    // すべてのデータをとりあえずすべて後半のbit木、頻度表に登録
    for (i = 1; i <= N; i++) {
        add(bits, A[i]);  // 後半のデータをbit木に追加
        vs[A[i]]++;       // 後半のデータを頻度表にも追加
    }
    
    same = 0; // 前後半における同じ値のデータの累積和
    x = A[1], add(bitf, x), sub(bits, x);
    vf[x]++, vs[x]--, same += vs[x];
    for (i = 2; i < N; i++) {
        x = A[i];
        sub(bits, x), vs[x]--, same -= vf[x];
        int sf = sum(bitf, x-1), ss = sum(bits, x-1);
        cnt[x] += (ll)sf*ss + (ll)(i-1-sf-vf[x])*(N-i-ss-vs[x]) - (same-vf[x]*vs[x]);
        add(bitf, x), same += vs[x], vf[x]++;
    }
    cum[0] = 0; for (i = 1; i <= N; i++) cum[i] = cum[i-1] + cnt[i];
    Q = in(); while (Q--) {
        int L = in(), H = in();
        if ((L = lower_bound(L)) > N || L < 1 ||
            (H = upper_bound(H)) <= 1 || H == L) pc('0');
        else {
            ll r = cum[A[t[H-1].id]] - cum[A[t[L].id]-1];
            if (r <= 0) pc('0');
            else out(r);
        }
        pc('\n');
    }
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0