Submission #1249623

#TimeUsernameProblemLanguageResultExecution timeMemory
1249623FernandoJC073개의 봉우리 (IOI25_triples)C++20
Compilation error
0 ms0 KiB
#include <vector>
#define ll long long
#define vi vector<int>
#define vl vector<ll>
#define For(i, a, n) for(int i = a; i<n; ++i)
using namespace std;

ll query13(vi H){
    int n = H.size();
    int N = n;
    function<bool(int, int, int)> triple = [&](int a, int b, int c){
    if(a>b || a>c || b>c) return false;
    if(c>=n) return false;
    if(a<0) return false;
    vi q1 = {b-a, c-b, c-a};
    vi q2 = {H[a], H[b], H[c]};
    sort(q1.begin(), q1.end());
    sort(q2.begin(), q2.end());
    return (q1 == q2);
};
    ll ans = 0;
    For(i, 0, n){
        For(j, i+1, n){
            if(H[i] == H[j]){
                if(j-i != H[i]) continue;
                int k = j+H[i];
                if(k>=N) continue;
                ans += (H[k] == 2*H[i]);
                continue;
            }
                if(j-i != H[i] && j-i != H[j]){
                int k2 = i+j+H[i]+H[j];
                if(k2&1) continue;
                k2/=2;
                if(k2>=N) continue;
                ans += (H[k2] == j-i && k2-j == min(H[i], H[j]) && k2-i == max(H[i], H[j]));
                continue;
            }
            if(j-i == H[i]){
                ans += triple(i, j, i+H[j]);
                ans += triple(i, j, j+H[j]);
            }
            else if(j-i == H[j]){
                ans += triple(i, j, i+H[i]);
                ans += triple(i, j, j+H[i]);
            }
        }
    }
    return ans;
}

ll count_triples(vi H){
    int n = H.size();
    if(n<=2000) return query13(H);
}
vi construct_range(int M, int K){
    vi ans(M);
    For(i, 0, M) if(i&1) ans[i] = 2; else ans[i]=1;
    return ans;
}

Compilation message (stderr)

triples.cpp: In function 'long long int query13(std::vector<int>)':
triples.cpp:11:5: error: 'function' was not declared in this scope
   11 |     function<bool(int, int, int)> triple = [&](int a, int b, int c){
      |     ^~~~~~~~
triples.cpp:2:1: note: 'std::function' is defined in header '<functional>'; did you forget to '#include <functional>'?
    1 | #include <vector>
  +++ |+#include <functional>
    2 | #define ll long long
triples.cpp:11:32: error: expression list treated as compound expression in functional cast [-fpermissive]
   11 |     function<bool(int, int, int)> triple = [&](int a, int b, int c){
      |                                ^
triples.cpp:11:14: error: expected primary-expression before 'bool'
   11 |     function<bool(int, int, int)> triple = [&](int a, int b, int c){
      |              ^~~~
triples.cpp:40:24: error: 'triple' was not declared in this scope
   40 |                 ans += triple(i, j, i+H[j]);
      |                        ^~~~~~
triples.cpp:44:24: error: 'triple' was not declared in this scope
   44 |                 ans += triple(i, j, i+H[i]);
      |                        ^~~~~~
triples.cpp: In function 'long long int count_triples(std::vector<int>)':
triples.cpp:55:1: warning: control reaches end of non-void function [-Wreturn-type]
   55 | }
      | ^