11575번: Affine Cipher

문제 출처 : https://www.acmicpc.net/problem/11575

11575번: Affine Cipher

언어 : Kotlin

 

문제 설명 :

서쪽나라에서 특수훈련을 받은 정은이는 동쪽나라로 침투를 하게 되었다. 뛰어난 스파이였던 정은이는 동쪽나라의 정보를 입수하게 되었고 정보를 안전하게 서쪽나라로 전달하기 위해 아핀 암호(Affine Cipher)를 이용하기로 하였다.

아핀 암호는 다음과 같은 식을 통해 구할 수 있다.

E(X) = (aX + b) mod 26

A부터 Z까지의 알파벳을 차례대로 0, 1, 2, ... , 25 라고 하자. a = 3이고, b = 1인 경우에 알파벳 A를 아핀 암호식에 대입하면 E(0) = (3 × 0 + 1) mod 26 이 되어 암호화된 결과는 B가 된다.

a와 b, 그리고 알파벳 대문자로만 구성된 평문이 주어졌을 때, 이를 암호문으로 치환하는 프로그램을 작성하라.

 

입력 :

입력의 첫 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 50) 가 주어진다.

각 테스트 케이스의 첫 번째 줄에는 두 정수 a와 b(0 < a, b ≤ 1,000,000)의 값이 주어진다. a는 26과 서로소이다.

각 테스트케이스의 두 번째 줄에는 평문 s가 주어진다. 평문의 길이 |s|는 0보다 크고 1,000,000보다 작다.

 

출력 :

각 테스트 케이스마다 한 줄에 한 개씩 평문 s를 암호문으로 치환한 결과를 출력한다.

 

제한 사항 :

• 시간 제한 : 5초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
2 3 1 IAMSPY 5 3 ABCDEFGHIJKLMNOPQRSTUVWXYZ	ZBLDUV DINSXCHMRWBGLQVAFKPUZEJOTY

 

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val t = readLine().toInt()
    repeat(t) {
        val sb = StringBuilder()
        val (a, b) = readLine().split(" ").map { it.toInt() }
        readLine().forEach {
            sb.append(((a * (it.code - 65) + b) % 26 + 65).toChar())
        }

        if (it < t) bw.appendLine(sb.toString()) else bw.append(sb.toString())
        bw.flush()
    }
    bw.close()
}