Eliminación de caracteres

Descripción del problema:
escriba un programa, primero ingrese una cadena str (la longitud no exceda 20), y luego ingrese un solo carácter ch, y luego el programa borrará todos los caracteres ch que aparecen en la cadena str para obtener uno nuevo La cadena str2, y luego imprima esta cadena.

Formato de entrada:
hay dos líneas de entrada, la primera línea es una cadena (sin espacios en el interior) y la segunda línea es un carácter.

Formato de salida:
cadena después del procesamiento.
Entrada y salida de muestra

Entrada de muestra:
123 $ 45 $ 678
$

Salida de muestra:
12345678

c código de referencia:

#include <stdio.h>
#include <string.h>

char str[21];
char str2[21];
char ch;

int main()
{
    
    
    gets(str);
	scanf("%c",&ch);
	
	int i,j=0;
	for(i=0;i<strlen(str);i++)
	{
    
    
		if(str[i]==ch)
		 continue;
		str2[j++]=str[i];
    }
    
    for(i=0;i<strlen(str2);i++)
     printf("%c",str2[i]);
	return 0;
}

Código de referencia de C ++:

#include <iostream>
#include <string>
#include <cstring>

using namespace std;

char str[21];
char str2[21];
char ch;

int main()
{
    
    
    gets(str);
	
	scanf("%c",&ch);
	
	int j=0;
	for(int i=0;i<strlen(str);i++)
	{
    
    
		if(str[i]!=ch)
		 str2[j++]=str[i];
    }
    
    cout<<str2<<endl;
	return 0;
}