Submission #1263015

#	Time	Username	Problem	Language	Result	Execution time	Memory
1263015		nerrrmin	Triple Peaks (IOI25_triples)	C++20	34 / 100	2093 ms	3512 KiB

triples

#include "triples.h"
#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxn = 2e5 + 10;
int n;
long long a[maxn];
long long count_triples(std::vector<int> H)
{
    n = H.size();
    for (int i = 1; i <= n; ++ i)
    {
        a[i] = H[i-1];
    }
    long long ans = 0;
    for (int i = 1; i <= n; ++ i)
    {
        for (int j = i + 2; j <= n; ++ j)
        {
            int dist = j - i;
            if(a[i] == dist || a[j] == dist)
            {
                int other = a[i] + a[j] - dist;
                int pos = i + other;
                int is = 0;
                if(i < pos && pos < j && a[pos] == j - pos)
                {
                    is = 1;
                    ans ++;
                }
                pos = j - other;
                if(is && pos == i + other)continue;
                if(i < pos && pos < j && a[pos] == pos - i)ans ++;
            }
            else
            {
                if(a[i] + a[j] == dist)
                {
                    int pos = i + a[i];
                    int is = 0;
                    if(i < pos && pos < j && a[pos] == dist)
                    {
                        is = 1;
                        ans ++;
                    }
                    pos = j - a[i];
                    if(is && pos == i + a[i])continue;
                    if(i < pos && pos < j && a[pos] == dist)ans ++;
                }
            }
        }
    }
    return ans;
}

std::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: 16 / 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