AcWing 3734. 求和 DFS枚举 区间优化处理

题目

链接

题解思路

策略就是枚举到符合1e9的数，这样需要搜到10位，枚举完排序，对应区间进行优化，暴力会TLE，枚举数到目标数可以直接做乘法乘数量来减少枚举的数量。

数据类型记得longlong int 直接爆。

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

long long   a[1300000];
int p;
void dfs(int step, long long  s)
{
    if (step == 10 )
    {
        a[p] = s;
        p++;
        return ;
    }
    if (step != 0 )
    {
        a[p] = s;
        p++;
    }
    dfs(step+1,s*10+4);
    dfs(step+1,s*10+7);
}
int main ()
{

    int l,r;
    cin>>l>>r;
    dfs(0,0);
    sort(a,a+p);
    long long ans = 0 ;
    int pit = 0;
    for (long long  i = l ; i <= r ; i=a[pit]+1 )
    {
        while( i > a[pit])
            pit++;
        if ( r > a[pit] )
        {
            ans += (a[pit]-i+1)*a[pit];
        }else
        {
            ans += (r-i+1)*a[pit];
        }
    }
    cout<<ans<<"\n";
    return 0;
}