Submission #1250060

#	Time	Username	Problem	Language	Result	Execution time	Memory
1250060		Aliyyiakbar	Triple Peaks (IOI25_triples)	C++20	18 / 100	2095 ms	1860 KiB

triples

#include "triples.h"
#include "bits/stdc++.h"
using namespace std;

bool is(vector<int> &h, int i, int j, int k)
{
    if (j < i || k < i || k >= (int)h.size())
    {
        return 0;
    }
    vector<int> d = {abs(i - j), abs(j - k), abs(k - i)};
    vector<int> c = {h[i], h[j], h[k]};
    sort(d.begin(), d.end());
    sort(c.begin(), c.end());
    return (d == c);
}

long long count_triples(vector<int> h)
{
    int cnt = 0;
    for (int i = 0; i < (int)h.size(); ++i)
    {
        int j = i + h[i];
        if (j < (int)h.size() && h[j] != h[i])
        {
            if (i + h[j] != j + h[i])
            {
                cnt += is(h, i, j, i + h[j]);
            }
            if (i + h[j] != j - h[j])
            {
                cnt += is(h, i, j, j - h[j]);
            }
            cnt += is(h, i, j, j + h[j]);
        }
        for (j = i + 1; j + h[i] < (int)h.size(); ++j)
        {
            cnt += is(h, i, j, j + h[i]);
        }
    }
    return cnt;
}

vector<int> construct_range(int m, int k)
{
    return {1, 1, 1};
}

• Subtask 0: 0 / 0
• Subtask 1: 8 / 8
• Subtask 2: 0 / 6
• Subtask 3: 10 / 10
• Subtask 4: 0 / 11
• Subtask 5: 0 / 16
• Subtask 6: 0 / 19
• Subtask 7: 0 / 5
• Subtask 8: 0 / 5
• Subtask 9: 0 / 5
• Subtask 10: 0 / 5
• Subtask 11: 0 / 5
• Subtask 12: 0 / 5