C++特別數的和（列舉）

小明對數位中含有 2、0、1、9 的數字很感興趣（不包括前導 0），在 1 到 40 中這樣的數包括 1、2、9、10 至 32、39 和 40，共 28 個，他們的和是 574。
請問，在 1到 n中，所有這樣的數的和是多少？
輸入格式
共一行，包含一個整數 n。
輸出格式
共一行，包含一個整數，表示滿足條件的數的和。
資料範圍
1≤n≤10000
輸入樣例：
40
輸出樣例：
574

由於本題資料範圍非常小，進行常規的列舉和拆分數字就行了。

AC程式碼：

#include<iostream>

using namespace std;

int n;
long long ans=0;

bool check(int x)
{
    int d=x,t;
    bool flag=false;
    while(d)
    {
        t=d%10;
        if(t==2||t==0||t==1||t==9)
        {
            flag=true;
            break;
        }
        d/=10;
    }
    return flag;
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;++i)
        if(check(i)) ans+=i;
    cout<<ans;
    return 0;
}