【BZOJ3771】Triple

【题目链接】

• 点击打开链接
  

【思路要点】

• 分别考虑丢失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把斧子的方案数。