HDU - 6606

제목 링크: HDU - 6606

      。

      ,     dp ,dp[i]=max{dp[j]+1},  j+1 i      mid  。

         max,        ,      ,    ,         。

AC 코드:
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include
#define int long long
using namespace std;
const int N=2e5+10,M=N*4;
int n,k,a[N],mx[M],dp[N],lc[M],rc[M],rt,cnt,up;
vector<int> v;
#define mid (l+r>>1)
void change(int &p,int l,int r,int x,int v){
    if(!p)    p=++cnt,mx[p]=-1e18,lc[p]=rc[p]=0;
    if(l==r){mx[p]=max(mx[p],v);    return ;}
    if(x<=mid)    change(lc[p],l,mid,x,v);
    else     change(rc[p],mid+1,r,x,v);
    mx[p]=max(mx[lc[p]],mx[rc[p]]);
}
int ask(int p,int l,int r,int ql,int qr){
    if(!p||ql>qr)    return -1e18;
    if(l==ql&&r==qr)    return mx[p];
    if(qr<=mid)    return ask(lc[p],l,mid,ql,qr);
    else if(ql>mid) return ask(rc[p],mid+1,r,ql,qr);
    else return max(ask(lc[p],l,mid,ql,mid),ask(rc[p],mid+1,r,mid+1,qr));
}
#undef mid
inline int get(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
inline int check(int mid){
    rt=cnt=0; change(rt,1,up,get(0),0);
    for(int i=1;i<=n;i++){
        dp[i]=ask(rt,1,up,get(v[a[i]-1]-mid),up)+1;
        change(rt,1,up,a[i],dp[i]);
        if(dp[i]>=k)    return 1;
    }
    return 0;
}
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){
    int x=0,f=1; char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
    return x*f;
}
inline void solve(){
    n=read(),k=read(); mx[0]=-1e18;
    for(int i=1;i<=n;i++) a[i]=read(),a[i]+=a[i-1],v.push_back(a[i]); v.push_back(0);
    sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); up=v.size();
    for(int i=1;i<=n;i++)	a[i]=get(a[i]);
    int l=-2e14,r=2e14;
    while(l<r){
        int mid=l+r>>1;
        if(check(mid))    r=mid;
        else l=mid+1;
    }
    cout<<l<<'
'
; } signed main(){ int T; T=read(); while(T--) solve(); return 0; }

좋은 웹페이지 즐겨찾기