結果

問題 No.1233 割り切れない気持ち
ユーザー Shinya Fujita
提出日時 2024-10-06 02:03:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,086 ms / 3,153 ms
コード長 2,910 bytes
コンパイル時間 359 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 125,824 KB
最終ジャッジ日時 2024-10-06 02:03:27
合計ジャッジ時間 15,437 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

class SegmentTree:
    def __init__(self, a):
        self.padding = 0
        self.n = len(a)
        self.N = 2 ** (self.n-1).bit_length()
        self.seg_data = [self.padding]*(self.N-1) + a + [self.padding]*(self.N-self.n)
        for i in range(2*self.N-2, 0, -2):
            self.seg_data[(i-1)//2] = self.seg_data[i] + self.seg_data[i-1]
    
    def __len__(self):
        return self.n
    
    def __getitem__(self, i):
        return self.seg_data[self.N-1+i]
    
    def __setitem__(self, i, x):
        idx = self.N - 1 + i
        self.seg_data[idx] = x
        while idx:
            idx = (idx-1) // 2
            self.seg_data[idx] = self.seg_data[2*idx+1] + self.seg_data[2*idx+2]

    def query(self, i, j):
        # [i, j)
        if i == j:
            return 0
        else:
            idx1 = self.N - 1 + i
            idx2 = self.N - 2 + j # 閉区間にする
            result = self.padding
            while idx1 < idx2 + 1:
                if idx1&1 == 0: # idx1が偶数
                    result = result + self.seg_data[idx1]
                if idx2&1 == 1: # idx2が奇数
                    result = result + self.seg_data[idx2]
                    idx2 -= 1
                idx1 //= 2
                idx2 = (idx2 - 1)//2
            return result
    
    def kth_left_idx(self, fr, k):
        if self.query(0, fr+1) < k:
            return -1
        remain = k
        now = fr + self.N - 1
        while self.seg_data[now] < remain:
            if now % 2:
                remain -= self.seg_data[now]
                now -= 1
            else:
                now = (now - 1) // 2
        
        while now < self.N - 1:
            nl = 2*now + 1
            nr = nl + 1
            if self.seg_data[nr] < remain:
                remain -= self.seg_data[nr]
                now = nl
            else:
                now = nr
        
        return now - (self.N - 1)

    def kth_right_idx(self, fr, k):
        if self.query(fr, self.n) < k:
            return -1
        remain = k
        now = fr + self.N - 1
        while self.seg_data[now] < remain:
            if now % 2 == 0:
                remain -= self.seg_data[now]
                now += 1
            else:
                now //= 2
        
        while now < self.N - 1:
            nl = 2*now + 1
            nr = nl + 1
            if self.seg_data[nl] < remain:
                remain -= self.seg_data[nl]
                now = nr
            else:
                now = nl
        
        return now - (self.N - 1)


from collections import defaultdict


N = int(input())
A = list(map(int, input().split()))
S = sum(A)
maxA = max(A)
cnt = defaultdict(int)


st = SegmentTree([0] * (maxA + 1))
for a in A:
    cnt[a] += 1
    st[a] += 1

ans = S * N
for a, n in cnt.items():
    left = a
    while left <= maxA:
        ans -= a*n*st.query(left, maxA+1)
        left += a

print(ans)
0