Submission #338832

#	Time	Username	Problem	Language	Result	Execution time	Memory
338832		beksultan04	Mountain Trek Route (IZhO12_route)	C++14	0 / 100	1 ms	364 KiB

route

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<int,int>
#define OK puts("OK");
#define NO puts("NO");
#define YES puts("YES");
#define fr first
#define sc second
#define ret return
#define scan1(a) scanf("%lld",&a);
#define scan2(a,b) scanf("%lld %lld",&a, &b);
#define scan3(a,b,c) scanf("%lld %lld %lld",&a,&b,&c);
#define all(s) s.begin(),s.end()
#define allr(s) s.rbegin(),s.rend()
#define pb push_back
#define sz(v) (int)v.size()
#define endi puts("");
const int N = 1e6+12,INF=1e9+7;
int q[N],n;

inline void fun(int x,int &ans){
    for (int i=1;i<=n;++i){
        if (q[i] < x)ans+=x-q[i];
    }
}

main(){
    int i,j,k,ans=0;
    scan2(n,k)
    for (i=1;i<=n;++i){
        scan1(q[i])
    }
    q[n+1]=q[1];
    for (i=1;i<=n;++i){
        ans+=abs(q[i+1]-q[i]);
    }
    int l=1,r=INF;
    while (r-l>1){
        int m = l+r>>1;
        int x = 0;
        fun(m,x);
        if (x > k)r=m;
        else l=m;
    }
    int x = 0;
    fun(r,x);
    if (x <= k)l=r;
    for (i=1;i<=n;++i){
        if (q[i] < l){
            k-=l-q[i];
            q[i] = l;
        }
    }
    q[n+1]=q[1];
    q[0]=q[n];
    for (i=1;i<=n;++i){
        if (q[i-1] > q[i] && q[i] < q[i+1]){
            int mn = abs(q[i-1]-q[i]);
            mn = min(mn,abs(q[i+1]-q[i]));
            mn = min(mn,k);
            k -= mn;
            q[i]+=mn;
        }
    }
    q[n+1]=q[1];
    int y =0;
    for (i=1;i<=n;++i){
        y+=abs(q[i+1]-q[i]);
    }

    cout <<ans-y;
}

Compilation message (stderr)

route.cpp:28:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   28 | main(){
      |      ^
route.cpp: In function 'int main()':
route.cpp:40:18: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   40 |         int m = l+r>>1;
      |                 ~^~
route.cpp:29:11: warning: unused variable 'j' [-Wunused-variable]
   29 |     int i,j,k,ans=0;
      |           ^
route.cpp:12:25: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   12 | #define scan2(a,b) scanf("%lld %lld",&a, &b);
      |                    ~~~~~^~~~~~~~~~~~~~~~~~~~
route.cpp:30:5: note: in expansion of macro 'scan2'
   30 |     scan2(n,k)
      |     ^~~~~
route.cpp:11:23: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   11 | #define scan1(a) scanf("%lld",&a);
      |                  ~~~~~^~~~~~~~~~~
route.cpp:32:9: note: in expansion of macro 'scan1'
   32 |         scan1(q[i])
      |         ^~~~~

• Subtask 1: 0 / 100