AT_tupc2022_d Zeta Sum

题目描述

对于正整数 $n$,定义 $\zeta_n := \cos{\left(\dfrac{2\pi}{n}\right)} + \sqrt{-1}\sin{\left(\dfrac{2\pi}{n}\right)}$。 给定正整数 $N$,请计算 \[ \sum_{a=1}^{N}\sum_{b=1}^{N}\sum_{i=0}^{a-1}\sum_{j=0}^{b-1}(\zeta_a^i+\zeta_b^j)^N \] 已知,这个值是整数。由于答案可能非常大,请输出其对输入给定素数 $P$ 的余数。

输入格式

输入以如下格式从标准输入读入: > $N$ $P$

输出格式

输出该值对 $P$ 取余的结果。

说明/提示

### 样例解释 1 该式等于 $(1+1)^2+(1+1)^2+(1-1)^2+(1+1)^2+(-1+1)^2+(1+1)^2+(1-1)^2+(-1+1)^2+(-1-1)^2=20$。 ### 样例解释 3 请计算其对素数 $520232023$ 的余数。 ### 数据范围 - $1 \leq N \leq 2\times 10^5$ - $10^8 \leq P \leq 10^9$ - $P$ 为素数 - 输入均为整数 由 ChatGPT 5 翻译