2-2 查找字符

题目：编写函数在给定的字符串中，查找给定的字符。若查找成功，函数返回该字符的地址（指针）；若查找失败，则返回的指针为NULL。
要求：编写主函数测试该函数，在主函数中输入源字符串和要查找的字符，如果找到，显示字符在源字符串中的位置序号；如果找不到，显示“not found”。

样例1：查找成功的情况

• 输入(第一个参数是源字符串，第二个是要查找的字符)：

  ABCDEFG
  D
  

• 输出：

  4
  

样例2：查找不成功的情况

• 输入：

  ABCDEFG
  V
  

• 输出：

  not found
  

#include<iostream>
#include<cstring>
using namespace std;

int main(void){
    string s;
    cin >> s;
    char key;
    int flag=0;
    cin >> key;
    for(int i=0;i<s.length();i++){
        if(s[i]==key){
            cout << i+1;
            flag=1;
            break;
        }
    }
    if(flag==0) cout << "not found";
}