题目描述:
小爱正在参加一个闯关游戏。游戏的关卡分为两类,一类与文艺相关,一类与体育相关。每类游戏都有 n 关,每通一关,需要的时间是不一样的。每种类型的关卡,必须按照顺序通关,即必须先通过文艺类(或体育)的前一关,才能挑战文艺类(或体育)的下一关。
小爱挑战这些关卡的总时间有限,记为t,在挑战中,她可以自由地在两类关卡中切换。请问如何才能在有限的时间内通过尽量多的关卡呢?
输入格式:
第一行:两个整数 n 和 t;
第二行:n 个整数 a1,a2,…,an,表示文艺类关卡的通关时间;
第三行:n 个整数 b1,b2,…,bn,表示体育类关卡的通关时间。
输出格式:
单个整数:表示在时限内能够完成挑战的最大关数。
数据范围:
- 对于 40% 的数据,1 ≤ n ≤ 100;
- 对于 60% 的数据,1 ≤ n ≤ 1000;
- 对于 100% 的数据,1 ≤ n ≤ 100000,1 ≤ t ≤ 1,000,000,000,1 ≤ ai , bi ≤ 10000。
样例输入:
4 22 6 8 10 7 7 11 9 9
样例输出:
3
提示:
选择通关6,7,8
时间限制: 1000ms
空间限制: 256MB
代码实现:
这道题我刚刚看到时,还以为是一道贪心题,于是写了一段20分的代码:
#include<bits/stdc++.h>
using namespace std;
long long a[10001],b[10001],n,t1=1,t2=1,times,cnt;
int main(){
cin>>n>>times;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
for(int i=1;i<=n;i++){
scanf("%d",&b[i]);
}
while(1){
if(times<min(a[t1],b[t2]))break;
if(a[t1]<b[t2]){
times-=a[t1];
t1++;
}
else{
times-=b[t2];
t2++;
}
cnt++;
}
cout<<cnt;
return 0;
}
当然没有过,让我非常苦恼,这道题也被搁置了好久。
终于有一天,我开窍了!如果一个很大的数后面跟了几个特别小的数,这个贪心程序是检测不到的,这个时候得用动态规划(没学过,不知道怎么写),于是又搁置了……有一天,我又开窍了,终于知道为什么这道题挂了{前缀和、二分}的标签了!
以下是AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,t,a[1000005],b[10000005],tmp,ans;
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>t;
for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];//前缀和计算到这个关卡所需的总时间
for(int j=1;j<=n;j++)cin>>b[j],b[j]+=b[j-1];//同上
for(int i=1;i<=n;i++){
if(a[i]<=t){
tmp=lower_bound(b,b+1+n,t-a[i])-b-1;//二分查找,查找到a[i]时剩下的时间在b中能闯几关(二分代码也可以自己写)
ans=max(ans,tmp/*b中闯过的关卡*/+i/*a中闯过的关卡*/);//找出最大闯过的关卡数
}
else break;
}
cout<<ans;
return 0;
}
无注释版:
#include<bits/stdc++.h>
using namespace std;
int n,t,a[1000005],b[10000005],tmp,ans;
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>t;
for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];
for(int j=1;j<=n;j++)cin>>b[j],b[j]+=b[j-1];
for(int i=1;i<=n;i++){
if(a[i]<=t){
tmp=lower_bound(b,b+1+n,t-a[i])-b-1;
ans=max(ans,tmp+i);
}
else break;
}
cout<<ans;
return 0;
}
然后,数据更新了…………靠!
又开始改代码ing…………
我发现,把lower_bound改成upper_bound就可以过了
#include<bits/stdc++.h>
using namespace std;
int n,t,a[1000005],b[10000005],tmp,ans;
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>t;
for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];
for(int j=1;j<=n;j++)cin>>b[j],b[j]+=b[j-1];
for(int i=1;i<=n;i++){
if(a[i]<=t){
tmp=upper_bound(b,b+1+n,t-a[i])-b-1;
ans=max(ans,tmp+i);
}
else break;
}
cout<<ans;
return 0;
}