寻找比目标字母大的最小字母（YYOJ）

题目描述

LZY又来搞字符串了，他现在给你一个只包含小写字母的有序字符串str 和一个目标字母 target，寻找有序数组里面比目标字母大的最小字母。 里字母的顺序是循环的。举个例子，如果目标字母target = ‘z’ 并且有序字符串为“ab”，则答案返回 'a’。

输入

测试样例由多组测试数据组成。每组测试数据第一行输入一个字符串str ( 1 <= str.length <= 2600000 ）和一份目标字符 c ( ‘a’ <= c <= ‘z’ ）

输出

输出寻找到的比目标字母大的最小字母

样例输入

cfj a
cfj c
cfj j

样例输出

c
f
c

提示

由于数据过大，建议使用scanf,printf进行输入输出

AC代码

#include<bits/stdc++.h>
using namespace std;
int main(){
 ios::sync_with_stdio(false);
 string s;
 char a, b;
 while(cin >> s >> a){
  b = 'z';
  int left = 0;
  int right = s.length();
  int flag = 0;
  while(left <= right){
   int mid = (left + right) / 2;
   if(s[mid] > a){
    flag = 1;
    b = min(b, s[mid]);
    right = mid - 1;
   }else{
    left = mid + 1;
   }
  }
  if(flag == 0)cout << s[0] << endl;
  else cout << b << endl;
 }
    return 0;
}

思路
一开始为b赋值为最大的 ‘ z ’ ，每次将头尾和除以二为mid，并判断是s[mid]是否小于b，如果小于改变b的值。
flag用来判断是否找到比输入的字符a大。

超时代码

#include<bits/stdc++.h>
using namespace std;
int main(){
 string str;
 char ch;
 while(cin>>str >>ch){
  int flag=0;
  for(char i=ch+1;i<='z';i++){
   int top=0;
   int botton=str.length()-1;
   while(top<=botton){
    int mid=(top+botton)/2;
    if(str[mid]==i){
     cout<<i<<endl;
     flag=1;
     break;
    }
    if(str[mid]>i){
     botton=mid-1;
    }
    if(str[mid]<i){
     top=mid+1;
    }
   }
   if(flag==1)
    break; 
  }
  if(flag==0)
   cout<<str[0]<<endl;
 }
 return 0;
}