Как создать меню с возвратом в консоли?

#include "stdafx.h"
#include <iostream>
#include<string.h>
#include<stdio.h>
#include<stdlib.h>
#include<conio.h>
#define word unsigned int
#define inf 100000
 
using namespace std;
 
int prompt_menu_item()
{
    setlocale(LC_ALL, "Russian");
    cout<<"|---------------------------------------------------------|"<<endl;
    cout<<"|Курсовая работа по теме: \"Поиск кратчайшего пути в графе\"| "<<endl;
    cout<<"|---------------------------------------------------------|\n"<<endl;
    // Выбрать вариант меню...
    int variant;
    cout << "Выберите метод поиска кратчайшего пути в графе...\n" << endl;
    cout << "1. алгоритм Дейкстры\n"
          << "2. алгоритм Флойда\n"
          << "3. алгоритм Форда\n"
          << "4. выйти\n" << endl;
    cout << ">>> ";
    cin >> variant;
    return variant;
}
//переменные для Дейкстры
int i, j, n, p, xn, xk;
int flag[11];
word c[11][11], l[11];
char s[80], path[80][11];
//переменные для Флойда
const int maxV=1000;
int Matrix[maxV][maxV];
//-----------------------------------------------------------
//алгоритм Флойда
void floid(int D[][maxV], int V)
{
    int k;
    for (i=0; i<V; i++) D[i][i]=0;
    
    for (k=0; k<V; k++)
        for (i=0; i<V; i++)
            for (j=0; j<V; j++)
                if (D[i][k] && D[k][j] && i!=j)
                    if (D[i][k]+D[k][j]<D[i][j] || D[i][j]==0)
                        D[i][j]=D[i][k]+D[k][j];
    for (i=0; i<V; i++)
    {
        for (j=0; j<V; j++) cout<<D[i][j]<<"\t";
        cout<<endl;
    }
}
//-------------------------------------------------
//вспомогательная процедура для алгоритма Дейкстры
int min(int n)
{
    int i, result;
    for(i=0;i<n;i++)
        if(!(flag[i])) result=i;
    for(i=0;i<n;i++)
        if((l[result]>l[i])&&(!flag[i])) result=i;
    return result;
}
 
word minim(word x, word y)
{
    if(x<y) return x;
    return y;
}
//-----------------------------------------------
struct Edges{
int u, v, w;
};
const int Vmax=1000;
const int Emax=Vmax*(Vmax-1)/2;
int e, start;
Edges edge[Emax];
int d[Vmax];
//алгоритм Беллмана-Форда
void bellman_ford(int n, int s)
{
int i, j;
for (i=0; i<n; i++) d[i]=inf;
d[s]=0;
for (i=0; i<n-1; i++)
for (j=0; j<e; j++)
if (d[edge[j].v]+edge[j].w<d[edge[j].u])
d[edge[j].u]=d[edge[j].v]+edge[j].w;
for (i=0; i<n; i++) if (d[i]==inf)
cout<<endl<<start<<"->"<<i+1<<"="<<"Not";
else cout<<endl<<start<<"->"<<i+1<<"="<<d[i];
}
//-----------------------------------------------
//-----------------------------------------------
//начало работы программы
int main(int argc, char* argv[])
{
    int variant = prompt_menu_item(); 
 
    switch (variant) 
    {
        case 1:
            {
            cout << "Поиск минимального пути в графе методом Дейкстры" << endl;
            cout<<"Введите количество вершин ";
    cin>>n; 
    for(i=0;i<n;i++)
        for(j=0;j<n;j++) 
            c[i][j]=0;
    for(i=0;i<n;i++)
        for(j=i+1;j<n;j++)
        {
            cout<<"Укажите  растояние от  x"<<i+1<<" до x"<<j+1<<": ";
            cin>>c[i][j];
        }
    cout<<"   ";
    for(i=0;i<n;i++) cout<<"    X"<<i+1;
    cout<<endl<<endl;
    for(i=0;i<n;i++)
    {
        printf("X%d",i+1);
        for(j=0;j<n;j++)
        {
            printf("%6d",c[i][j]);
            c[j][i]=c[i][j];
        }
        printf("\n\n");
    }
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            if(c[i][j]==0) c[i][j]=65535; //бесконечность
    cout<<"Введите начальную точку: ";
    cin>>xn;
    cout<<"Введите конечнную точку: ";
    cin>>xk;
    xk--;
    xn--;
    if(xn==xk)
    {
        cout<<"Начальная и конечная точки совподают."<<endl;
        _getche();
        return 0;
    }
    //------proba---------
    if (xk>n)
    {
        cout<<"Такой конечной точки не существует."<<endl;
        _getche();
        return 0;
    }
 
    for(i=0;i<n;i++)
    {
        flag[i]=0;
        l[i]=65535;
    }
    l[xn]=0;
    flag[xn]=1;
    p=xn;
    itoa(xn+1,s,10);
        for(i=1;i<=n;i++)
        {
            strcpy(path[i],"X");
            strcat(path[i],s);
        }
        do
        {
            for(i=0;i<n;i++)
                if((c[p][i]!=65535)&&(!flag[i])&&(i!=p))
                {
                    if(l[i]>l[p]+c[p][i])
                    {
                        itoa(i+1,s,10);
                        strcpy(path[i+1],path[p+1]);
                        strcat(path[i+1],"-X");
                        strcat(path[i+1],s);
                    }
                    l[i]=minim(l[i],l[p]+c[p][i]);
                }
            p=min(n);
            flag[p]=1;
        }
        while(p!=xk);
    if(l[p]!=65535)
    {
        cout<<"Путь: "<<path[p+1]<<endl;
        cout<<"Длинна пути: "<<l[p]<<endl;
    }
    else
        cout<<"Путь не существует!"<<endl;
    _getche();
            }
//-----------------------конец дейкстры-------------------------
            break;
        case 2:
            {
            cout << "Поиск минимального пути в графе методом Флойда" << endl;
//для Флойда
{
    cout<<"Введите количество вершин в графе > "; 
    cin>>n;
    cout<<"Введите матрицу весов ребер:\n";
    for (i=0; i<n; i++)
        for (j=0; j<n; j++)
        {
            cout<<"Введите вес ребера графа ["<<i+1<<"]["<<j+1<<"] > ";
            cin>>Matrix[i][j];
        }
        cout<<"Матрица кратчайших путей:"<<endl;
        floid (Matrix, n);
        getch();
}
               }
//-----------------------конец Флойда-------------------------
            break;
        case 3:
            {
            cout << "Поиск минимального пути в графе методом Форда" << endl;
            {
int w;
cout<<"Количество вершин > "; cin>>n;
e=0;
for (i=0; i<n; i++)
for (j=0; j<n; j++)
{
cout<<"Вес "<<i+1<<"->"<<j+1<<" > "; cin>>w;
if (w!=0)
{
edge[e].v=i;
edge[e].u=j;
edge[e].w=w;
e++;
}
}
cout<<"Стартовая вершина > "; cin>>start;
cout<<"Список кратчайших путей:";
bellman_ford(n, start-1);
system("pause>>void");
}
            }
            break;
        case 4:
            {
            cout << "Выход из программы..." << endl;
            exit(EXIT_SUCCESS);
            }
            break;
        default:
            cerr << "Вы выбрали неверный вариант.\nНажмите любую клавишу для выхода из программы" << endl;
            getch();
    }
    return 0;
}

int main()
{
    ...
    while(true)
    {
        switch (prompt_menu_item()) 
        {
             ...
        }
    }
}

#include <iostream>
 
using namespace std;
void floyd()
{
    cout << "1";
}
void ford()
{
    cout << "2";
}
//******
 
int main()
{
    char ch = ' ';
    while(ch != '4')
    {
        switch(ch)
        {
        case '1': floyd();
                break;
        case '2': ford();
                break;
        default: cout << "net takogo pynkta v menu\n";
        }
 
        cout << "Выберите метод поиска кратчайшего пути в графе...\n" << endl;
        cout << "1. алгоритм Дейкстры\n"
          << "2. алгоритм Флойда\n"
          << "3. алгоритм Форда\n"
          << "4. выйти\n" << endl;
        cin >> ch;
    }
    return 0;
}