【题目链接】
【思路要点】
- 分别考虑丢失1把、2把、3把斧子的方案数。
- 定义多项式\(A\),满足\(A_i=[i\ exist\ in\ input]\)。
- 丢失1把斧子的方案数显然就是\(A_i\)。
- 丢失2把斧子的方案数可以通过容斥原理计算,考虑先计算出\(B=A^2\)。
- 对于一种方案\((x,y)\),当\(x=y\),它会被计算1次,否则它会被计算2次,我们只希望统计\(x\ne y\)的方案恰好一次。
- 那么在\(B\)中先减去\(x=y\)的方案,然后将\(B\)整体除以2即可得到丢失2把斧子的方案数。