1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
| class Solution { public: int buildTransferStation(vector<vector<int>>& area) { vector<int> xx,yy; for(int i=0;i<area.size();++i){ for(int j=0;j<area[i].size();++j){ if(area[i][j]==1){ xx.push_back(i); yy.push_back(j); } } } sort(xx.begin(),xx.end());sort(yy.begin(),yy.end()); int x=xx[xx.size()/2],y=yy[yy.size()/2]; cout<<x<<" "<<y<<"\n"; int ans=0; for(int i=0;i<area.size();++i){ for(int j=0;j<area[i].size();++j){ if(area[i][j]==1){ ans+=abs(i-x)+abs(j-y); } } } return ans; } };
|