Submission #613885

#	Time	Username	Problem	Language	Result	Execution time	Memory
613885		AugustinasJucas	Hexagonal Territory (APIO21_hexagon)	C++14	9 / 100	1 ms	340 KiB

hexagon

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

using namespace std;
int n = 0;
const long long mod = 1e9 + 7;
long long pw(long long x, long long y) {
    if(y == 0) return 1;
    auto sq = pw(x, y/2);
    sq = sq * sq % mod;
    if(y & 1) sq = sq * x % mod;
    return sq;
}
long long inv(long long x) {
    return pw(x, mod-2);
}
long long S(long long ind) {
    ind--;
    long long ret = ind*ind%mod*ind%mod;
    ret = (ret + 3ll*ind % mod * ind % mod) % mod;
    ret = (ret + 2ll * ind % mod) % mod;
    ret = ret * inv(3) % mod;
    return ret;
}
int draw_territory(int N, int A, int B,
                   std::vector<int> D, std::vector<int> L) {
    n = N;
    if (n == 3) {
        long long side = (L[0] + 1) % mod;
        long long area = (side * (side + 1) / 2ll) % mod;
      //  cout << "area = " << area << endl;
        long long ret = (area * A % mod + B * S(side) % mod) % mod;
        return ret;
    }


    return 0;
}
/*
3 1 0
1 3
3 3
5 3
*/

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