CF1575F.Finding Expected Value

普及/提高-

通过率:0%

AC君温馨提醒

该题目为【codeforces】题库的题目,您提交的代码将被提交至codeforces进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

Mr. Chanek opened a letter from his fellow, who is currently studying at Singanesia. Here is what it says.

Define an array bb ( 0bi<k0 \leq b_i < k ) with nn integers. While there exists a pair (i,j)(i, j) such that bibjb_i \ne b_j , do the following operation:

  • Randomly pick a number ii satisfying 0i<n0 \leq i < n . Note that each number ii has a probability of 1n\frac{1}{n} to be picked.
  • Randomly Pick a number jj satisfying 0j<k0 \leq j < k .
  • Change the value of bib_i to jj . It is possible for bib_i to be changed to the same value.

Denote f(b)f(b) as the expected number of operations done to bb until all elements of bb are equal.

You are given two integers nn and kk , and an array aa ( 1ai<k-1 \leq a_i < k ) of nn integers.

For every index ii with ai=1a_i = -1 , replace aia_i with a random number jj satisfying 0j<k0 \leq j < k . Let cc be the number of occurrences of 1-1 in aa . There are kck^c possibilites of aa after the replacement, each with equal probability of being the final array.

Find the expected value of f(a)f(a) modulo 109+710^9 + 7 .

Formally, let M=109+7M = 10^9 + 7 . It can be shown that the answer can be expressed as an irreducible fraction pq\frac{p}{q} , where pp and qq are integers and q≢0(modM)q \not \equiv 0 \pmod{M} . Output the integer equal to pq1modMp \cdot q^{-1} \bmod M . In other words, output such an integer xx that 0x<M0 \le x < M and xqp(modM)x \cdot q \equiv p \pmod{M} .

After reading the letter, Mr. Chanek gave the task to you. Solve it for the sake of their friendship!

输入格式

The first line contains two integers nn and kk ( 2n1052 \leq n \leq 10^5 , 2k1092 \leq k \leq 10^9 ).

The second line contains nn integers a1,a2,,ana_1, a_2, \ldots, a_n ( 1ai<k-1 \leq a_i < k ).

输出格式

Output an integer denoting the expected value of f(a)f(a) modulo 109+710^9 + 7 .

输入输出样例

  • 输入#1

    2 2
    0 1

    输出#1

    2
  • 输入#2

    2 2
    0 -1

    输出#2

    1
  • 输入#3

    3 3
    0 1 1

    输出#3

    12
  • 输入#4

    3 3
    -1 -1 -1

    输出#4

    11
  • 输入#5

    10 9
    -1 0 -1 1 1 2 2 3 3 3

    输出#5

    652419213
首页