CF1172C1.Nauuo and Pictures (easy version)

提高+/省选-

通过率:0%

时间限制:4.00s

内存限制:256MB

AC君温馨提醒

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

题目描述

The only difference between easy and hard versions is constraints.

Nauuo is a girl who loves random picture websites.

One day she made a random picture website by herself which includes nn pictures.

When Nauuo visits the website, she sees exactly one picture. The website does not display each picture with equal probability. The ii-th picture has a non-negative weight wiw_i, and the probability of the ii-th picture being displayed is wi∑j=1nwj\frac{w_i}{\sum_{j=1}^nw_j}. That is to say, the probability of a picture to be displayed is proportional to its weight.

However, Nauuo discovered that some pictures she does not like were displayed too often.

To solve this problem, she came up with a great idea: when she saw a picture she likes, she would add 11 to its weight; otherwise, she would subtract 11 from its weight.

Nauuo will visit the website mm times. She wants to know the expected weight of each picture after all the mm visits modulo 998244353998244353. Can you help her?

The expected weight of the ii-th picture can be denoted by qipi\frac {q_i} {p_i} where gcd⁡(pi,qi)=1\gcd(p_i,q_i)=1, you need to print an integer rir_i satisfying 0≤ri<9982443530\le r_i \lt 998244353 and ri⋅pi≡qi(mod998244353)r_i\cdot p_i\equiv q_i\pmod{998244353}. It can be proved that such rir_i exists and is unique.

简单版本与困难版本的唯一区别在于约束条件。

Nauuo 是一位喜爱随机图片网站的女孩。

某天,她自己创建了一个随机图片网站,其中包含 nn 张图片。

当 Nauuo 访问该网站时,她恰好看到一张图片。该网站并非以相等的概率显示每张图片。第 ii 张图片具有一个非负权重 wiw_i,其被显示的概率为 wi∑j=1nwj\frac{w_i}{\sum_{j=1}^nw_j}。换言之,一张图片被显示的概率与其权重成正比。

然而,Nauuo 发现,一些她不喜欢的图片出现得过于频繁。

为解决这一问题,她想出了一个绝妙的主意:当她看到一张喜欢的图片时,就将其权重加 11;否则,就将其权重减 11。

Nauuo 将访问该网站 mm 次。她希望知道经过全部 mm 次访问后,每张图片的期望权重(对 998244353998244353 取模)。你能帮她吗?

第 ii 张图片的期望权重可表示为 qipi\frac {q_i} {p_i},其中 gcd⁡(pi,qi)=1\gcd(p_i,q_i)=1。你需要输出一个整数 rir_i,满足 0≤ri<9982443530\le r_i \lt 998244353 且 ri⋅pi≡qi(mod998244353)r_i\cdot p_i\equiv q_i\pmod{998244353}。可以证明这样的 rir_i 存在且唯一。

输入格式

The first line contains two integers nn and mm (1≤n≤501\le n\le 50, 1≤m≤501\le m\le 50) — the number of pictures and the number of visits to the website.

The second line contains nn integers a1,a2,…,ana_1,a_2,\ldots,a_n (aia_i is either 00 or 11) — if ai=0a_i=0 , Nauuo does not like the ii-th picture; otherwise Nauuo likes the ii-th picture. It is guaranteed that there is at least one picture which Nauuo likes.

The third line contains nn integers w1,w2,…,wnw_1,w_2,\ldots,w_n (1≤wi≤501\le w_i\le50) — the initial weights of the pictures.

第一行包含两个整数 nn 和 mm(1≤n≤501\le n\le 50,1≤m≤501\le m\le 50)—— 分别表示图片数量和网站访问次数。

第二行包含 nn 个整数 a1,a2,…,ana_1,a_2,\ldots,a_n(每个 aia_i 为 00 或 11)—— 若 ai=0a_i=0,则 Nauuo 不喜欢第 ii 张图片;否则 Nauuo 喜欢第 ii 张图片。保证至少存在一张 Nauuo 喜欢的图片。

第三行包含 nn 个整数 w1,w2,…,wnw_1,w_2,\ldots,w_n(1≤wi≤501\le w_i\le50)—— 表示各张图片的初始权重。

输出格式

The output contains nn integers r1,r2,…,rnr_1,r_2,\ldots,r_n — the expected weights modulo 998244353998244353.

输出包含 nn 个整数 r1,r2,…,rnr_1,r_2,\ldots,r_n —— 即期望权重对 998244353998244353 取模的结果。

输入输出样例

  • 输入#1

    2 1
    0 1
    2 1

    输出#1

    332748119
    332748119
  • 输入#2

    1 2
    1
    1

    输出#2

    3
  • 输入#3

    3 3
    0 1 1
    4 3 5

    输出#3

    160955686
    185138929
    974061117

说明/提示

In the first example, if the only visit shows the first picture with a probability of 23\frac 2 3, the final weights are (1,1)(1,1); if the only visit shows the second picture with a probability of 13\frac1 3, the final weights are (2,2)(2,2).

So, both expected weights are 23⋅1+13⋅2=43\frac2 3\cdot 1+\frac 1 3\cdot 2=\frac4 3 .

Because 332748119⋅3≡4(mod998244353)332748119\cdot 3\equiv 4\pmod{998244353}, you need to print 332748119332748119 instead of 43\frac4 3 or 1.33333333331.3333333333.

In the second example, there is only one picture which Nauuo likes, so every time Nauuo visits the website, w1w_1 will be increased by 11.

So, the expected weight is 1+2=31+2=3.

Nauuo is very naughty so she didn't give you any hint of the third example.

在第一个例子中,若唯一的一次访问展示第一张图片(概率为 23\frac{2}{3}),则最终权重为 (1,1)(1,1);若唯一的一次访问展示第二张图片(概率为 13\frac{1}{3}),则最终权重为 (2,2)(2,2)。

因此,两个权重的期望值均为 23⋅1+13⋅2=43\frac{2}{3}\cdot 1+\frac{1}{3}\cdot 2=\frac{4}{3}。

由于 332748119⋅3≡4(mod998244353)332748119\cdot 3\equiv 4\pmod{998244353},你需要输出 332748119332748119,而非 43\frac{4}{3} 或 1.33333333331.3333333333。

在第二个例子中,只有一张 Nauuo 喜欢的图片,因此每次 Nauuo 访问网站时,w1w_1 都会增加 11。

因此,期望权重为 1+2=31+2=3。

Nauuo 非常调皮,因此她没有给你第三个例子的任何提示。

输入解题思路,AI测评打分。不知道怎么写?

首页