Lucky Substring(CodeForces - 122B)

Input
047
Output
4
Input
16
Output
-1
Input
472747
Output
7

題意:給出的字符串的子串滿足下面3個條件,輸出字典序最小的那個。(1)只含有4或7(2)子串中個數多的優先(3)不能有前導0

思路:因爲n<50,所以可以先找出所有字串,然後排序,之後輸出滿足條件的第一個。關鍵就是如何存儲那裏,我是先用S數組存出所有字串,順帶寫個map標記一下個數,最後就是cmp排序函數了。

#include <iostream>
#include<cstring>
#include<stdio.h>
#include<algorithm>
#include <vector>
#include<iomanip>
#include <queue>
#include<stdlib.h>
#include<time.h>
#include<map>
#include<stddef.h>
using namespace std;

string S[101010];
map<string,int> ma;
bool cmp(string p,string q){
    if(ma[p]==ma[q])
        return p < q;
    return ma[p]>ma[q];
}
int main()
{
    string s;
    cin>>s;
    int n=0;
    for(int i=0;i<s.size();i++){
        for(int j=1;i+j<=s.size();j++){
          S[n++]=s.substr(i,j);
          ma[s.substr(i,j)]++;
        }
    }
    sort(S,S+n,cmp);
    int flag=0;
    int i,j;
    string ans;
    for(int i=0;i<n;i++){
       // cout<<S[i]<<endl;
        for(j=0;j<S[i].size();j++){
            if(S[i][j]!='4'&&S[i][j]!='7')
                break;
            if(S[i][0]=='0')
                break;
        }
        if(j>=S[i].size()){
            flag=1;
            //cout<<"***"<<endl;
            ans=S[i];
            break;
        }
    }
    if(flag==1)
        cout<<ans<<endl;
    else
        cout<<"-1"<<endl;
    return 0;
}

 

發佈了350 篇原創文章 · 獲贊 717 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章