51NOD-1082-与7无关的数

1082 与7无关的数

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1082

题目

一个正整数，如果它能被7整除，或者它的十进制表示法中某个位数上的数字为7，则称其为与7相关的数。求所有小于等于N的与7无关的正整数的平方和。
例如：N = 8，<= 8与7无关的数包括：1 2 3 4 5 6 8，平方和为：155。
Input
第1行：一个数T，表示后面用作输入测试的数的数量。（1 <= T <= 1000)
第2 - T + 1行：每行1个数N。（1 <= N <= 10^6)
Output
共T行，每行一个数，对应T个测试的计算结果。
Input示例
5
4
5
6
7
8
Output示例
30
55
91
91
155

题解：提前预处理10^6以内的与7相关的数。用longlong和平方和公式计算1-N的平方和，然后减去比N小的与7相关的数即可。求平方和需要特殊处理，不然会爆longlong。

AC代码：

#include<iostream>using namespace std;long long n;long long ans;int k[1000007];int t;int d = 0;int judge(long long x){    while(x>0)    {        long long y = x%10;        if(y==7)        {            //cout<<x<<endl;            return 1;        }        x/=10;    }    return 0;}void intc(){    for(long long i=1;i<1000000;i++)    {        if(i%7==0)        {            k[d++]=i;        }        else if(judge(i))        {            k[d++]=i;        }    }}int main(){    cin>>t;    intc();    while(t--)    {        cin>>n;        if(n&1)ans = (2*n+1)*(n+1)/2*n/3;        else if(n%6==0)ans = n/6*(n+1)*(2*n+1);        else ans = (2*n+1)*(n+1)/3*(n/2);        for(int i=0;n>=k[i];i++)        {            ans-=(long long)k[i]*k[i];        }        cout<<ans<<endl;    }    return 0;}