Codeforces Round #417 (Div. 2) C. Sagheer and Nubian Market 二分答案 +排序

Codeforces Round #417 (Div. 2) C. Sagheer and Nubian Market

二分答案 +排序
题意 有 a[ i ] 个数 要求选最多的数 使其和不超过 S ，且在此情况下，和最小
选最多数情况下 和最小 且 每个数有加成
如果选了 k个数 那么加成后 就是 a[ i ] + k*i ;

题解 二分mid 表示选了个数
加成一下，将加成以后结果排序一下 ， 若前 mid数 和大于 s 则此方案不可行

PS 要用 long long ..... 还有 codeforces 要用 %I64d

 

#include <cstdio>
#include <cstring>
#include <iomanip>
#include <algorithm>
#define ll long long 
using namespace std ; 

const ll maxn = 100011 ; 
ll n,s,l,r,mid ; 
ll a[maxn],b[maxn] ; 

inline bool check(ll mid) 
{
    ll sum = 0 ; 
    for(ll i=1;i<=n;i++) b[ i ] = a[ i ] + 1ll*i*mid ;
    sort(b+1,b+n+1) ; 
    for(ll i=1;i<=mid;i++) 
        sum+=b[ i ] ; 
    return sum <= s ; 
}

int main() 
{
    scanf("%I64d%I64d",&n,&s) ; 
    for(ll i=1;i<=n;i++) scanf("%I64d",&a[ i ]),b[ i ] = a[ i ] ;
    sort(b+1,b+n+1) ; 
     l = 0 ; r = n ; 
    while(l<r) 
    {
        mid = (l+r+1)>>1 ;
        if(check(mid)) l = mid ; 
            else r = mid - 1 ; 
    }
    printf("%I64d ",l) ; 
    ll sum = 0 ;
    for(ll i=1;i<=n;i++) b[ i ] = a[ i ] + 1ll*i*l ;
    sort(b+1,b+n+1) ; 
    for(ll i=1;i<= l ;i++) 
        sum+=b[ i ] ; 
    printf("%I64d\n",sum) ; 
    return 0 ; 
}