URAL 1998 The old Padawan

题目只给了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;
}