P2946 [USACO09MAR]Cow Frisbee Team S

https://www.luogu.com.cn/problem/P2946

题解: 

一道需要进行许多处理的01背包问题

首先由于物品数量有2000多,N*R最高将达到惊人的2×10^8,N*N*R,空间不够用

第一时间想到优化成一维滚动数组

如下

#include<stdio.h>
#include<iostream>
#include<cstdlib>
#include<string.h>
#include<algorithm>
using namespace std;
int n,f;
const int N=2010;
const int MOD=1e8;
const int R=3e7;
int d[R];//前i头牛,体积恰好为N
int v[N];
int V;
int main()
{
	//freopen("uva.txt","r",stdin);
	d[0]=1;
	scanf("%d%d",&n,&f);
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&v[i]);
		V+=v[i];
	}
	int res=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=V;j>=v[i];j--)
		{
			d[j]=d[j]+d[j-v[i]];
			d[j]%=MOD;
		}
	}
	for(int i=f;i<=V;i+=f)
	{
		res+=d[i];
		res%=MOD;
	}
	cout<<res;
	return 0;
}

一发交上去,只能拿80分,优化成1维滚动数组仍然不能拿到满分.

其实显然,即使优化成1维,d的第二维表示的体积(能力值)依然不会被优化到,N*R也是无法接受的

必须要进行压缩.

赛后看了一下题解,是将体积也进行取模缩小范围,存储v[i]时,v[i]%f

由d[j-v[i]]改成d[(j-v[i]+f)%f] 

最初的状态转移方程为d[i][j]=d[i-1][j]+d[i-1][j-v[i]] 

而有了取模之后

由于原v[i]的大小关系发生了变化(如10模5变成0,2模5模成2,  10反而在了2前面),这样会导致第i层和第i-1层混淆(也就是一些点提前更新了),而我们01背包每次都要用i-1层的数据.

所以只能改成二维,不过此时dp数组第二维最大大小设为F=1000即可.

原来的d[0][0]=1也改成d[i][v[i]]=1  (v[i]事先已对 f 取模)  就是说取v[i]的方法数为1

后面也需要把d[i][j]=改成d[i][j]+=

AC代码

#include<stdio.h>
#include<iostream>
#include<cstdlib>
#include<string.h>
#include<algorithm>
using namespace std;
int n,f;
const int N=2010;
const int MOD=1e8;
const int R=3e7;
const int F=1010;
int d[N][F];//前i头牛,体积恰好为N
int v[N];
int main()
{
	//freopen("uva.txt","r",stdin);
	scanf("%d%d",&n,&f);
	//d[0][0]=1;d[0][0]=1换到了for循环里面的
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&v[i]);
		v[i]%=f;	
	}
	
	for(int i=1;i<=n;i++)
	{
		d[i][v[i]]=1;
		for(int j=0;j<f;j++)
		{
			d[i][j]+=d[i-1][j]+d[i-1][(j-v[i]+f)%f];
			d[i][j]%=MOD;
		}
	}
	cout<<d[n][0];
	return 0;
}