abc360 E题解 (AtCoder Beginner Contest 360)

 

E

对于位置2~n，它们的概率是相等的。

n*n个(x,y)对。其中x可以等于y。

 

对于x/y，y的逆元rev(y)为mul(y,mod-2)。

加、减、乘、除都可以做。比如48/9和16/3的结果是一样的，48*rev(9)%mod = 16*rev(3)%mod。比如3*rev(2)%mod = (rev(2)+rev(2)+rev(2))%mod.

 

对于每次操作，有多少概率黑球在位置1：

a. 黑球上次在位置1，这次不进行修改

　　概率：(n-1)*(n-1)+1 / n^2

b. 黑球上次在其它地方，这次移动到位置1

　　概率：2 / n^2

举个例子，以n为3为例，

(x,y)总共有1 1, 1 2, 1 3, 2 1, 2 2, 2 3, 3 1, 3 2, 3 3

对于a，有2 2, 2 3, 3 2, 3 3, 1 1。共5个

对于b，对于数字2，有2 1, 1 2, 对于数字3，有3 1, 3 2。共2个

递推式为：

 

 

注意n*n-2*n+2，要先取模，否则它再乘上其它数，有可能大于Long Long范围。

 

#include <bits/stdc++.h>
using namespace std;
#define LL long long
#define ULL unsigned long long

const LL mod=998244353;

const double eps_1=1e-5;
const double eps_2=1e-10;

const int maxn=1e5+10;

LL a[maxn];

LL mul(LL x, LL y)
{
    LL z=1;
    while (y)
    {
        if (y&1)
            z=z*x%mod;
        x=x*x%mod;
        y>>=1;
    }
    return z;
}

LL rev(LL x)
{
    return mul(x,mod-2);
}

int main()
{
    LL n,k,i,rev_n,one,other;
    cin>>n>>k;
    a[0]=1;
    rev_n = rev(n);
    for (i=1;i<=k;i++)
    {
        //n*n-2*n+2 放在前面
        a[i] = ( (n*n-2*n+2) % mod *a[i-1] % mod + (1-a[i-1]+mod) * 2 % mod ) % mod
             * rev_n % mod * rev_n % mod;
    }
    one = a[k];
    other = (1-one+mod) * rev(n-1) % mod;
    cout << ( one + (n-1) * (2+n) % mod * rev(2) % mod * other % mod ) % mod;

    return 0;
}

 

 

F

 

 

G