Submission #1250669

#	Time	Username	Problem	Language	Result	Execution time	Memory
1250669		Cebrayil09	Triple Peaks (IOI25_triples)	C++20	29.29 / 100	2096 ms	20596 KiB

triples

#include "triples.h"
#include <bits/stdc++.h>

using namespace std;

#define ll long long

long long count_triples(vector<int> h) {
    int mx = *max_element(h.begin(), h.end());

    ll cnt = 0;
    if(mx <= 10) {
        for(int i = 0;i < h.size();i++) {
            for(int k = i+1;k <= min((int)h.size()-1, i+10);k++) {
                for(int j = i+1;j < k;j++) {
                    vector<int> a = {h[i], h[j], h[k]};
                    vector<int> b = {j-i, k-i, k-j};

                    sort(a.begin(), a.end());
                    sort(b.begin(), b.end());

                    cnt += (a == b);
                }
            }
        }

        return cnt;
    }

    set<vector<int>> s;
    for(int i = 0;i < h.size();i++) {
        for(int j = i+1;j < h.size()-1;j++) {
            for(int k : {i+h[i], i+h[j], i-h[i], i-h[j], j+h[i], j+h[j], j-h[i], j-h[j]}) {
                if(k < 0 || k >= h.size()) continue;

                vector<int> a = {j-i, abs(k-i), abs(k-j)};
                vector<int> b ={h[i], h[j], h[k]};

                sort(a.begin(), a.end());
                sort(b.begin(), b.end());

                if(a == b) {
                    vector<int> in = {i,j,k};
                    sort(in.begin(), in.end());

                    s.insert(in);
                }
            }
        }
    }

    cnt = s.size();
    return cnt;
}

vector<int> construct_range(int M, int K) {
    vector<int> v;
    for(int i = 0, j = 0;i < M;i++) {
        if(j < 2) {
            v.push_back(1);
            j++;
        }
        else {
            v.push_back(2);
            j = 0;
        }
    }

    return v;
}

• Subtask 0: 0 / 0
• Subtask 1: 8 / 8
• Subtask 2: 6 / 6
• Subtask 3: 10 / 10
• Subtask 4: 0 / 11
• Subtask 5: 0 / 16
• Subtask 6: 0 / 19
• Subtask 7: 3 / 5
• Subtask 8: 1.245 / 5
• Subtask 9: 0.4998 / 5
• Subtask 10: 0.2143 / 5
• Subtask 11: 0.25 / 5
• Subtask 12: 0.0833 / 5