尺取法 POJ 3061

反复地推进区间的开头和末尾,来求取满足条件的最小区间的方法被称为尺取法。     ——《挑战程序设计竞赛》

POJ 3061

因为t最多变化n次,因此只需O(n)的复杂度就可以解决这个问题了。

AC code:

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<cstdlib>
#define ll long long
#define mem(a,x) memset((a),(x),sizeof ((a)))//x只能是0或-1或false或true
#define debug(x) cout<<"X: "<<(x)<<endl
#define de cout<<"************"<<endl
#define lowbit(x) ((x)&(-x))
#define lson rt<<1
#define rson rt<<1|1
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/(__gcd(a,b))
#define inf 0x3f3f3f3f//1e9+6e7
#define eps 1e-8
#define mod 1e9+7
#define N 100010
const double pi=acos(-1.0);
using namespace std;
int a[N];
void solve(int a[],int n,int k)    //在序列a中求出总和不小于k的连续子序列的长度的最小值,n为序列a的长度
{
    int res=n+1;
    int s=0,t=0,sum=0;
    while(1)
    {
        while(t<n&&sum<k)
        {
            sum+=a[t++];
        }
        if(sum<k)
            break;
        res=min(res,t-s);
        sum-=a[s++];
    }
    if(res>n)    //解不存在
        res=0;
    printf("%d\n",res);
}
int main()
{
    int T;
    cin>>T;
    int n,k;
    while(T--)
    {
        mem(a,0);
        cin>>n>>k;
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        solve(a,n,k);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/flyzer/article/details/83317122