BOJ 17205 : 진우의 비밀번호

• 문제 링크 : boj.kr/17205
• 난이도 : G5
• 태그 : 조합론

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    int n;
    cin >> n;
    string s;
    cin >> s;
    vector<ll> arr(n);
    for(int i = 0; i < s.length(); i++) {
        arr[i] = s[i] - 'a' + 1;
    }

    ll ans = 0;
    for(int i = 0; i < n; i++) {
        if(!arr[i]) break;
        ll temp = 1;
        ll sum = 0;
        for(int j = 0; j < n - i; j++) {
            sum += temp;
            temp *= 26;
        }
        ans += (arr[i]-1) * sum + 1;
    }
    cout << ans;
    return 0;
}

---

풀이

앞에서부터 보았을 떄, 현재 자리의 앞부분은 고정하고 현재 자리는 지금 알파벳보다 작은 알파벳이 올 수 있는 경우를 구해서 더한다.

예를 들어 3자리이고 암호가 cba라면,

첫 번째 자리에선 a~bzz까지 + c를 구한다.

두 번째 자리에선 c~caz까지+ cb를 구한다.

세 번째 자리에선 cb~cb까지+ cba를 구한다.