반전수
시간 제한2초메모리 제한1024 MB
값 i가 각각 a_i개씩 있을 때, 무작위로 고른 수열의 역전 수를 제곱한 값의 기댓값을 998244353으로 나눈 나머지로 구합니다.
문제
개의 , 개의 , , 개의 으로 이루어진 수열들을 생각하자. 이 수들로 이루어진 수열 중 하나를 균일한 확률로 무작위로 고를 때, 반전수의 제곱의 기댓값을 으로 나눈 나머지를 구하여라.
수열 의 반전수는 이고 를 만족하는 순서쌍 의 개수로 정의한다.
입력
첫 번째 줄에 수열에 있는 수의 최댓값 이 주어진다. ()
두 번째 줄에 각 수의 개수 가 공백을 사이에 두고 주어진다. (; )
출력
정답을 으로 나눈 나머지를 출력하여라.
힌트
반전수의 제곱의 기댓값은 유리수이다. 이 수를 서로소인 정수 , 로 라 할 때, 는 의 배수가 될 수 없음을 증명할 수 있다. 정답은 을 만족하는 음이 아닌 정수 중 가장 작은 수이다.
이 1개, 가 1개인 수열은 와 두 가지가 있고, 반전수는 각각 , 이다. 따라서 반전수의 제곱의 기댓값은 이고, 이므로 정답은 이다.