結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

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

###セグメントツリー###
#####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()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0