题目只给了500ms,注意超时问题,一开始的几发都超时了,后来想到了预处理,从后往前推即可,为了防止t的大小可能有问题,所以进行了排序,还有人用二分做的,比较犀利先贴一个我的思路
#include<iostream>
#include<cstdio>
#include<list>
#include<algorithm>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<cmath>
#include<memory.h>
#include<set>
#define ll long long
#define eps 1e-7
#define inf 0xfffffff
const ll INF = 1ll<<61;
using namespace std;
//vector<pair<int,int> > G;
//typedef pair<int,int > P;
//vector<pair<int,int> > ::iterator iter;
//
//map<ll,int >mp;
//map<ll,int >::iterator p;
//
int n,m,k;
int w[100000 + 5];
int t[100000 + 5];
int sum[100000 * 2];//当前往前面第一次超过k的石头和
int step[100000 * 2];//当前往前需要回头几步
void clear() {
memset(w,0,sizeof(w));
memset(t,0,sizeof(t));
memset(sum,0,sizeof(sum));
memset(step,0,sizeof(step));
}
void init() {
sum[n] = w[n];
int tempn = n - 1;
int cnt = 1;
while(sum[n] <= k && tempn >=1) {//最后一个要单独拎出来
cnt++;
sum[n] += w[tempn--];
}
step[n] = cnt;
int tmpn = n - 1;
while(tmpn) {
sum[tmpn] = sum[tmpn + 1] - w[tmpn + 1];
cnt--;
while(sum[tmpn] <=k &&tempn >= 1) {
cnt++;
sum[tmpn] += w[tempn--];
}
step[tmpn] = cnt;
if(tempn <= 0) {
for(int i=tmpn;i>0;i--)//前面和已经无法超过k,所以退到直至第一次为止
step[i] = i;
break;
}
tmpn--;
}
}
int main() {
while(scanf("%d %d %d",&n,&m,&k) == 3) {
clear();
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
for(int i=1;i<=m;i++)
scanf("%d",&t[i]);
sort(t + 1,t + m + 1);
init();
int ans = 0;
int mark = 0;
for(int i=1;i<=m;i++) {
int cha = (t[i] - t[i-1]);
mark += cha;
if(mark > n) {
mark -= cha;
break;
}
ans += cha;
mark = mark - 1 - step[mark - 1];
}
ans += n - mark;
printf("%d\n",ans);
}
return EXIT_SUCCESS;
}
/*
5 2 4
1
2
3
4
5
4
5
5 2 4
3
2
3
4
5
3
4
*/
接下来是二分做的,
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cstring>
using namespace std;
#define N 100010
int a[N],t[N],sum[N],n,m,k,ans;
int find(int x)
{
int l=0,r=n,res=0;
while(l<=r)
{
int mid=(l+r)/2;
if(sum[mid]<x)
{
res=mid;
l=mid+1;
}
else
{
r=mid-1;
}
}
return res;
}
int main()
{
while(scanf("%d%d%d",&n,&m,&k)!=EOF)
{
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
for(int i=1;i<=m;i++)
scanf("%d",&t[i]);
m++;
t[m]=2000000000;
ans=0;
int pos=0,to;
for(int i=1;i<=m;i++)
{
if(n-pos<t[i]-ans)
{
ans+=n-pos;
break;
}
else
{
to=t[i]-ans+pos;
ans+=t[i]-ans;
pos=find(sum[to-1]-k);
}
}
printf("%d\n",ans);
}
return 0;
}