Submission #69342428

Source Code Expand

#include <iostream>
using namespace std;
long long arr[300001];
long long pfx[300001];
int main() {
    long long n, r;
    cin >> n >> r;
    for(long long i = 1; i<=n; i++){
        cin >> arr[i];
        pfx[i] = arr[i] + pfx[i-1];
    }
    long long ans = 0;
    long long s = r+1;
    for(long long i = 1; i<=r; i++){
        if(arr[i] != 1){s = i;break;}
    }
    ans += pfx[r] - pfx[s-1];
    ans += r-s +1;
    s = r;
    for(long long i = n; i>r; i--){
        if(arr[i] != 1){s = i; break;}
    }
    ans += pfx[s] - pfx[r];
    ans += s - r;
    cout << ans;
    return 0;
}

Submission Info

Submission Time
Task C - Lock All Doors
User tonystar0420
Language C++ 17 (gcc 12.2)
Score 300
Code Size 613 Byte
Status AC
Exec Time 20 ms
Memory 6844 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 33
Set Name Test Cases
Sample sample00.txt, sample01.txt, sample02.txt
All sample00.txt, sample01.txt, sample02.txt, testcase00.txt, testcase01.txt, testcase02.txt, testcase03.txt, testcase04.txt, testcase05.txt, testcase06.txt, testcase07.txt, testcase08.txt, testcase09.txt, testcase10.txt, testcase11.txt, testcase12.txt, testcase13.txt, testcase14.txt, testcase15.txt, testcase16.txt, testcase17.txt, testcase18.txt, testcase19.txt, testcase20.txt, testcase21.txt, testcase22.txt, testcase23.txt, testcase24.txt, testcase25.txt, testcase26.txt, testcase27.txt, testcase28.txt, testcase29.txt
Case Name Status Exec Time Memory
sample00.txt AC 1 ms 3464 KiB
sample01.txt AC 1 ms 3584 KiB
sample02.txt AC 1 ms 3588 KiB
testcase00.txt AC 13 ms 5852 KiB
testcase01.txt AC 18 ms 6652 KiB
testcase02.txt AC 17 ms 6252 KiB
testcase03.txt AC 19 ms 6696 KiB
testcase04.txt AC 4 ms 4060 KiB
testcase05.txt AC 18 ms 6836 KiB
testcase06.txt AC 14 ms 5812 KiB
testcase07.txt AC 19 ms 6700 KiB
testcase08.txt AC 8 ms 4680 KiB
testcase09.txt AC 19 ms 6608 KiB
testcase10.txt AC 15 ms 5840 KiB
testcase11.txt AC 19 ms 6596 KiB
testcase12.txt AC 2 ms 3764 KiB
testcase13.txt AC 19 ms 6592 KiB
testcase14.txt AC 17 ms 6380 KiB
testcase15.txt AC 20 ms 6656 KiB
testcase16.txt AC 18 ms 6576 KiB
testcase17.txt AC 18 ms 6688 KiB
testcase18.txt AC 1 ms 3528 KiB
testcase19.txt AC 20 ms 6588 KiB
testcase20.txt AC 11 ms 5320 KiB
testcase21.txt AC 19 ms 6804 KiB
testcase22.txt AC 18 ms 6364 KiB
testcase23.txt AC 18 ms 6804 KiB
testcase24.txt AC 16 ms 6212 KiB
testcase25.txt AC 18 ms 6844 KiB
testcase26.txt AC 17 ms 6572 KiB
testcase27.txt AC 19 ms 6696 KiB
testcase28.txt AC 18 ms 6800 KiB
testcase29.txt AC 18 ms 6672 KiB