CSES - Datatähti 2021 alku - Results
Submission details
Task:Alitaulukot
Sender:xnor
Submission time:2020-09-28 17:29:12 +0300
Language:Rust
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.01 s1, 2, 3details
#20.01 s1, 2, 3details
#30.01 s1, 2, 3details
#40.01 s1, 2, 3details
#50.01 s1, 2, 3details
#60.01 s2, 3details
#70.01 s2, 3details
#80.01 s2, 3details
#90.01 s2, 3details
#100.01 s2, 3details
#110.01 s3details
#120.01 s3details
#130.01 s3details
#140.01 s3details
#150.01 s3details
#160.01 s3details
#170.01 s3details

Code

use std::io::{self, BufRead};

fn main() {
    let mut lines = io::BufReader::new(io::stdin()).lines();

    let line1 = lines.next().unwrap().unwrap();
    let line2 = lines.next().unwrap().unwrap();
    let mut line1 = line1.split_whitespace();
    let mut line2 = line2.split_whitespace();

    let n: i32 = line1.next().unwrap().parse().unwrap();
    let k: i32 = line1.next().unwrap().parse().unwrap();

    let mut arr = Vec::<i32>::new();
    for _ in 0..n {
        arr.push(line2.next().unwrap().parse().unwrap());
    }
    
    println!("{}", f(&arr, k));
}


fn f(arr: &[i32], k: i32) -> u64 {
    if arr.len() < 2 {
        return arr.len() as u64;
    }

    if arr.len() == 2 {
        if (arr[0] - arr[1]).abs() <= k {
            return 3;
        } else {
            return 2;
        }
    }

    let mid = arr.len() / 2;
    let mut min = (arr[mid], mid);
    let mut max = (arr[mid], mid);
    let mut left = mid;
    let mut right = mid;

    while max.0 - min.0 <= k && (left > 0 || right < arr.len() - 1) {
        if mid - left < right - mid {
            while left > 0 && arr[left] >= min.0 && arr[left] <= max.0 {
                left -= 1;
            }
            if arr[left] < min.0 {
                min = (arr[left], left);
            }
            if arr[left] > max.0 {
                max = (arr[left], left);
            }
        } else {
            while right > 0 && arr[right] >= min.0 && arr[right] <= max.0 {
                right += 1;
            }
            if arr[right] < min.0 {
                min = (arr[right], right);
            }
            if arr[right] > max.0 {
                max = (arr[right], right);
            }
        }
    }

    let final_left = min.1.min(max.1);
    let final_right = min.1.max(max.1);
    let diff = (final_right - final_left) as u64;

    diff * diff + f(&arr[0..final_right], k) + f(&arr[final_left+1..arr.len()], k)
}

Test details

Test 1

Group: 1, 2, 3

Verdict:

input
100 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5050

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 100 but the index is 100', inpu...

Test 2

Group: 1, 2, 3

Verdict:

input
100 2
5 5 2 4 3 5 3 4 3 2 3 4 5 4 4 ...

correct output
317

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 6 but the index is 6', input/co...

Test 3

Group: 1, 2, 3

Verdict:

input
100 10
71 60 61 96 25 10 10 9 84 85 1...

correct output
119

user output
202

Test 4

Group: 1, 2, 3

Verdict:

input
100 990000000
111122929 961821360 578238211 ...

correct output
4006

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 71 but the index is 71', input/...

Test 5

Group: 1, 2, 3

Verdict:

input
100 1000000000
553190572 453407680 667300705 ...

correct output
5050

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 100 but the index is 100', inpu...

Test 6

Group: 2, 3

Verdict:

input
2000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
2001000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 2000 but the index is 2000', in...

Test 7

Group: 2, 3

Verdict:

input
2000 2
4 4 1 4 2 3 1 2 1 3 5 2 2 4 4 ...

correct output
6340

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 6 but the index is 6', input/co...

Test 8

Group: 2, 3

Verdict:

input
2000 10
65 88 33 88 41 10 17 38 22 3 8...

correct output
2413

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 3 but the index is 3', input/co...

Test 9

Group: 2, 3

Verdict:

input
2000 999000000
746120950 772769620 721488968 ...

correct output
1287776

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 595 but the index is 595', inpu...

Test 10

Group: 2, 3

Verdict:

input
2000 1000000000
621947980 510355354 756705418 ...

correct output
2001000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 2000 but the index is 2000', in...

Test 11

Group: 3

Verdict:

input
100000 0
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
5000050000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 100000'...

Test 12

Group: 3

Verdict:

input
100000 2
3 3 1 3 3 1 1 5 1 2 5 4 1 3 1 ...

correct output
317066

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 7 but the index is 7', input/co...

Test 13

Group: 3

Verdict:

input
100000 10
10 3 6 3 43 60 5 48 15 27 86 4...

correct output
123292

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 3 but the index is 3', input/co...

Test 14

Group: 3

Verdict:

input
100000 999990000
460235639 963048588 47270983 3...

correct output
4946886742

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 99399 but the index is 99399',...

Test 15

Group: 3

Verdict:

input
100000 1000000000
885457070 18257718 927615960 3...

correct output
5000050000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 100000'...

Test 16

Group: 3

Verdict:

input
100000 50000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
3750075000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 50002 but the index is 50002',...

Test 17

Group: 3

Verdict:

input
100000 50000
100000 99999 99998 99997 99996...

correct output
3750075000

user output
(empty)

Error:
thread 'main' panicked at 'index out of bounds: the len is 50002 but the index is 50002',...