encode string C++

给一个string, 让把里面连续的超过3个的相同char替换成char#个数。少于3个的不变。
比如：AAABCCDDDDEEF
变成：A#3BCCD#4EEF
---------
挺简单的。。

0. string encode(string &input){  
1.   string output;  
2.   if(input.empty()) return output;  
3.   int count = 1;  
4.   int i;  
5.   for(i = 0; i < input.length() - 1; i++){  
6.     if(input[i] != input[i+1]){  
7.           if(count < 3){  
8.               while(count-- > 0)  
9.               output.push_back(input[i]);               
10.          }  
11.          else{  
12.              output.push_back(input[i]);  
13.              output.push_back('#');  
14.              output.push_back(to_string(count));  
15.          }  
16.         count = 1;  
17.     }  
18.   else count++;  
19.  }  
20.   
21. output.push_back(input[i]);  
22. if(count > 1){  
23.   output.push_back('#');  
24.   output.push_back(to_string(count));  
25. }  
26. if(count == 2) output.push_back(input[i]);  
27.   
28. return output;  
29. }