您的位置:首页 > 产品设计 > UI/UE

POJ 3061 Subsequence 二分或者尺取法

2014-03-11 08:49 417 查看
http://poj.org/problem?id=3061

题目大意:

给定长度为n的整列整数a[0],a[1],……a[n-1],以及整数S,求出总和不小于S的连续子序列的长度的最小值。

思路:

方法一:

首先求出各项的和sum[i],这样可以在O(1)的时间内算出区间上的总和,这样,枚举每一个起点i,然后二分搜索出结果大于sum[i]+tot的最小下标。(tot是题目中的S)

总的时间为O(nlogn)

方法二:

设以a[s]开始的总和最初大于S时的连续子序列为a[s]+a[s+1]+……a[t-1],这时,a[s+1]+a[s+1]+……a[t-2]<a[s]+a[s+1]+……a[t-2]<S,所以如果从a[s+1]开始总和最初超过S的连续子序列是a[s+1]+……a[t‘-1],则t<=t'。

故可以设计如下算法:

1.初始s=t=sum=0

2.只要依然有sum<S,就不断将sum加上a[t],并且t=t+1;

3.如果2中无法满足sum>=s则终止,否则ans=min(ans,t-s);

4.将sum减去a[s],s+=1后回到2

总复杂度为O(N)

1.二分法

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=100000+10;
const int INF=0x7fffffff;
int a[MAXN],sum[MAXN];
int search(int L,int R,int target) //(L,R]
{
while(L<R-1)
{
int m=(L+R)>>1;
if(sum[m]<target)
L=m;
else
R=m;
}
return R;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,tot;
scanf("%d%d",&n,&tot);
sum[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
int ans=INF;
for(int i=1;i<=n&&sum[i]+tot<=sum
;i++)
{
int t=search(i,n,tot+sum[i]);//sum[t]-sum[i]>=tot -> sum[t]>=tot+sum[i]
ans=min(ans,t-i);
}
if(ans==INF)
printf("0\n");
else
printf("%d\n",ans);
}
return 0;
}


2.尺取法

#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=100000+10;
const int INF=0x7fffffff;
int a[MAXN];

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,tot;
scanf("%d%d",&n,&tot);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
int s=0,t=0,sum=0,ans=INF;
while(true)
{
while(sum<tot && t<n)
{
sum+=a[t++];
}
if(sum<tot) break;
ans=min(ans,t-s);
sum-=a[s++];
}
if(ans==INF)
printf("0\n");
else
printf("%d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: