[BOJ] 1394 암호

문제

1394번: 암호

유진이는 현수의 암호를 알아내려고 한다. 유진이는 사전 조사를 통해 임현수의 컴퓨터에 어떤 문자들이 쓰이는지 알아내었고, 하나씩 대입해보려고 한다. 대입하는 순서는 유진이가 메모한 문자 집합의 순서대로이고, 한 글자부터 암호가 풀릴 때까지 모두 대입해본다.

https://www.acmicpc.net/problem/1394

풀이

코드

#include <iostream>
#include <cstdio>
#include <map>
#define MOD 900528
using namespace std;
typedef long long ll;
 
string s,m;
map<char, int> mp;
ll ans = 0;
 
int main() {
    cin >> s; // 암호로 사용할 수 있는 문자
    for(int i = 0; i < s.size(); i++) mp[s[i]] = i+1;
    cin >> m; // 암호
    int sz = m.size(); // 암호 크기
    ll k = s.size();
    for(int i = 1; i < sz; i++) {
        ans = (ans + k) % MOD;
        k = (k * s.size()) % MOD;
    }
    k = 1;
    for(int i = sz-1; i >= 0; i--) {
        ll t = ((mp[m[i]] - mp[s[0]]) % MOD * k) % MOD;
        ans = (ans + t) % MOD;
        k = (k * s.size()) % MOD;
    }
    printf("%lld", (ans + 1) % MOD);
}