Submission #743391

#	Time	Username	Problem	Language	Result	Execution time	Memory
743391		saayan007	Hexagonal Territory (APIO21_hexagon)	C++17	3 / 100	1 ms	256 KiB

hexagon

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

#define ll long long
#define fr first
#define sc second
#define eb emplace_back
#define nl '\n';

const ll mod = 1000000007;

ll mul(ll a, ll b) {
    return ((a * b) % mod + mod) % mod;
}

ll sum(ll a, ll b) {
    return ((a + b) % mod + mod) % mod;
}

int draw_territory(int N, int A, int B, vector<int> D, vector<int> L) {
    /* int sz = 10; */
    /* int dist[sz + 2][sz + 2]; */
    /* for(int i = 0; i < sz + 2; ++i) for(int j = 0; j < sz + 2; ++j) if(i > 0 && i <= sz && j > 0 && j <= sz) dist[i][j] = -1; else dist[i][j] = 0; */

    /* dist[sz/2][sz/2] = 0; */
    /* queue<pair<int, int>> q; */
    /* q.emplace(sz/2, sz/2); */
    /* while(!q.empty()) { */
    /*     int x = q.front().fr, y = q.front().sc; */
    /*     q.pop(); */
    /*     for(int dx : {-1, 0, 1}) for(int dy : {-1, 0, 1}) if(dx * dy != 1) { */
    /*         int nx = x + dx, ny = y + dy; */
    /*         if(dist[nx][ny] == -1) { */
    /*             dist[nx][ny] = dist[x][y] + 1; */
    /*             q.emplace(nx, ny); */
    /*         } */
    /*     } */
    /* } */

/*     for(int i = 1; i <= sz; ++i) { */
/*         for(int j = 1; j <= sz; ++j) { */
/*             cout << dist[i][j] << ' '; */
/*         } */
/*         cout << nl; */
/*     } */

/*     cout << "INPUT" << nl; */
/*     cout << N << ' ' << A << ' ' << B << nl; */
/*     for(int i = 0; i < N; ++i) { */
/*         cout << D[i] << ' ' << L[i] << nl; */
/*     } */
/*     for(int i = 0; i < N; ++i) { */
/*         --D[i]; */
/*         if(D[i] >= 3) D[i] = 8 - D[i]; */
/*     } */

    /* int res = 0; */

    ll n = L[0];
    ll res = (n & 1 ? mul(A, mul((n + 1) / 2, n + 2)) : mul(A, mul((n + 2) /2, n + 1)));
    res = sum(res, mul(B, (n * (n + 1) * (n + 2)) / 3));
    /* cout << (n * (n + 1) * (n + 2)) / 3 << nl; */
    return res;
}

• Subtask 1: 3 / 3
• Subtask 2: 0 / 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