Simple 6 line solution


#1

vector<vector>v((A.size()*2)-1);
for(int i=0;i<A.size();i++){
for(int j=0;j<A.size();j++){
v[i+j].push_back(A[i][j]);
}
}
return v;ere.