求教Airbnb的CSV parser

在magicsets老哥的基础上改进了一下，原本的test input最后一行1后面的\n好像是个typo.
而且这里省掉了一个状态，算是一个简化。

1. class Solution {
   
2. 
   
3.     public static void main(String[] args) {
   
4.         String doc =
   
5.                 "John,Smith,[email]john.smith@gmail.com[/email],Los Angeles,1\n" +
   
6.                         "Jane,Roberts,[email]janer@msn.com[/email],\"San Francisco, CA\",0\n" +
   
7.                         "\"Alexandra \"\"Alex\"\"\",Menendez,[email]alex.menendez@gmail.com[/email],Miami,1 " +
   
8.                         "\"\"\"Alexandra, Alex\"\"\"";
   
9. 
   
10.         System.out.println(CSVParser.Convert(doc));
    
11.     }
    
12. 
    
13.     static class CSVParser {
    
14.         private enum State {
    
15.             FIELD_START,
    
16.             QUOTE_START,
    
17.             QUOTE_IN_QUOTE
    
18.         }
    
19. 
    
20.         public static String Convert(String doc) {
    
21.             StringBuilder sb = new StringBuilder();
    
22.             State state = State.FIELD_START;
    
23. 
    
24.             for (int i = 0; i < doc.length(); ++i) {
    
25.                 char ch = doc.charAt(i);
    
26. 
    
27.                 switch (state) {
    
28.                     case FIELD_START:
    
29.                         if (ch == '"') {
    
30.                             state = State.QUOTE_START;
    
31.                             continue;
    
32.                         } else if (ch == ',') {
    
33.                             ch = '|';
    
34.                         }
    
35.                         break;
    
36.                     case QUOTE_START:
    
37.                         if (ch == '"') {
    
38.                             state = State.QUOTE_IN_QUOTE;
    
39.                             continue;
    
40.                         }
    
41.                         break;
    
42.                     case QUOTE_IN_QUOTE:
    
43.                         switch (ch) {
    
44.                             case '"': state = State.QUOTE_START; break;
    
45.                             case ',': state = State.FIELD_START; ch = '|'; break;
    
46.                             default:  Error(i, ch);
    
47.                         }
    
48.                         break;
    
49.                 }
    
50.                 sb.append(ch);
    
51.             }
    
52. 
    
53.             return sb.toString();
    
54.         }
    
55. 
    
56.         private static void Error(int position, char ch) {
    
57.             throw new RuntimeException(
    
58.                     "Invalid character " + ch + " at position " + position);
    
59.         }
    
60.     }
    
61. }
    

复制代码

今天国内的WorldQuant也问这个题了，就15分钟没写出来……来学习一下