[백준 15874] Caesar Cipher (C++/Python)

C++)

#include <iostream>
#include <string>

using namespace std;

int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);

	int k, s;
	string st;

	cin >> k >> s;
	cin.ignore();
	getline(cin, st);

	k %= 26;

	for (auto ch : st) {
		if (isupper(ch)) 
			cout << char((ch - 'A' + k) % 26 + 'A');
		else if (islower(ch)) 
			cout << char((ch - 'a' + k) % 26 + 'a');
		else
			cout << ch;
	}

	return 0;
}

 

 

 

Python)

import sys
read = sys.stdin.readline

k, s = map(int, read().rstrip().split())
k %= 26

st = read().rstrip()
ans = []

for ch in st:
    if ch.isupper():
        ans.append(chr((ord(ch) - ord('A') + k) % 26 + ord('A')))
    elif ch.islower():
        ans.append(chr((ord(ch) - ord('a') + k) % 26 + ord('a')))
    else:
        ans.append(ch)

print(''.join(ans))