[AtCoder] ABC 152 D – Handstand 2
問題
方針
\( a(i, j) \) を先頭の数字が \( i \) で末尾が \( j \) である \( N \) 以下の正の整数の個数とします。これをもちいて、求める答えは、
\[\sum_{i = 1}^{9} \sum_{j = 1}^{9} a(i, j) \times a(j, i) \]
となります。
コード
#include <bits/stdc++.h> using namespace std; typedef long long ll; int main() { int N; cin >> N; ll a[10][10]{}; for (int i = 1; i <= N; i++) { string s = to_string(i); a[s[0] - '0'][s[s.length() - 1] - '0']++; } ll ans = 0; for (int i = 1; i < 10; i++) { for (int j = 1; j < 10; j++) { ans += a[i][j] * a[j][i]; } } cout << ans << "\n"; return 0; }
ディスカッション
コメント一覧
まだ、コメントがありません