L1-033 出生年

題目：

以上是新浪微博中一奇葩貼：“我出生於1988年，直到25歲才遇到4個數字都不相同的年份。”也就是說，直到2013年才達到“4個數字都不相同”的要求。本題請你根據要求，自動填充“我出生於y年，直到x歲才遇到n個數字都不相同的年份”這句話。

輸入格式：

輸入在一行中給出出生年份y和目標年份中不同數字的個數n，其中y在[1, 3000]之間，n可以是2、或3、或4。註意不足4位的年份要在前面補零，例如公元1年被認為是0001年，有2個不同的數字0和1。

輸出格式：

根據輸入，輸出x和能達到要求的年份。數字間以1個空格分隔，行首尾不得有多餘空格。年份要按4位輸出。註意：所謂“n個數字都不相同”是指不同的數字正好是n個。如“2013”被視為滿足“4位數字都不同”的條件，但不被視為滿足2位或3位數字不同的條件。

輸入樣例1：

1988 4

輸出樣例1：

25 2013

輸入樣例2：

1 2

輸出樣例2：

0 0001

上代碼：

#include <string>
#include <iostream>
using namespace std;
bool judge(int s,int n)
{
    string str=to_string(s);
    while(str.length()<4){
        str="0"+str;
    }
    int count=0;
    int arr[10]={0};
    for(int i=0;i<str.length();i++)
    {
        arr[str[i]-'0']++;
    }
    for(int i:arr){
        if(i!=0)
            count++;
    }
    if(count==n)
        return true;
    else 
        return false;
}
int main() {
    int n,s;
    int count=0;
    cin>>s>>n;
    while(!judge(s,n))
    {
        s++; count++;
    }
    printf("%d %04d",count,s);
    return 0;
}

 自己寫的一遍過感覺最爽了。