結果

問題 No.2338 Range AtCoder Query
ユーザー Forested
提出日時 2022-08-23 17:08:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 899 ms / 4,000 ms
コード長 2,859 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 82,848 KB
実行使用メモリ 194,896 KB
最終ジャッジ日時 2024-10-04 09:34:34
合計ジャッジ時間 22,739 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

# https://github.com/shakayami/ACL-for-python/blob/master/fenwicktree.py
class fenwick_tree():
    n=1
    data=[0 for i in range(n)]
    def __init__(self,N):
        self.n=N
        self.data=[0 for i in range(N)]
    def add(self,p,x):
        assert 0<=p<self.n,"0<=p<n,p={0},n={1}".format(p,self.n)
        p+=1
        while(p<=self.n):
            self.data[p-1]+=x
            p+=p& -p
    def sum(self,l,r):
        assert (0<=l and l<=r and r<=self.n),"0<=l<=r<=n,l={0},r={1},n={2}".format(l,r,self.n)
        return self.sum0(r)-self.sum0(l)
    def sum0(self,r):
        s=0
        while(r>0):
            s+=self.data[r-1]
            r-=r&-r
        return s

class OfflineRectangleAddPointGet:
    def __init__(self, n):
        self.n = n
        self.add = [[] for _ in range(n + 1)]
        self.sub = [[] for _ in range(n + 1)]
        self.q = 0
        self.queries = [[] for _ in range(n)]
    
    def add_rect(self, xl, xr, yl, yr):
        self.add[xl].append((yl, yr))
        self.sub[xr].append((yl, yr))
    
    def add_query(self, x, y):
        self.queries[x].append((y, self.q))
        self.q += 1
    
    def solve(self):
        ret = [0 for _ in range(self.q)]
        fw = fenwick_tree(self.n + 1)
        for x in range(self.n):
            for (l, r) in self.add[x]:
                fw.add(l, 1)
                fw.add(r, -1)
            for (l, r) in self.sub[x]:
                fw.add(l, -1)
                fw.add(r, 1)
            for (y, qi) in self.queries[x]:
                ret[qi] = fw.sum(0, y + 1)
        return ret

def main():
    import sys
    input = sys.stdin.readline
    
    n, m, q = map(int, input().split())
    p = [0 for _ in range(n)]
    is_ac = [False for _ in range(n)]
    for i in range(n):
        pi, si = input().split()
        p[i] = int(pi) - 1
        is_ac[i] = si == "AC"
    l = [0 for _ in range(q)]
    r = [0 for _ in range(q)]
    for i in range(q):
        li, ri = map(int, input().split())
        l[i] = li - 1
        r[i] = ri
    
    prev_ac = [-1 for _ in range(n)]
    next_ac = [n for _ in range(n)]
    pv = [-1 for _ in range(m)]
    for i in range(n):
        prev_ac[i] = pv[p[i]]
        if is_ac[i]:
            pv[p[i]] = i
    nt = [n for _ in range(m)]
    for i in reversed(range(n)):
        next_ac[i] = nt[p[i]]
        if is_ac[i]:
            nt[p[i]] = i
    
    ac = OfflineRectangleAddPointGet(n + 1)
    pena = OfflineRectangleAddPointGet(n + 1)
    for i in range(n):
        if is_ac[i]:
            ac.add_rect(prev_ac[i] + 1, i + 1, i + 1, n + 1)
        else:
            pena.add_rect(prev_ac[i] + 1, i + 1, next_ac[i] + 1, n + 1)
    for i in range(q):
        ac.add_query(l[i], r[i])
        pena.add_query(l[i], r[i])
    acc = ac.solve()
    penac = pena.solve()
    for i in range(q):
        print(acc[i], penac[i])

main()
0