Submission #1080964

#	Time	Username	Problem	Language	Result	Execution time	Memory
1080964		pawned	Hexagonal Territory (APIO21_hexagon)	C++17	9 / 100	1 ms	348 KiB

hexagon

#pragma GCC optimize("O1,O2,O3,Ofast,unroll-loops")

#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
#define pb push_back
typedef long long ll;
typedef pair<ll, ll> ii;
typedef vector<ll> vi;

#include "hexagon.h"

const ll MOD = 1e9 + 7;

ll nr(ll x) {
	x %= MOD;
	x += MOD;
	x %= MOD;
	return x;
}

ll binpow(ll x, ll y) {
	x = nr(x);
	ll res = 1;
	while (y > 0) {
		if (y % 2 == 1)
			res = nr(res * x);
		x = nr(x * x);
		y /= 2;
	}
	return res;
}

ll inv(ll x) {
	return binpow(x, MOD - 2);
}

int N;

ll A, B;

vi D, L;

int draw_territory(int N_g, int A_g, int B_g, vector<int> D_g, vector<int> L_g) {
	N = N_g; A = A_g; B = B_g;
	for (int x : D_g)
		D.pb(x);
	for (int x : L_g)
		L.pb(x);
	ll x1 = nr(nr((L[0] + 1) * (L[0] + 2)) * inv(2));
	ll x2 = nr(nr(nr(L[0] * (L[0] + 1)) * (L[0] + 2)) * inv(3));
//	cout<<"x1, x2: "<<x1<<" "<<x2<<endl;
	return nr(x1 * A + x2 * B);
}

• 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