我好弱啊、、只会刷水题啊。。
这题我们首先考虑暴力的n^3 dp,f[i][j]代表前i个种类的蚂蚁构成size为j的集合的方案数
转移枚举第i类蚂蚁选了几个,可以在n^3的复杂度内解决此题
不过 应该是T*n^2,。。。所以显然过不去
考虑优化
我们在n^3时的dp转移:f[i][j]=sigma(f[i-k][j-1])
那么显然i-k是一段区间的和
我们可以用前缀和优化
然后考虑优化空间,f[i][j]只与f[k][j-1]有关,所以我们可以使用滚存压掉i这一维
时间O(T*N),空间O(N)
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <string>
#include <map>
#include <cstring>
#include <ctime>
#include <vector>
#define inf 1e9
#define ll long long
#define For(i,j,k) for(int i=j;i<=k;i++)
#define Dow(i,j,k)