【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-i\over n-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 long
using 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);
}