[AtCoder] ABC 152 D – Handstand 2

2020年1月20日2020年12月15日

問題

方針

\( 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;
}