【NOIP2017提高组模拟6.30】tty's maze

Description

Input

Output

输出一行一个数表示最大的宝藏之和

Sample Input

input 1:
1 5 3
1 2 3 4 5
input 2:
2 5 6
5 1 3 4 2

Sample Output

output 1:
6
output 2:
22

Solution

一道水题，我太垃圾了，没有看出来

根据题目的特性：从最上面可以到最下面，说明无论从哪个位置进入一列，都可以不重复的走完

那么枚举i表示这一次最多往右走到走到第i列，那么这之前的每列一定都要至少选一个
那么还有k−in−1列可以选完整，那么肯定是选前几大的，还有多几个不够一整列，那就再全部选次小的
这个用权值线段树维护，前k大的和和第k大的值

Code

#include<cstdio>#include<algorithm>#include<cmath>#define fo(i,a,b) for(int i=a;i<=b;i++)#define N 1001000#define ll long longusing namespace std;int mx=0,a[N];ll ans,k,n,m,t[N*10],s[N*10];void insert(int v,int i,int j,ll x){    t[v]++;s[v]+=x;    if(i==j) return;    int m=(i+j)/2;    if(x<=m) insert(v*2,i,m,x);    else insert(v*2+1,m+1,j,x);}ll get1(int v,int i,int j,ll x){    if(i==j) return (ll)i*x;    if(x==t[v]) return s[v];    int m=(i+j)/2;    ll jy=get1(v*2+1,m+1,j,min(t[v*2+1],x));    x-=t[v*2+1];    if(x>0) jy+=get1(v*2,i,m,x);    return jy;}ll get2(int v,int i,int j,ll x){    if(i==j) return (ll)i;    int m=(i+j)/2;    if(t[v*2+1]>=x) return get2(v*2+1,m+1,j,x);    else return get2(v*2,i,m,x-t[v*2+1]);}int main(){    scanf("%d%d%lld",&n,&m,&k);    fo(i,1,m) scanf("%d",&a[i]),mx=max(mx,a[i]);    int x=(k-1)/n;    fo(i,1,x) insert(1,1,mx,a[i]);    for(x++;x<=min(k,m);x++)    {        insert(1,1,mx,a[x]);        ll jy=s[1];        int kk=n-1==0?0:(k-x)/(n-1);        jy+=(ll)(n-1)*get1(1,1,mx,kk);        jy+=(ll)(k-x-(n-1)*kk)*get2(1,1,mx,kk+1);        ans=max(ans,jy);    }    printf("%lld\n",ans);}