习题11-5 指定位置输出字符串 (20分) --- PTA

本题要求实现一个函数，对给定的一个字符串和两个字符，打印出给定字符串中从与第一个字符匹配的位置开始到与第二个字符匹配的位置之间的所有字符。

函数接口定义：

char *match( char *s, char ch1, char ch2 );

函数match应打印s中从ch1到ch2之间的所有字符，并且返回ch1的地址。

裁判测试程序样例：

#include <stdio.h>

#define MAXS 10

char *match( char *s, char ch1, char ch2 );

int main()
{
    char str[MAXS], ch_start, ch_end, *p;

    scanf("%s\n", str);
    scanf("%c %c", &ch_start, &ch_end);
    p = match(str, ch_start, ch_end);
    printf("%s\n", p);

    return 0;
}

/* 你的代码将被嵌在这里 */

输入样例1：

program
r g

输出样例1：

rog
rogram

我的答案：

char *match( char *s, char ch1, char ch2 )
{
    char *p = NULL;
    int flag = 0;
    while(*s != '\0')
    {
        if (*s == ch1)
        {
            p = s;
            flag = 1;
            break;
        }
        s++;
    }
    if (flag == 1)
    {
        s = p;
        while(*s != '\0')
        {
            if (*s != ch2)
            {
            	printf("%c", *s);
			} else {
				printf("%c", ch2);
				break;
			}
			
            s++;
        }
		printf("\n"); 
		return p;
    } else {
        printf("\n");
    	return  s;
	}   
}

其实我觉得这题干有点bug，为什么ch1没找到要返回s最后’\0’的地址而不是返回空或者s的起始地址，题目也没有说清楚；
格式也没说清楚怎么输出，最后是边提交边改格式，加了几个莫名其妙的换行才结果正确。