ZOJ1006 Do the Untwist

Cryptography deals with methods of secret communication that transform a message (the plaintext) into a disguised form (the ciphertext) so that no one seeing the ciphertext will be able to figure out the plaintext except the intended recipient. Transforming the plaintext to the ciphertext is encryption; transforming the ciphertext to the plaintext is decryption. Twisting is a simple encryption method that requires that the sender and recipient both agree on a secret key k, which is a positive integer.

The twisting method uses four arrays: plaintext and ciphertext are arrays of characters, and plaincode and ciphercode are arrays of integers. All arrays are of length n, where n is the length of the message to be encrypted. Arrays are origin zero, so the elements are numbered from 0 to n - 1. For this problem all messages will contain only lowercase letters, the period, and the underscore (representing a space).

The message to be encrypted is stored in plaintext. Given a key k, the encryption method works as follows. First convert the letters in plaintext to integer codes in plaincode according to the following rule: '_' = 0, 'a' = 1, 'b' = 2, ..., 'z' = 26, and '.' = 27. Next, convert each code in plaincode to an encrypted code in ciphercode according to the following formula: for all i from 0 to n - 1,

ciphercode[i] = (plaincode[ki mod n] - i) mod 28.
(Here x mod y is the positive remainder when x is divided by y. For example, 3 mod 7 = 3, 22 mod 8 = 6, and -1 mod 28 = 27. You can use the C '%' operator or Pascal 'mod' operator to compute this as long as you add y if the result is negative.) Finally, convert the codes in ciphercode back to letters in ciphertext according to the rule listed above. The final twisted message is in ciphertext. Twisting the message cat using the key 5 yields the following:

Array 0 1 2
plaintext 'c' 'a' 't'
plaincode 3 1 20
ciphercode 3 19 27
ciphertext 'c' 's' '.'
Your task is to write a program that can untwist messages, i.e., convert the ciphertext back to the original plaintext given the key k. For example, given the key 5 and ciphertext 'cs.', your program must output the plaintext 'cat'.

The input file contains one or more test cases, followed by a line containing only the number 0 that signals the end of the file. Each test case is on a line by itself and consists of the key k, a space, and then a twisted message containing at least one and at most 70 characters. The key k will be a positive integer not greater than 300. For each test case, output the untwisted message on a line by itself.

Note: you can assume that untwisting a message always yields a unique result. (For those of you with some knowledge of basic number theory or abstract algebra, this will be the case provided that the greatest common divisor of the key k and length n is 1, which it will be for all test cases.)

Example input:

5 cs.
101 thqqxw.lui.qswer
3 b_ylxmhzjsys.virpbkr
0
Example output:

cat
this_is_a_secret
beware._dogs_barking
Source: Zhejiang University Local Contest 2001

import java.util.Scanner;


/**  
 * Do the Untwist
 * 
 * This class is to do decryption from twist
 * 
 * Twist is an encryption algorithm
 *  
 * @author Emma Zhao
 *
 * 2011-7-3 下午07:49:46
 */
public class Main{
	/**
	 * to identify the length of each array
	 */
	private int[] plaincode = new int[70];
	private char[] plaintext;
	private int[] cliphercode = new int[70];
	private char[] cliphertext;
	/**
	 * each index of array code represents the item in it
	 */
	private static char[] code = {'_','a','b','c','d','e','f','g','h','i','j','k',
									'l','m','n','o','p','q','r','s','t','u','v','w','x','y','z','.'};
	
	public char[] decryption(int k,String s){
		int n = s.length();
		cliphertext = s.toCharArray();
		plaintext = new char[n];
		
		for(int i = 0;i<n;i++){
			if('.' == cliphertext[i]){
				cliphercode[i] = 27;
			}else if('_' == cliphertext[i])
				cliphercode[i] = 0;
			else
				cliphercode[i] = cliphertext[i] -96;
		}
		
		int temp,count;
		for(int i = 0;i<n;i++){
			count = 0;
			temp = cliphercode[i] + 28 * count + i;
			while(!(temp>=0 && temp <= 27)){
				if(temp > 27)
					count--;
				if(temp<0)
					count++;
				temp = cliphercode[i] + 28 * count + i;
			}
			plaincode[k*i%n] = temp;
			plaintext[k*i%n] = code[plaincode[k*i%n]];

		}
		
		
		return plaintext;
	}
	
	public static void main(String[] args){
		Main m = new Main();
		Scanner scanner = new Scanner(System.in);
		String temp = scanner.nextLine();
		int k;
		while(!temp.equals("0")){
			k = Integer.parseInt(temp.split(" ")[0]);
			String s = temp.split(" ")[1].trim();
			char[] plaintext = m.decryption(k, s);
			for(int i = 0;i< plaintext.length;i++){
				System.out.print(plaintext[i]);
			}
			temp = scanner.nextLine();
			System.out.println();
		}
	}
}