ACM模版

描述

题解

这道题如果暴力解题，超时是一定的，所以需要用到打表来解决。这里需要注意的是数据溢出问题，开始时因为忽略了这个问题，i * i时发生了数据溢出，所以导致大数据测试数据系数WA，找了好久才找到问题所在，于是将i定义为long long后成功AC了。

代码

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int MAXN = 1e6 + 7;

bool A[MAXN];
long long sum[MAXN];

void debug()
{
    for (int i = 0; i < MAXN; i++)
    {
        printf("%4d %4d\n", i, A[i]);
    }
    return ;
}

void init()
{
    memset(A, true, sizeof(A));
    // 筛除7的倍数
    for (int i = 7; i < MAXN; i += 7)
    {
        A[i] = false;
    }
    // 筛选数位含7
    int radix;
    for (int i = 1; i < MAXN; i *= 10)
    {
        radix = i * 10;
        for (int j = i * 7; j < MAXN; j += radix)
        {
            for (int k = 0; k < i; k++)
            {
                A[k + j] = false;
            }
        }
    }
    // 调试
// debug();
    return ;
}

void solve()
{
    sum[0] = 0;
    sum[1] = 1;
    for (long long i = 2; i < MAXN; i++)    // 这里的必须是long long
    {
        if (A[i])
        {
            sum[i] = sum[i - 1] + i * i;    // 因为这里i * i，所以要考虑到溢出
        }
        else
        {
            sum[i] = sum[i - 1];
        }
    }
    return ;
}

int main(int argc, const char * argv[])
{
    init();
    solve();

// freopen("input.txt", "r", stdin);

    int T, N;
    cin >> T;

    while (T--)
    {
        scanf("%d", &N);
        printf("%lld\n", sum[N]);
    }

    return 0;
}