結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー rpy3cpp
提出日時 2015-07-13 23:08:48
言語 PyPy2
(7.3.15)
結果
AC  
実行時間 2,756 ms / 5,000 ms
コード長 1,824 bytes
コンパイル時間 124 ms
コンパイル使用メモリ 77,056 KB
実行使用メモリ 323,956 KB
最終ジャッジ日時 2024-06-28 18:37:01
合計ジャッジ時間 8,690 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

def compress(As):
    '''とびとびの値のリストを連続する値のリストに変換する。変換後の最大値も返す。
    '''
    uniqA = list(set(As))
    uniqA.sort()
    val2idx = {a: i for i, a in enumerate(uniqA)}
    return [val2idx[a] for a in As], len(uniqA) - 1

def add_count_f(val, data, size):
    '''セグメントツリーへの要素 val を追加し、
    val 未満の要素の個数と、val 超の要素の個数を返す。
    '''
    pos = size + val
    count = 0
    while pos:
        if pos & 1:
            count += data[pos-1]
        data[pos] += 1
        pos >>= 1
    return count, data[1] - count - data[size + val]

def count_aba(As, maxA):
    '''数値のリストAs の部分列で、a, b, a となるパターンの個数を返す。
    ある数値 a が、p0, p1, p2, ... pn-1 にあるとすると、
    sum((p[i] - p[i-1] - 1) * i * (n-i) for i in range(1, n)) 個だけ、abaとなるパターンがある。
    '''
    posList = [[] for i in range(maxA+1)]
    for i, a in enumerate(As):
        posList[a].append(i)
    count = 0
    for poss in posList:
        n = len(poss)
        for i in xrange(1, n):
            count += (poss[i] - poss[i-1] - 1) * i * (n-i)
    return count

N = int(raw_input())
As = list(map(int, raw_input().split()))
As, maxA = compress(As)
size = 1 << (len(bin(N)) - 2)
segL = [0] * (size * 2)
segR = [0] * (size * 2)
belowL = [0] * N
aboveL = [0] * N

for i, a in enumerate(As):
    below, above = add_count_f(a, segL, size)
    belowL[i] = below
    aboveL[i] = above
result = 0
for i, a in enumerate(As[::-1], 1):
    below, above = add_count_f(a, segR, size)
    result += below * belowL[-i] + above * aboveL[-i]

# 左右が同じ値の疑似門松列の個数を差し引く
result -= count_aba(As, maxA)
print(result)
0