Submission #1254401

#	Time	Username	Problem	Language	Result	Execution time	Memory
1254401		osmiyum	Addk (eJOI21_addk)	C++20	92 / 100	39 ms	5704 KiB

Main

#include <iostream>

using namespace std;

long long int A[100005];
long long int pref[100005];
long long int suf[100005];
long long int sim[100005];
long long int simsuf[100005];

int main() {

    long long int N,K;
    scanf("%lld %lld",&N,&K);

    for(int i=1;i<=N;++i) scanf("%lld",&A[i]);

    sim[0] = 0;
    for(int i=1;i<=N;++i) sim[i] = sim[i-1] + A[i];

    pref[0] = 0;
    for(int i=1;i<=N;++i) pref[i] = pref[i-1] + i*A[i];

    suf[N+1] = 0;
    for(int i=N;i>0;--i) suf[i] = suf[i+1] + (N-i+1)*A[i];

    simsuf[N+1] = 0;
    for(int i=N;i>0;--i) simsuf[i] = simsuf[i+1] + A[i];

    long long int Q;
    scanf("%lld",&Q);

    for(int q=0;q<Q;++q) {
        int t;
        scanf("%d",&t);

        if (t==1) {
            int a;
            scanf("%d",&a);
        }
        else {
            long long int res=0;
            long long int l,r,m;
            scanf("%lld%lld%lld",&l,&r,&m);
            long long int a = min(m,r-l+1-m+1);
            if (a < 0) a = 0;
            res += pref[l-1+a] - pref[l-1] - (l-1)*(sim[l-1+a]-sim[l-1]);
            

            long long int b = r-l+1 - 2*a + 1;
            res += a*(sim[b+a+l-1] - sim[l-1 + a]);
            
            long long int c = a-1;
            res += (suf[l+b+a] - suf[r+1]) - (N-r)*(simsuf[l+b+a] - simsuf[r+1]);
            
            printf("%lld\n",res);
        }


    }



}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:14:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   14 |     scanf("%lld %lld",&N,&K);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~
Main.cpp:16:32: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   16 |     for(int i=1;i<=N;++i) scanf("%lld",&A[i]);
      |                           ~~~~~^~~~~~~~~~~~~~
Main.cpp:31:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 |     scanf("%lld",&Q);
      |     ~~~~~^~~~~~~~~~~
Main.cpp:35:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   35 |         scanf("%d",&t);
      |         ~~~~~^~~~~~~~~
Main.cpp:39:18: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   39 |             scanf("%d",&a);
      |             ~~~~~^~~~~~~~~
Main.cpp:44:18: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   44 |             scanf("%lld%lld%lld",&l,&r,&m);
      |             ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 36 / 36
• Subtask 2: 56 / 56
• Subtask 3: 0 / 8