Task: | Equilateral numbers |
Sender: | Viestipojat |
Submission time: | 2024-11-16 16:25:25 +0200 |
Language: | Rust (2021) |
Status: | READY |
Result: | ACCEPTED |
test | verdict | time | |
---|---|---|---|
#1 | ACCEPTED | 0.10 s | details |
#2 | ACCEPTED | 0.12 s | details |
#3 | ACCEPTED | 0.10 s | details |
#4 | ACCEPTED | 0.17 s | details |
#5 | ACCEPTED | 0.17 s | details |
#6 | ACCEPTED | 0.16 s | details |
#7 | ACCEPTED | 0.16 s | details |
#8 | ACCEPTED | 0.16 s | details |
#9 | ACCEPTED | 0.10 s | details |
#10 | ACCEPTED | 0.10 s | details |
#11 | ACCEPTED | 0.10 s | details |
#12 | ACCEPTED | 0.16 s | details |
Code
use std::str::FromStr;use std::fmt::Debug;use std::collections::HashSet;fn read<T: FromStr>() -> Vec<T>where <T as FromStr>::Err: Debug {let stdin = std::io::stdin();let line = stdin.lines().next().unwrap().unwrap();line.split_whitespace().map(T::from_str).map(|x| x.unwrap()).collect()}fn main() {let n = read::<usize>()[0];let nums = (1..=1_501_000).map(|n| n * (n + 1) / 2).collect::<HashSet<_>>();if nums.contains(&n) {println!("1");} else if nums.iter().any(|s| nums.contains(&n.wrapping_sub(*s))) {println!("2");} else {println!("3");}}
Test details
Test 1
Verdict: ACCEPTED
input |
---|
1 |
correct output |
---|
1 |
user output |
---|
1 |
Test 2
Verdict: ACCEPTED
input |
---|
2 |
correct output |
---|
2 |
user output |
---|
2 |
Test 3
Verdict: ACCEPTED
input |
---|
3 |
correct output |
---|
1 |
user output |
---|
1 |
Test 4
Verdict: ACCEPTED
input |
---|
5 |
correct output |
---|
3 |
user output |
---|
3 |
Test 5
Verdict: ACCEPTED
input |
---|
33 |
correct output |
---|
3 |
user output |
---|
3 |
Test 6
Verdict: ACCEPTED
input |
---|
12385719843 |
correct output |
---|
3 |
user output |
---|
3 |
Test 7
Verdict: ACCEPTED
input |
---|
10935032 |
correct output |
---|
3 |
user output |
---|
3 |
Test 8
Verdict: ACCEPTED
input |
---|
659023495928 |
correct output |
---|
3 |
user output |
---|
3 |
Test 9
Verdict: ACCEPTED
input |
---|
913591235689 |
correct output |
---|
2 |
user output |
---|
2 |
Test 10
Verdict: ACCEPTED
input |
---|
999999911791 |
correct output |
---|
1 |
user output |
---|
1 |
Test 11
Verdict: ACCEPTED
input |
---|
1000000000000 |
correct output |
---|
2 |
user output |
---|
2 |
Test 12
Verdict: ACCEPTED
input |
---|
999999999998 |
correct output |
---|
3 |
user output |
---|
3 |