[SHOI2015]自动刷题机

[SHOI2015]自动刷题机

题目描述

曾经发明了信号增幅仪的发明家 SHTSC 又公开了他的新发明：自动刷题机——一种可以自动 AC 题目的神秘装置。

自动刷题机刷题的方式非常简单：首先会瞬间得出题目的正确做法，然后开始写程序。每秒，自动刷题机的代码生成模块会有两种可能的结果：

1.写了x行代码。

2.心情不好，删掉了之前写的y行代码（如果y大于当前代码长度则相当于全部删除）。

对于一个 OJ，存在某个固定的长度n ，一旦自动刷题机在某秒结束时积累了大于等于n行的代码，它就会自动提交并 AC 此题，然后新建一个文件（即弃置之前的所有代码）并开始写下一题。

SHTSC 在某个 OJ 上跑了一天的自动刷题机，得到了很多条关于写代码的日志信息。他突然发现自己没有记录这个 OJ 的n究竟是多少。

所幸他通过自己在 OJ 上的 Rank 知道了自动刷题机一共切了k道题，希望你计算n可能的最小值和最大值。

solution

二分答案，只有当刚好等于K时才记录答案

如果不存在输出-1，然后我输出了两个-1，就GG了

#include<bits/stdc++.h>

using namespace std;

const int MAXN=10000000+10; 

typedef long long ll;

inline ll read()
{
    ll f=1,x=0;
    char ch;
    do
    {
        ch=getchar();
        if(ch=='-') f=-1;
    }while(ch<'0'||ch>'9');
    do
    {
        x=(x<<3)+(x<<1)+ch-'0';
        ch=getchar();
    }while(ch>='0'&&ch<='9');
    return f*x;
} 

ll n,k;
ll a[MAXN];
ll ans1,ans;

inline ll check(ll x)
{
    ll cnt=0,tot=0;
    for(int i=1;i<=n;i++)
    {
        cnt+=a[i];
        if(cnt>=x) tot++,cnt=0;
        if(cnt<0) cnt=0;
    }
    return tot;
}

int main()
{
    n=read();k=read();
    for(int i=1;i<=n;i++)
    {
        a[i]=read();
    }
    ll l=1LL,r=(1LL)<<62;
    ans1=ans=-1LL;
    while(l<=r)
    {
        ll mid=(l+r)/2;
        ll sum=check(mid);
        if(sum==k) ans=mid;
        if(sum<=k) r=mid-1;
        else l=mid+1; 
    }
    l=1,r=(1LL)<<62;
    ans1=-1LL;
    if(ans==-1)
    {
        cout<<-1<<" ";
        return 0;
    }
    while(l<=r)
    {
        ll mid=(l+r)/2;
        ll sum=check(mid);
        if(sum==k) ans1=mid;
        if(sum<k) r=mid-1;
        else l=mid+1; 
    }
    cout<<ans<<" "<<ans1;
}