588D - Duff in Beach - CodeForces Solution


*2100

Please click on ads to support us..

Python Code:

from collections import Counter
mod = 10 ** 9 + 7
n, l, k = list(map(int, input().split()))
if k == 1:
    print(l % mod)
    exit()
a = list(map(int, input().split()))
if l < n:
    a = a[:l]
    n = len(a)
m = {v : i + 1 for i, v in enumerate(sorted(set(a)))}
b = [m[i] for i in a]
g = l // n
r = l % n
k = min(k, g + int(r > 0))
d = Counter(b)
dr = Counter(b[:r])
sd = sorted(d.items())

dp = [0] * (len(m) + 1)
dp[0] = 1
ret = 0
for kk in range(k):
    tmp = [0] * len(dp)
    psum = dp[0]
    for v, c in sd:
        psum = (psum + dp[v]) % mod
        tmp[v] = psum * c % mod
        rep = (g - kk) % mod * c % mod + dr.get(v, 0)
                ret = (ret + psum * rep % mod) % mod
    dp = tmp
print(ret)


Comments

Submit
0 Comments
More Questions

454A - Little Pony and Crystal Mine
2A - Winner
1622B - Berland Music
1139B - Chocolates
1371A - Magical Sticks
1253A - Single Push
706B - Interesting drink
1265A - Beautiful String
214A - System of Equations
287A - IQ Test
1108A - Two distinct points
1064A - Make a triangle
1245C - Constanze's Machine
1005A - Tanya and Stairways
1663F - In Every Generation
1108B - Divisors of Two Integers
1175A - From Hero to Zero
1141A - Game 23
1401B - Ternary Sequence
598A - Tricky Sum
519A - A and B and Chess
725B - Food on the Plane
154B - Colliders
127B - Canvas Frames
107B - Basketball Team
245A - System Administrator
698A - Vacations
1216B - Shooting
368B - Sereja and Suffixes
1665C - Tree Infection