SPOJ.COM - Thuật toán bài TOANDFRO - To and Fro

#include <iostream>
using namespace std;

int main()
{
	//freopen("input.txt","r",stdin);
	int NUM_COL = 0;
	while(true)
	{
	    // Nhập số lượng cột
		cin >> NUM_COL;
		if(NUM_COL == 0) break;

		// Xâu đầu vào
		char str[201];
		cin >> str;

		// Tính độ dài của xâu
		int length = 0;
		while(str[length] != '\0') length++;

		// Lưu ma trận
		char Mat[101][21];

	    // Lưu toạ độ hàng và cột đang đứng
		int col = 0, row = 0;

		// biến thiên chỉ số cột: = 1 nếu đi từ trái sang phải
		// ngược lại = -1
		int delta = 1;

		// Duyệt xâu để chuyển xâu thành ma trận
		for(int i = 0; i < length; i++)
		{
			// Ban đầu là đi từ trái sang phải => chỉ số cột tăng
			Mat[row][col] = str[i];
			col += delta;

			// Khi đi đến cuối thì quay về trái
			if(col >= NUM_COL)
			{
				Mat[row][col] = '\0';
				row++;
				Mat[row][col] = '\0';
				delta = -1;
				col += delta;
			}
			else if (col < 0)	// Khi đi đến đầu thì quay sang bên phải
			{
				row++;
				delta = 1;
				col += delta;
			}
		}

		// Tìm ra kết quả
		int count = 0;
		for(int j = 0; j < NUM_COL; j++)
		{
			for(int i = 0; i < row; i++)
				str[count++] = Mat[i][j];
		}
		str[count] = '\0';
		cout << str << endl;
	}
	return 0;
}

while(1):
    # Nhap so luong cot
    num_col = int(raw_input())
    if num_col == 0:
        break

    # Xau dau vao
    str_input = raw_input()

    # Tinh so luong cot
    num_row = len(str_input) / num_col;

    # Chuyen xau dau vao thanh dang ma tran
    # va luu dang xau binh thuong
    string = ""
    for i in range(0, num_row):
        if i % 2 == 0:
            string += str_input[i*num_col : (i+1)*num_col]
        else:
            string += (str_input[i*num_col : (i+1)*num_col])[::-1]# dao nguoc xau

    # xau ket qua
    result = ""
    for j in range(0, num_col):
        for i in range(0, num_row):
            result += string[i*num_col + j]

    print result