Zgrade n stanovi n stuff

#include <iostream>

using namespace std;

void main(){
    int u,s;
    cout << "Unesi broj ulaza zatim broj spratova:\n";
    do{
        cin >> u >> s;//3,5
        if (!(u >= 1 && s >= 1))
            cout << "Zgrada mora imati bar po 1 ulaz i 1 sprat\n";
    } while (u < 1 || s < 1);
    cout << "Unesi broj stana za koji zelis ispis susjednih stanova: "; //novo
    int z;  //novo
    cin >> z; //novo
    int b=0;
    cout << endl << "Zgrada ima " << s << " spratova, i na svakom spratu " << u << " ulaza.\n";
    for (int i = 1; i <= s; i++){
        for (int j = 1; j <= u; j++){
            b++;    //novo
            if (z == b){    //novo
                cout << "Stan broj " << b << ". ima susjede: "; //novo ++b prebaceno u samo b
                if (i == 1){    //prvi sprat
                    if (j == 1){    //prvi stan
                        cout << b + 1 << ", " << b + u << endl;
                    }
                    else if (j == u){   //zadnji stan
                        cout << b - 1 << ", " << b + u << endl;
                    }
                    else{   //stanovi izmedju
                        cout << b - 1 << ", " << b + 1 << ", " << b + u << endl;
                    }
                }
                else if (i == s){   //zadnji sprat
                    if (j == 1){
                        cout << b + 1 << ", " << b - u << endl;
                    }
                    else if (j == u){
                        cout << b - 1 << ", " << b - u << endl;
                    }
                    else{
                        cout << b - 1 << ", " << b + 1 << ", " << b - u << endl;
                    }
                }
                else{   //spratovi izmedju
                    if (j == 1){
                        cout << b + 1 << ", " << b - u << ", " << b + u << endl;
                    }
                    else if (j == u){
                        cout << b - 1 << ", " << b - u << ", " << b + u << endl;
                    }
                    else{
                        cout << b - 1 << ", " << b + 1 << ", " << b - u << ", " << b + u << endl;
                    }
                }
            } //novo
        }
    }
    system("pause>0");
}