[백준 1855] 암호 (C#, C++) - soo:bak

문제 링크

1855번 - 암호

설명

지그재그 형태로 문자를 배열한 후, 열 방향으로 읽어 원래의 문자열을 복원하는 문제입니다.

문자열은 입력으로 주어진 열 개수에 따라 위에서 아래로, 왼쪽에서 오른쪽, 또는 오른쪽에서 왼쪽으로 번갈아 가며 채워집니다.

이후 열 단위로 위에서 아래 방향으로 읽었을 때 복원되는 문자열을 계산합니다.

예를 들어, 열이 3이고 암호 문자열이 "aeijfbcgklhd"인 경우, 다음과 같은 순서로 채워집니다:

a e i
j f b
c g k
l h d

이를 위에서 아래로 열 기준으로 읽으면 "abcdefghijkl"이 됩니다.

접근법

• 암호 문자열을 열 개수에 따라 행 x 열 형태로 나누고,
• 짝수 번째 행은 왼쪽에서 오른쪽, 홀수 번째 행은 오른쪽에서 왼쪽 방향으로 채워줍니다.
• 이후 열 기준으로 위에서 아래로 문자열을 읽어 원래 문자열을 복원합니다.

---

Code

C#

using System;
using System.Text;

class Program {
  static void Main() {
    int col = int.Parse(Console.ReadLine());
    string code = Console.ReadLine();
    int row = code.Length / col;
    char[,] table = new char[row, col];

    int idx = 0;
    for (int r = 0; r < row; r++) {
      if (r % 2 == 0) {
        for (int c = 0; c < col; c++)
          table[r, c] = code[idx++];
      } else {
        for (int c = col - 1; c >= 0; c--)
          table[r, c] = code[idx++];
      }
    }

    var sb = new StringBuilder();
    for (int c = 0; c < col; c++)
      for (int r = 0; r < row; r++)
        sb.Append(table[r, c]);

    Console.WriteLine(sb.ToString());
  }
}

C++

#include <bits/stdc++.h>

using namespace std;
typedef vector<char> vc;
typedef vector<vc> vvc;

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

  int col; string code; cin >> col >> code;
  int row = code.size() / col;
  vvc table(row, vc(col));

  for (int r = 0, i = 0; r < row; r++) {
    for (int c = 0; c < col; c++)
      table[r][r % 2 ? col - 1 - c : c] = code[i++];
  }

  for (int c = 0; c < col; c++) {
    for (int r = 0; r < row; r++)
      cout << table[r][c];
  }

  cout << "\n";

  return 0;
}