BZOJ 2023: [Usaco2005 Nov]Ant Counting 数蚂蚁——DP

BZOJ2023

我好弱啊、、只会刷水题啊。。  

这题我们首先考虑暴力的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)