生成函數
1.BZOJ3771
構造$f(x)=\sum x^{a[i]}$那麼選一個數的答案就是$f(x)$
選兩個數需要去掉選同一個數的情況,發現這部分竟然可以被表示爲$x^{2a[i]}$。那麼答案是$(f^{2}(x)-f(x^{2}))/2$
選三個數需要去掉兩種情況
1.一個數選三次,貢獻是$f(x^{3})$
2.兩個數一個一次一個兩次,$f^{2}(x)f(x^{2})$,但選一次的數是第幾次選不確定,因此要乘3,又由於這部分還包含了一個數選三次的情況,在計算結果時要加回來。
那麼答案是$(f^{3}(x)-3f^{2}(x)f(x^{2})+2f(x^{3}))/6$
三部分加起來就行了