LeetCode Spiral Matrix II a little modification required in Spiral Matrix done earlier

LeetCode Spiral Matrix II a little modification required in Spiral Matrix done earlier

class Solution {

public:

    vector<vector<int>> generateMatrix(int n) {

      vector < int > temp(n,0);

      vector < vector< int >  > ans(n,temp);

      int s1,s2;

      int e1=n,e2=n;

      int i=0,j=0,cnt=1;

      n=n*n;

     while(cnt<=n){

      s1=i;s2=j;

      while(s2<e2){

            ans[s1][s2]=cnt;

          s2++;cnt++;

      }

      if(cnt>n)

           break;

      s1=i+1;s2=e2-1;

      while(s1<e1){

           ans[s1][s2]=cnt;

           s1++;cnt++;

      }

      if(cnt>n)

           break;

      s1=e1-1;s2=e2-2;

      while(s2>=j){

          ans[s1][s2]=cnt;

          s2--;cnt++;

      }

      if(cnt>n)

           break;

      s1=e1-2;s2=j;

      while(s1>i){

          ans[s1][s2]=cnt;

          s1--;cnt++;

      }

      if(cnt>n)

           break;

      i=i+1;j=j+1;

      e1=e1-1;e2=e2-1;

     }

     return ans;

    }

};