#837276 (C++23) No.2203 POWER!!!!!

提出ソース

結果

問題	No.2203 POWER!!!!!
ユーザー	ゾビオ
提出日時	2023-02-03 22:33:37
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	32 ms / 2,000 ms
コード長	9,252 bytes
コンパイル時間	3,016 ms
コンパイル使用メモリ	248,500 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-07-02 20:39:52
合計ジャッジ時間	4,158 ms
ジャッジサーバーID （参考情報）	judge4 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	AC	2 ms 6,944 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	5 ms 6,944 KB
testcase_09	AC	12 ms 6,944 KB
testcase_10	AC	19 ms 6,940 KB
testcase_11	AC	27 ms 6,944 KB
testcase_12	AC	5 ms 6,940 KB
testcase_13	AC	32 ms 6,940 KB
testcase_14	AC	32 ms 6,944 KB
testcase_15	AC	32 ms 6,944 KB
testcase_16	AC	32 ms 6,944 KB
testcase_17	AC	31 ms 6,940 KB
testcase_18	AC	2 ms 6,944 KB
testcase_19	AC	2 ms 6,940 KB
testcase_20	AC	2 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
//#include <atcoder/all> // AtCoder
using namespace std;
//using namespace atcoder; // AtCoder
using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
//using mint = modint998244353; // AtCoder
using vll = vector<long long>;
using vvll = vector<vector<long long>>;
using vvvll = vector<vector<vector<long long>>>;
using vvvvll = vector<vector<vector<vector<long long>>>>;
using dll = deque<long long>;
using ddll = deque<deque<long long>>;
using dddll = deque<deque<deque<long long>>>;
using ddddll = deque<deque<deque<deque<long long>>>>;
using pll = pair<long long, long long>;
using vpll = vector<pair<long long, long long>>;
using vvpll = vector<vector<pair<long long, long long>>>;
#define overload2(_1, _2, name, ...) name
#define overload4(_1, _2, _3, _4, name, ...) name
#define rep1(n) for (ll i = 0; i < n; i++)
#define rep2(i, n) for (ll i = 0; i < n; i++)
#define rep3(i, a, b) for (ll i = a; i < b; i++)
#define rep4(i, a, b, c) for (ll i = a; i < b; i += c)
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)
#define reps1(n) for (ll i = 1; i <= n; i++)
#define reps2(i, n) for (ll i = 1; i <= n; i++)
#define reps(...) overload2(__VA_ARGS__, reps2, reps1)(__VA_ARGS__)
#define rrep1(n) for (int i = n - 1; i >= 0; i--)
#define rrep2(i, n) for (int i = n - 1; i >= 0; i--)
#define rrep3(i, a, b) for (int i = b - 1; i >= a; i--)
#define rrep4(i, a, b, c) for (int i = b - 1; i >= a; i -= c)
#define rrep(...) overload4(__VA_ARGS__,rrep4, rrep3, rrep2, rrep1)(__VA_ARGS__)
#define rreps1(n) for (long long i = n; i >= 1; i--)
#define rreps2(i, n) for (long long i = n; i >= 1; i--)
#define rreps(...) overload2(__VA_ARGS__, rreps2, rreps1)(__VA_ARGS__)
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()
#define PI acos(-1.0)
#define YES printf("YES\n"), exit(0)
#define NO printf("NO\n"), exit(0)
#define Yes printf("Yes\n"), exit(0)
#define No printf("No\n"), exit(0)
#define MINUS printf("-1\n"), exit(0)
#define cuot cout // typo
#define endk endl //typo
constexpr char ln = '\n';
constexpr long long MOD = 998244353LL;
constexpr long long LINF = 0x1fffffffffffffff; // 4倍までOK 10^18より大きい
constexpr int INF = 0x3fffffff; // 2倍までOK 10^9より大きい
template<class T> void setcout(T n) {cout << setprecision(n) << fixed;}
template<class... T> constexpr auto min(T... a){return min(initializer_list<common_type_t<T...>>{a...});}
template<class... T> constexpr auto max(T... a){return max(initializer_list<common_type_t<T...>>{a...});}
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; }
template<class T> long long acc(const T& a){ return accumulate(all(a), 0LL); }
template<class T> long double accl(const T& a){ return accumulate(all(a), 0.0L); }
template<class T> vector<T> mrui(const vector<T>& a) { vector<T> ret(a.size() + 1); for(int i = 0; i < a.size(); i++) { ret[i + 1] = ret[i] + a[i]; } return ret; }
template <class T> vector<vector<T>> mrui2(const vector<vector<T>> &a) { vector<vector<T>> ret(a.size() + 1, vector<T>(a.front().size() + 1)); for(int i = 0; i < (int)a.size(); i++) { for(int j = 0; j < (int)a.front().size(); j++) { ret[i + 1][j + 1] = ret[i][j + 1] + ret[i + 1][j] - ret[i][j] + a[i][j]; } } return ret; }

inline void in(){}
template <class Head, class... Tail> inline void in(Head& head, Tail&... tail){ cin >> head; in(tail...); }

#define INT(...) int __VA_ARGS__;in(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;in(__VA_ARGS__)
#define ULL(...) ull __VA_ARGS__;in(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;in(__VA_ARGS__)
#define CHR(...) char __VA_ARGS__;in(__VA_ARGS__)
#define DBL(...) double __VA_ARGS__;in(__VA_ARGS__)
#define LD(...) ld __VA_ARGS__;in(__VA_ARGS__)
#define vec(type, name, ...) vector<type> name(__VA_ARGS__)
#define VEC(type, name, size) vector<type> name(size); in(name)
#define vv(type, name, h, ...) vector<vector<type>>name(h, vector<type>(__VA_ARGS__))
#define VV(type, name, h, ...) vector<vector<type>>name(h, vector<type>(__VA_ARGS__)); in(name)
#define vvv(type, name, h, w, ...) vector<vector<vector<type>>>name(h, vector<vector<type>>(w, vector<type>(__VA_ARGS__)))

template <class T1, class T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p); // prototype

template <class T1, class T2>
istream &operator>>(istream &is, pair<T1, T2> &p); // prototype

template <class T>
ostream &operator<<(ostream &os, const vector<T> &v); // prototype

template <class T>
ostream &operator<<(ostream &os, const vector<vector<T>> &v); // prototype

template <class T>
ostream &operator<<(ostream &os, const vector<vector<vector<T>>> &v); // prototype

template <class T>
istream &operator>>(istream &is, vector<T> &v); // prototype

template <class T, class S>
ostream &operator<<(ostream &os, const map<T, S> &mp); // prototype

template <class T>
ostream &operator<<(ostream &os, const set<T> &st); // prototype

template <class T>
ostream &operator<<(ostream &os, const multiset<T> &st); // prototype

template <class T>
ostream &operator<<(ostream &os, queue<T> q); // prototype

template <class T>
ostream &operator<<(ostream &os, deque<T> q); // prototype

template <class T>
ostream &operator<<(ostream &os, stack<T> st); // prototype

template <class T, class Container, class Compare>
ostream &operator<<(ostream &os, priority_queue<T, Container, Compare> pq); // prototype

//ostream &operator<<(ostream &os, const mint &i); //atcoder // prototype

//ostream &operator<<(ostream &os, const vector<mint> &v); //atcoder // prototype

//ostream &operator<<(ostream &os, const modint &i); //atcoder // prototype

//ostream &operator<<(ostream &os, const vector<modint> &v); //atcoder // prototype

template <class T1, class T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p) {
	os << "(" << p.first << "," << p.second << ")";
	return os;
}

template <class T1, class T2>
istream &operator>>(istream &is, pair<T1, T2> &p) {
	is >> p.first >> p.second;
	return is;
}

template <class T>
ostream &operator<<(ostream &os, const vector<T> &v) {
	for (int i = 0; i < (int)v.size(); i++) {
		os << v[i] << (i + 1 != (int)v.size() ? " " : "");
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, const vector<vector<T>> &v) {
	for (int i = 0; i < (int)v.size(); i++) {
		os << v[i] << endl;
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, const vector<vector<vector<T>>> &v) {
	for (int i = 0; i < (int)v.size(); i++) {
		os << "i = " << i << endl;
		os << v[i];
	}
	return os;
}

template <class T>
istream &operator>>(istream &is, vector<T> &v) {
	for (T &in : v) {
		is >> in;
	}
	return is;
}

template <class T, class S>
ostream &operator<<(ostream &os, const map<T, S> &mp) {
	for (auto &[key, val] : mp) {
		os << key << ": " << val << " ";
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, const set<T> &st) {
	auto itr = st.begin();
	for (int i = 0; i < (int)st.size(); i++) {
		os << *itr << (i + 1 != (int)st.size() ? " " : "");
		itr++;
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, const multiset<T> &st) {
	auto itr = st.begin();
	for (int i = 0; i < (int)st.size(); i++) {
		os << *itr << (i + 1 != (int)st.size() ? " " : "");
		itr++;
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, queue<T> q) {
	while (q.size()) {
		os << q.front() << " ";
		q.pop();
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, deque<T> q) {
	while (q.size()) {
		os << q.front() << " ";
		q.pop_front();
	}
	return os;
}

template <class T>
ostream &operator<<(ostream &os, stack<T> st) {
	while (st.size()) {
		os << st.top() << " ";
		st.pop();
	}
	return os;
}

template <class T, class Container, class Compare>
ostream &operator<<(ostream &os, priority_queue<T, Container, Compare> pq) {
	while (pq.size()) {
		os << pq.top() << " ";
		pq.pop();
	}
	return os;
}

/*ostream &operator<<(ostream &os, const mint &i) { //AtCoder
	os << i.val();
	return os;
}*/

/*ostream &operator<<(ostream &os, const vector<mint> &v) { //AtCoder
	for (int i = 0; i < (int)v.size(); i++) {
		os << v[i].val() << (i + 1 != (int)v.size() ? " " : "");
	}
	return os;
}*/

/*ostream &operator<<(ostream &os, const modint &i) { //AtCoder
	os << i.val();
	return os;
}*/

/*ostream &operator<<(ostream &os, const vector<modint> &v) { //AtCoder
	for (int i = 0; i < (int)v.size(); i++) {
		os << v[i].val() << (i + 1 != (int)v.size() ? "" : "");
	}
	return os;
}*/


//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")

template<typename T>
T mpow(T a, T n, T m) {
	/*a^n % mを返す
	(例)
	pow(2, 10, 1000) --> 24
	計算量はlog(n)
	*/
	T ret = 1;
	while(n > 0) {
		if (n & 1) ret = ret % m * a % m;
		a = a % m * a % m;
		n >>= 1;
	}
	return ret;
}

int main() {
	ll n; cin>> n;
	vll a(n); cin >> a;
	vll b(9); rep(n) b[a[i]]++;
	ll ans = 0;
	rep(i, 1, 9) rep(j, 1, 9) { //i^j
		if(1) {
			ans += b[i] * b[j] * mpow(i, j, LINF);
		}
		else {
			ans += b[i] * (b[i] - 1) / 2 * mpow(i, i, LINF);
		}
	}
	cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード