結果
| 問題 |
No.742 にゃんにゃんにゃん 猫の挨拶
|
| コンテスト | |
| ユーザー |
 McGregorsh
|
| 提出日時 | 2022-11-10 13:38:07 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 208 ms / 2,500 ms |
| コード長 | 9,974 bytes |
| コンパイル時間 | 479 ms |
| コンパイル使用メモリ | 82,588 KB |
| 実行使用メモリ | 90,984 KB |
| 最終ジャッジ日時 | 2024-07-23 16:04:16 |
| 合計ジャッジ時間 | 4,033 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge4 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| other | AC * 16 |
ソースコード
###セグメントツリー###
#####segfunc#####
def segfunc(x, y):
return x + y
# 最小値 min(x, y)
# 最大値 max(x, y)
# 区間和 x + y
# 区間積 x * y
# 最大公約数 math.gcd(x, y)
# 排他的論理和 x ^ y
#################
#####ide_ele#####
ide_ele = 0
# 最小値 float('inf')
# 最大値 -float('inf')
# 区間和 0
# 区間積 1
# 最大公約数 0
# 排他的論理和 0
#################
class SegTree:
"""
init(init_val, ide_ele): 配列init_valで初期化 O(N)
update(k, x): k番目の値をxに更新 O(logN)
query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)
"""
def __init__(self, init_val, segfunc, ide_ele):
"""
init_val: 配列の初期値
segfunc: 区間にしたい操作
ide_ele: 単位元
n: 要素数
num: n以上の最小の2のべき乗
tree: セグメント木(1-index)
"""
n = len(init_val)
self.segfunc = segfunc
self.ide_ele = ide_ele
self.num = 1 << (n - 1).bit_length()
self.tree = [ide_ele] * 2 * self.num
# 配列の値を葉にセット
for i in range(n):
self.tree[self.num + i] = init_val[i]
# 構築していく
for i in range(self.num - 1, 0, -1):
self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1])
def update(self, k, x):
"""
k番目の値をxに更新
k: index(0-index)
x: update value
"""
k += self.num
self.tree[k] = x
while k > 1:
self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1])
k >>= 1
def query(self, l, r):
"""
[l, r)のsegfuncしたものを得る
l: index(0-index)
r: index(0-index)
"""
res = self.ide_ele
l += self.num
r += self.num
while l < r:
if l & 1:
res = self.segfunc(res, self.tree[l])
l += 1
if r & 1:
res = self.segfunc(res, self.tree[r - 1])
l >>= 1
r >>= 1
return res
###UnionFind###
from collections import defaultdict
class UnionFind():
"""
Union Find木クラス
Attributes
--------------------
n : int
要素数
root : list
木の要素数
0未満であればそのノードが根であり、添字の値が要素数
rank : list
木の深さ
"""
def __init__(self, n):
"""
Parameters
---------------------
n : int
要素数
"""
self.n = n
self.root = [-1]*(n+1)
self.rank = [0]*(n+1)
def find(self, x):
"""
ノードxの根を見つける
Parameters
---------------------
x : int
見つけるノード
Returns
---------------------
root : int
根のノード
"""
if(self.root[x] < 0):
return x
else:
self.root[x] = self.find(self.root[x])
return self.root[x]
def unite(self, x, y):
"""
木の併合
Parameters
---------------------
x : int
併合したノード
y : int
併合したノード
"""
x = self.find(x)
y = self.find(y)
if(x == y):
return
elif(self.rank[x] > self.rank[y]):
self.root[x] += self.root[y]
self.root[y] = x
else:
self.root[y] += self.root[x]
self.root[x] = y
if(self.rank[x] == self.rank[y]):
self.rank[y] += 1
def same(self, x, y):
"""
同じグループに属するか判定
Parameters
---------------------
x : int
判定したノード
y : int
判定したノード
Returns
---------------------
ans : bool
同じグループに属しているか
"""
return self.find(x) == self.find(y)
def size(self, x):
"""
木のサイズを計算
Parameters
---------------------
x : int
計算したい木のノード
Returns
---------------------
size : int
木のサイズ
"""
return -self.root[self.find(x)]
def roots(self):
"""
根のノードを取得
Returns
---------------------
roots : list
根のノード
"""
return [i for i, x in enumerate(self.root) if x < 0]
def group_size(self):
"""
グループ数を取得
Returns
---------------------
size : int
グループ数
"""
return len(self.roots()) - 1
def group_members(self):
"""
全てのグループごとのノードを取得
Returns
---------------------
group_members : defaultdict
根をキーとしたノードのリスト
"""
group_members = defaultdict(list)
for member in range(self.n):
group_members[self.find(member)].append(member)
return group_members
###素因数分解###
def prime_factorize(n: int) -> list:
return_list = []
while n % 2 == 0:
return_list.append(2)
n //= 2
f = 3
while f * f <= n:
if n % f == 0:
return_list.append(f)
n //= f
else:
f += 2
if n != 1:
return_list.append(n)
return return_list
###ある数が素数かどうかの判定###
def is_prime(n):
if n < 2:
return False
i = 2
while i * i <= n:
if n % i == 0:
return False
i += 1
return True
###N以下の素数列挙###
import math
def sieve_of_eratosthenes(n):
prime = [True for i in range(n+1)]
prime[0] = False
prime[1] = False
sqrt_n = math.ceil(math.sqrt(n))
for i in range(2, sqrt_n+1):
if prime[i]:
for j in range(2*i, n+1, i):
prime[j] = False
return prime
###N以上K以下の素数列挙###
import math
def segment_sieve(a, b):
sqrt_b = math.ceil(math.sqrt(b))
prime_small = [True for i in range(sqrt_b)]
prime = [True for i in range(b-a+1)]
for i in range(2, sqrt_b):
if prime_small[i]:
for j in range(2*i, sqrt_b, i):
prime_small[j] = False
for j in range((a+i-1)//i*i, b+1, i):
#print('j: ', j)
prime[j-a] = False
return prime
###n進数から10進数変換###
def base_10(num_n,n):
num_10 = 0
for s in str(num_n):
num_10 *= n
num_10 += int(s)
return num_10
###10進数からn進数変換###
def base_n(num_10,n):
str_n = ''
while num_10:
if num_10%n>=10:
return -1
str_n += str(num_10%n)
num_10 //= n
return int(str_n[::-1])
###複数の数の最大公約数、最小公倍数###
from functools import reduce
# 最大公約数
def gcd_list(num_list: list) -> int:
return reduce(gcd, num_list)
# 最小公倍数
def lcm_base(x: int, y: int) -> int:
return (x * y) // gcd(x, y)
def lcm_list(num_list: list):
return reduce(lcm_base, num_list, 1)
###約数列挙###
def make_divisors(n):
lower_divisors, upper_divisors = [], []
i = 1
while i * i <= n:
if n % i == 0:
lower_divisors.append(i)
if i != n // i:
upper_divisors.append(n//i)
i += 1
return lower_divisors + upper_divisors[::-1]
###順列###
def nPr(n, r):
npr = 1
for i in range(n, n-r, -1):
npr *= i
return npr
###組合せ###
def nCr(n, r):
factr = 1
r = min(r, n - r)
for i in range(r, 1, -1):
factr *= i
return nPr(n, r)/factr
###組合せMOD###
def comb(n,k):
nCk = 1
MOD = 10**9+7
for i in range(n-k+1, n+1):
nCk *= i
nCk %= MOD
for i in range(1,k+1):
nCk *= pow(i,MOD-2,MOD)
nCk %= MOD
return nCk
import sys, re
from fractions import Fraction
from math import ceil, floor, sqrt, pi, factorial, gcd
from copy import deepcopy
from collections import Counter, deque, defaultdict
from heapq import heapify, heappop, heappush
from itertools import accumulate, product, combinations, combinations_with_replacement, permutations
from bisect import bisect, bisect_left, bisect_right
from functools import reduce
from decimal import Decimal, getcontext, ROUND_HALF_UP
def i_input(): return int(input())
def i_map(): return map(int, input().split())
def i_list(): return list(i_map())
def i_row(N): return [i_input() for _ in range(N)]
def i_row_list(N): return [i_list() for _ in range(N)]
def s_input(): return input()
def s_map(): return input().split()
def s_list(): return list(s_map())
def s_row(N): return [s_input for _ in range(N)]
def s_row_str(N): return [s_list() for _ in range(N)]
def s_row_list(N): return [list(s_input()) for _ in range(N)]
def lcm(a, b): return a * b // gcd(a, b)
def get_distance(x1, y1, x2, y2):
d = sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
return d
def rotate(table):
n_fild = []
for x in zip(*table[::-1]):
n_fild.append(x)
return n_fild
sys.setrecursionlimit(10 ** 7)
INF = float('inf')
MOD = 10 ** 9 + 7
MOD2 = 998244353
###関数コピーしたか？###
def main():
N = int(input())
lines = [int(input()) for i in range(N)]
nums = [1] * (N + 1)
nums[0] = 0
seg = SegTree(nums, segfunc, ide_ele)
ans = 0
for i in range(N):
point = lines[i]
seg.update(point, 0)
score = seg.query(0, point)
ans += score
print(ans)
if __name__ == '__main__':
main()