UVA1121 Subsequence 题解
题目链接:UVA1121 Subsequence
题意:给定数组,找最短连续子序列使其和大于 $S$ ,多组数据
解法一
对于区间 $[l,r]$ ,若 $\sum_{i=l}^{r}a[i]\ge S$ ,那么 $\forall r’>r (r’ \le n)$ 该条件均成立
那么我们只要枚举左端点,二分查找满足条件的最左的右端点,然后找个最小值就行了
区间和用前缀和优化即可
时间复杂度 $O(n\log n)$
代码如下
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define R register
#define MAXN (int)(1e5+5)
#define INF 0x3f3f3f3f3f3f3f3f
int n,k;
int sum[MAXN];
void proc()
{
R int mn=INF;
for(R int l=1; sum[l]+k<=sum[n]; l++)
{
R int r=lower_bound(sum+l,sum+n,sum[l]+k)-sum-1;
mn=min(mn,r-l+1);
}
printf("%lld\n",mn==INF?0:mn);
}
bool Input()
{
if(!(~scanf("%lld%lld",&n,&k)))
return 0;
for(R int i=1,t; i<=n; i++)
{
scanf("%lld",&t);
sum[i]=sum[i-1]+t;
}
return 1;
}
signed main()
{
while(Input())proc();
return 0;
}
解法二
本解法就是尺取法
根据上面的结论,我们再想一想,有没有必要枚举左端点呢?
对于每个左端点,当且仅当右端点在尽可能左的位置时对答案有贡献
或者说对答案有贡献的每个左端点所对应的右端点是唯一确定的
显然右端点也是单调不降的
我们可以设 $l$ 表示当前左端点的位置, $r$ 不断自增 $1$ 直到 $[l,r]$ 的和大于等于 $S$
这时 $l,r$ 一定是对答案有贡献的一组,和最小值取个 $\min$
那么怎么继续呢?
我们把左端点 $l$ 不断自增 $1$ 直到此时 $[l,r]$ 的和小于 $S$ ,然后再继续拓展右端点
不断重复以上过程直到右端点无法拓展且左端点无解时结束
由于每个点只遍历至多 $2$ 次,因此
时间复杂度 $O(n)$
代码如下
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define R register
#define MAXN (int)(1e5+5)
#define INF 0x3f3f3f3f3f3f3f3f
int n,k;
int a[MAXN];
void proc(R int l,R int r)
{
R int now=0,mn=INF;
while(1)
{
while(now<k&&r<n)now+=a[++r]; // 拓展右端点
if(now>=k)mn=min(mn,r-l+1);
else break; // 无解退出
now-=a[l++]; // 收缩左端点
}
printf("%lld\n",mn==INF?0:mn);
}
bool Input()
{
if(!(~scanf("%lld%lld",&n,&k)))
return 0;
for(R int i=1; i<=n; i++)
scanf("%lld",&a[i]);
return 1;
}
signed main()
{
while(Input())proc(1,0); // 个人传参习惯 qwq
return 0;
}