SRM 114 DIV2 [550]

Topcoder的算法题，有兴趣的可以自己做做   PS：突然发现格式很丑，懒得弄了，不影响阅读 Problem Statement
	Let's say you have a binary string such as the following: 011100011 One way to encrypt this string is to add to each digit the sum of its adjacent digits. For example, the above string would become: 123210122 In particular, if P is the original string, and Q is the encrypted string, then Q[i] = P[i-1] + P[i] + P[i+1] for all digit positions i. Characters off the left and right edges of the string are treated as zeroes. An encrypted string given to you in this format can be decoded as follows (using 123210122 as an example): Assume P[0] = 0. Because Q[0] = P[0] + P[1] = 0 + P[1] = 1, we know that P[1] = 1. Because Q[1] = P[0] + P[1] + P[2] = 0 + 1 + P[2] = 2, we know that P[2] = 1. Because Q[2] = P[1] + P[2] + P[3] = 1 + 1 + P[3] = 3, we know that P[3] = 1. Repeating these steps gives us P[4] = 0, P[5] = 0, P[6] = 0, P[7] = 1, and P[8] = 1. We check our work by noting that Q[8] = P[7] + P[8] = 1 + 1 = 2. Since this equation works out, we are finished, and we have recovered one possible original string. Now we repeat the process, assuming the opposite about P[0]: Assume P[0] = 1. Because Q[0] = P[0] + P[1] = 1 + P[1] = 0, we know that P[1] = 0. Because Q[1] = P[0] + P[1] + P[2] = 1 + 0 + P[2] = 2, we know that P[2] = 1. Now note that Q[2] = P[1] + P[2] + P[3] = 0 + 1 + P[3] = 3, which leads us to the conclusion that P[3] = 2. However, this violates the fact that each character in the original string must be '0' or '1'. Therefore, there exists no such original string P where the first digit is '1'. Note that this algorithm produces at most two decodings for any given encrypted string. There can never be more than one possible way to decode a string once the first binary digit is set. Given a string message, containing the encrypted string, return a vector <string> with exactly two elements. The first element should contain the decrypted string assuming the first character is '0'; the second element should assume the first character is '1'. If one of the tests fails, return the string "NONE" in its place. For the above example, you should return {"011100011", "NONE"}.
Definition
	Class: BinaryCode Method: decode Parameters: string Returns: vector <string> Method signature: vector <string> decode(string message) (be sure your method is public)
Constraints
-	message will contain between 1 and 50 characters, inclusive.
-	Each character in message will be either '0', '1', '2', or '3'.
Examples
0)
	"123210122" Returns: { "011100011", "NONE" } The example from above.
1)
	"11" Returns: { "01", "10" } We know that one of the digits must be '1', and the other must be '0'. We return both cases.
2)
	"22111" Returns: { "NONE", "11001" } Since the first digit of the encrypted string is '2', the first two digits of the original string must be '1'. Our test fails when we try to assume that P[0] = 0.
3)
	"123210120" Returns: { "NONE", "NONE" } This is the same as the first example, but the rightmost digit has been changed to something inconsistent with the rest of the original string. No solutions are possible.
4)
	"3" Returns: { "NONE", "NONE" }
5)
	"12221112222221112221111111112221111" Returns: { "01101001101101001101001001001101001", "10110010110110010110010010010110010" }

Code:

1. #include<iostream>  
2. #include<vector>  
3. #include<string>  
4. using namespace std;  
5.   
6. class BinaryCode  
7.     {  
8. public:  
9.     vector<string> decode(string message)  
10.         {  
11.             vector<string> temp;  
12.             temp.push_back(message);  
13.             temp.push_back(message);  
14.             temp[0][0]='0';  
15.             temp[1][0]='1';  
16.             int len=message.size();  
17.             if (len==1)  
18.                 switch (message[0]){  
19.             case '0':  
20.                 temp[1].assign("NONE");  
21.                 break;  
22.             case '1':  
23.                 temp[0].assign("NONE");  
24.                 break;  
25.             default:  
26.                 temp[0].assign("NONE");  
27.                 temp[1].assign("NONE");  
28.                 break;  
29.             }  
30.             else{  
31.                 for (int i=1;i!=len;++i){  
32.                     if (i==1){  
33.                         temp[0][i]=message[i-1]-temp[0][i-1]+'0';  
34.                         temp[1][i]=message[i-1]-temp[1][i-1]+'0';  
35.                     }  
36.                     else{  
37.                         temp[0][i]=message[i-1]-temp[0][i-1]-temp[0][i-2]+'0'+'0';  
38.                         temp[1][i]=message[i-1]-temp[1][i-1]-temp[1][i-2]+'0'+'0';  
39.                     }  
40.                 }  
41.                 if (temp[0][len-1]+temp[0][len-2]!=message[len-1]+'0') temp[0].assign("NONE");  
42.                 if (temp[1][len-1]+temp[1][len-2]!=message[len-1]+'0') temp[1].assign("NONE");  
43.                 for (int i=1;i!=len;++i){  
44.                     if (temp[0][i]>'1' || temp[0][i]<'0'){  
45.                         temp[0].assign("NONE");  
46.                     }  
47.                     if (temp[1][i]>'1' || temp[1][i]<'0'){  
48.                         temp[1].assign("NONE");  
49.                     }  
50.                 }  
51.             }  
52.             return temp;  
53.     }  
54. };