CSES - HIIT Open 2024 - Results
Submission details
Task:Equilateral numbers
Sender:asdf4321
Submission time:2024-11-16 15:19:58 +0200
Language:C++ (C++17)
Status:READY
Result:ACCEPTED
Test results
testverdicttime
#1ACCEPTED0.00 sdetails
#2ACCEPTED0.00 sdetails
#3ACCEPTED0.00 sdetails
#4ACCEPTED0.00 sdetails
#5ACCEPTED0.00 sdetails
#6ACCEPTED0.02 sdetails
#7ACCEPTED0.00 sdetails
#8ACCEPTED0.10 sdetails
#9ACCEPTED0.04 sdetails
#10ACCEPTED0.03 sdetails
#11ACCEPTED0.05 sdetails
#12ACCEPTED0.13 sdetails

Code

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define rep(i, a, b) for (int i = a; i < (b); i++)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;


vi num;
bool Exist(int x)
{
    auto it = lower_bound(all(num), x);
    if(it == num.end())
        return false;
    return (*it) == x;
}

signed main() {
#ifdef LOCAL
freopen(".inp", "r", stdin);
freopen(".out", "w", stdout);
#endif

    int n, now = 0;
    cin >> n;
    
    for (int i = 1; now <= n; i++) {
        now += i;
        num.push_back(now);
    }

    if(Exist(n)) {
        cout << 1;
        return 0;
    }


    for(int x: num)
    {
        int t = n - x;
        if(Exist(t)) {
            cout << 2;
            return 0;
        }

    }

    cout << 3;
    return 0;
}

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