第十屆藍橋杯省賽C++A組 平方和

試題 A: 平方和

本題總分：5 分

【問題描述】

        小明對數位中含有 2、0、1、9 的數字很感興趣，在 1 到 40 中這樣的數包括 1、2、9、10 至 32、39 和 40，共 28 個，他們的和是 574，平方和是 14362。注意，平方和是指将每個數分别平方後求和。   

        請問，在 1 到 2019 中，所有這樣的數的平方和是多少？

【答案送出】

        這是一道結果填空的題，你隻需要算出結果後送出即可。本題的結果為一 個整數，在送出答案時隻填寫這個整數，填寫多餘的内容将無法得分。

        提示：如果你編寫程式計算，發現結果是負的，請仔細檢查自己的程式， 不要懷疑考場的程式設計軟體。

答案：2658417853

思路：這道題和B組的第6題十分類似，隻是多了一個求平方。

代碼：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;

bool check(long long n)
{
	while(n)
	{
		int t=n%10;
		if(t==2||t==0||t==1||t==9)
			return true;
		n/=10;
	}
	return false;
}

int main()
{
	long long ans=0;
	for(long long i=1;i<=2019;i++)
	{
		if(check(i))
			ans+=i*i;
	}
	cout<<ans<<endl;
	return 0;
}