Print U Shaped String

uuuouou

就是确定竖线和横线的长度，需要注意的是，算竖线长度的时候，包括与横线的交点，因为这个WA了2次。
样例输入中helloworld少了个l，@管理员

1. class Solution {
   
2. public:
   
3.     vector<string> uFormat(string &s) {
   
4.         int n = s.size();
   
5.         if(n <= 3) return vector<string>({s});
   
6. 
   
7.         int a = n/3, r = n%3, b;
   
8.         if(r == 0){
   
9.             b = a;
   
10.         }
    
11.         else{
    
12.             b = n-a*2;
    
13.         }
    
14.         if(b < 3 || b <= a){
    
15.             --a;
    
16.             b += 2;
    
17.         }
    
18.         
    
19.         string padding(b-2, ' ');
    
20.         vector<string> res(a+1);
    
21.         for(int i = 0; i < a; ++i){
    
22.             res【i】 = s【i】 + padding + s[n-1-i];
    
23.         }
    
24.         res[a] = s.substr(a, b);
    
25.         return res;
    
26.     }
    
27. };
    

复制代码