题意:
思路:
注意到这个题是求若干个数的组合数,
(
a
,
b
)
,
(
b
,
a
)
(a,b),(b,a)
(a,b),(b,a)视为一种方案,所以我们考虑生成一个普通型生成函数。
考虑到每个数只能选一次,但是如果我们生成函数相乘的话是不能控制每个数选多少次的,可以简单脑补一下两个循环相乘,得到的结果。由于选的物品最多有三个,所以我们考虑分开讨论。
我们构造函数
a
(
x
)
a(x)
a(x)表示每个物品选一次,
b
(
x
)
b(x)
b(x)表示每个物品选两次,
c
(
x
)
c(x)
c(x)表示每个物品选三次。
(
1
)
(1)
(1)对于只选一个物品,答案直接为
a
(
x
)
a(x)
a(x)即可。
(
2
)
(2)
(2)对于选了两个物品的情况,我们如果直接计算
a
2
(
x
)
a^2(x)
a2(x)的话会发现有重复的部分,这个重复的部分就是每个数选了两次,所以要减去
b
(
x
)
b(x)
b(x),由于其组合有
2
2
2种,答案即为
a
2
(
x
)
−
b
(
x
)
2
\frac{a^2(x)-b(x)}{2}
2a2(x)−b(x)。
(
3
)
(3)
(3)对于选了三个物品的情况,直接计算
a
3
(
x
)
a^3(x)
a3(x)也是有很多重复的,考虑从三个中选两个相同的位置,这个时候方案数即为
(
3
2
)
∗
a
(
x
)
∗
b
(
x
)
\binom{3}{2}*a(x)*b(x)
(23)∗a(x)∗b(x),再考虑从三个中选三个相同的位置,由于刚才已经减去了
3
3
3倍的了,所以需要加上
2
∗
c
(
x
)
2*c(x)
2∗c(x)。由于各自的组合有
6
6
6种,所以最终答案为
a
3
(
x
)
−
3
∗
a
(
x
)
∗
b
(
x
)
+
2
∗
c
(
x
)
6
\frac{a^3(x)-3*a(x)*b(x)+2*c(x)}{6}
6a3(x)−3∗a(x)∗b(x)+2∗c(x)。
直接
F
F
T
FFT
FFT卷一下求答案即可。
//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#include<random>
#include<cassert>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid ((tr[u].l+tr[u].r)>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;
//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
const int N=10000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6,PI=acos(-1);
int n,m;
int A[N],B[N],C[N];
int rev[N];
int bit,limit;
struct Complex {
double x,y;
Complex operator + (const Complex& t) const { return {x+t.x,y+t.y}; }
Complex operator - (const Complex& t) const { return {x-t.x,y-t.y}; }
Complex operator * (const Complex& t) const { return {x*t.x-y*t.y,x*t.y+y*t.x}; }
}a[N],b[N],c[N],ans[N];
void fft(Complex a[],int inv) {
for(int i=0;i<limit;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int mid=1;mid<limit;mid<<=1) {
Complex w1=Complex({cos(PI/mid),inv*sin(PI/mid)});
for(int i=0;i<limit;i+=mid*2) {
Complex wk=Complex({1,0});
for(int j=0;j<mid;j++,wk=wk*w1) {
Complex x=a[i+j],y=wk*a[i+j+mid];
a[i+j]=x+y; a[i+j+mid]=x-y;
}
}
}
}
int main()
{
// ios::sync_with_stdio(false);
// cin.tie(0);
cin>>n;
int mx=0;
for(int i=1;i<=n;i++) {
int x; scanf("%d",&x);
a[x].x++; b[x*2].x++; c[x*3].x++;
mx=max(mx,x*3);
}
while((1<<bit)<=mx) bit++;
limit=1<<bit;
for(int i=0;i<limit;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
fft(a,1); fft(b,1); fft(c,1);
for(int i=0;i<limit;i++) {
Complex x={3,0},y={2,0},z={1.0/6,0},h={1.0/2,0};
ans[i]=ans[i]+(a[i]*a[i]*a[i]-x*a[i]*b[i]+y*c[i])*z;
ans[i]=ans[i]+(a[i]*a[i]-b[i])*h;
ans[i]=ans[i]+a[i];
}
fft(ans,-1);
for(int i=0;i<limit;i++) {
int val=(int)(ans[i].x/limit+0.5);
if(val) printf("%d %d\n",i,val);
}
return 0;
}