Завершение программы 1/0 - C (СИ)

Формулировка задачи:

Нужно сделать так чтобы на 1 программа продолжалась, а на 0 завершалась!

// Функция.cpp: определяет точку входа для консольного приложения.
//
 
#include "stdafx.h"
#include "stdio.h"
#include "conio.h"
#include <locale.h>
#include <iostream>
 
int A[100];
int razm()
{
    int r;
    printf("Введите размерность:");
    scanf("%d", &r);
    return r;
}
 
void vvod(int r)
{
    printf("Вектор:\n");
    for (int i=0;i<r;i++)
    {       
        printf("A[%d]=", i+1);
        scanf("%d", &A[i]);
    }
}
void vivod(int r)
{
    printf("Вывод:\n");
    for (int i=0;i<r;i++)
    {   
        printf("%d ", A[i]);
    }
        
}
 
void main()
{
    setlocale(LC_ALL,"Russian");    
    int r;
    r=0;
    r=razm();
    printf("Размерность = %d\n", r);
    vvod(r);
    vivod(r);
    getch();
}

Решение задачи: «Завершение программы 1/0»

Листинг программы

#include <stdio.h>
 
int main(void) {
    
    int r;
    printf("0-zaverwenu9,1-prodolwenue,drygoe-povtornuy vvod\n");
    for(;;)
    {
        scanf("%d", &r);
        if(!r) 
        {
            printf("0\n");
            return 0;
        }
        else if(r==1)
        {
            printf("1\n");
            break;
        }
        else
        {
            printf("%d\n ", r);
        }
        
    }
    printf("4to-to ewe");
    return 0;
}

Объяснение кода листинга программы

Подключение стандартной библиотеки для работы с консолью (stdio.h).
Объявление переменной типа int с именем r, которая будет использоваться в цикле.
Вывод на экран сообщения с просьбой ввести число от 0 до 9, либо 1.
Бесконечный цикл, который будет выполняться до тех пор, пока пользователь не введет число 0.
Считывание введенного числа с помощью функции scanf и сохранение его в переменной r.
Проверка значения переменной r на равенство 0.
Если r равно 0, то выводится сообщение 0 и программа завершается.
Если r равно 1, то выводится сообщение 1 и цикл прерывается.
Если r не равно 0 или 1, то выводится сообщение с числом r.
После завершения цикла выводится сообщение 4to-to ewe.
Программа возвращает 0, что означает успешное завершение.