oj.uz

Submission #1035236

# Submission time Handle Problem Language Result Execution time Memory
1035236 2024-07-26T07:19:58 Z devariaota Addk (eJOI21_addk) C++17
0 / 100
 36 ms 1712 KB 
#include<bits/stdc++.h>
using namespace std;
#define int long long 

signed main(){
    int n,k;
    cin>>n>>k;
    if(k==1){
        int a[n+1];
        for(int q=1;q<=n;q++){
            cin>>a[q];
        }
        int pref[n+1],suf[n+2],sum[n+1],blk[n+1];
        pref[0]=0;
        suf[n+1]=0;
        sum[0]=0;
        blk[n+1]=0;

        for(int q=1;q<=n;q++){
            pref[q]=pref[q-1]+q*a[q];
            sum[q]=sum[q-1]+a[q];
        }
        for(int q=n;q>=1;q--){
            suf[q]=suf[q+1]+(n-q+1)*a[q];
            blk[q]=blk[q+1]+a[q];
        }

        int q;
        cin>>q;
        for(int w=1;w<=q;w++){
            int type;
            cin>>type;
            if(type==1){
                int r;
                cin>>r;
            }
            else{
                int l,r,m;
                cin>>l>>r>>m;
                int ans=0;

                if((r-l+1)%2==0){
                    int posl=min(l-1+(r-l+1)/2,l+m-1);
                    int posr=max(r+1-(r-l+1)/2,r-m+1);
                    ans+=pref[posl]-pref[l-1]-(l-1)*(sum[posl]-sum[l-1]);
                    ans+=suf[posr]-suf[r+1]-(n-r)*(blk[posr]-blk[r+1]);
                    posl++;
                    posr--;
                    if(posl<=posr){
                        ans+=m*(sum[posr]-sum[posl-1]);
                    }
                }
                else{
                    int posl=min(l+m-1,l-1+(r-l+1)/2);
                    int posr=max(r-m+1,r+1-(r-l+1)/2);
                    //cout<<posl<<" "<<posr<<endl;
                    ans+=pref[posl]-pref[l-1]-(l-1)*(sum[posl]-sum[l-1]);
                    
                    ans+=suf[posr]-suf[r+1]-(n-r)*(blk[posr]-blk[r+1]);
                    //cout<<ans<<endl;

                    posl++;
                    posr--;
                    if(posl<=posr){
                        ans+=m*(sum[posr]-sum[posl-1]);
                    }
                }
                cout<<ans<<endl;
            }
        }
    }
}

Subtask #1
0 / 36.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 56.0

# Verdict Execution time Memory Grader output
1 Incorrect 36 ms 1712 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -