题目

试题2:玩具(toy)

源代码:toy.cpp

输入文件:toy.in

输出文件:toy.out

时间限制:1s

空间限制:256MB

题目描述

商店正在出售小C最喜欢的系列玩具,在接下来的n周中,每周会出售其中的一款,同一款玩具不会重复出现。

由于是小C最喜欢的系列,他希望尽可能多地购买这些玩具,但是同一款玩具小C只会购买一个。同时,小C的预算只有m元,因此他无法将每一款都纳入囊中。此外,小C不能连续两周都购买玩具,否则他会陷入愧疚。现在小C想知道,他最多可以买多少款不同的玩具呢?

输入说明

输入文件共2行;

第一行两个正整数n和m,中间用一个空格隔开;

第二行共n个正整数,第i个正整数表示第i周出售的玩具的价格。

输出说明

输出文件只有一行,包含一个整数,表示小C最多能买多少款不同的玩具。

样例输入

3 8

4 4 5

样例输出

1

数据范围

对于30%的数据,n≤10;

对于60%的数据,n≤100,m≤1000;

对于100%的数据,n≤1000,m≤1000000,单个玩具的价格≤1000。

分析

首先,根据“同一款玩具小C只会购买一个”我们得知大体的做题方向是01背包的DP

所以我们用f[i]记录花i块钱最多能购买到的玩具数量

  但是! 题目中说“不能连续两周都购买玩具”,所以我们可以打标记避免连续两周都购买玩具。

粗略的分析完这些,我们就可以写代码了!

可惜的是你会发现你无法开下一个n*m的数组去打标记(因为n≤1000,m≤1000000)。

如何缩小一个数组的空间呢?

要么是状态压缩,要么是滚动数组。

而现在的情况用滚动数组会更简单一点。

代码

#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
using namespace std;
int n,m,a[11001];
int f[1100001];//f[i]记录花i块钱最多能购买到的玩具数量
bool flag[1100001][3];//f[i][j]记录f[i]在上周是否购买了玩具(j是用来滚动数组的)
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
int o1=1,o2=2;
for(int i=1;i<=n;i++)
{
swap(o1,o2);//滚动一次flag数组
for(int j=0;j<=m;j++) flag[j][o2]=0;
for(int j=m;j>=a[i];j--)
{
if(f[j-a[i]]+1>f[j] && !flag[j-a[i]][o1])
{
flag[j][o2]=1;
f[j]=f[j-a[i]]+1;
}
}
}
cout<<f[m];
return 0;
}

  

05-27 22:24