爱奇艺-笔试刷题

题目描述：

/**
牛牛选择了一个正整数X,然后把它写在黑板上。
然后每一天他会擦掉当前数字的最后一位,直到他擦掉所有数位。
在整个过程中,牛牛会把所有在黑板上出现过的数字记录下来,
然后求出他们的总和sum.

例如X = 509, 在黑板上出现过的数字依次是509, 50, 5, 他们的和就是564.
牛牛现在给出一个sum,牛牛想让你求出一个正整数X经过上述过程的结果是sum.

输入描述:
输入包括正整数sum(1 ≤ sum ≤ 10^18)

输出描述:
输出一个正整数,即满足条件的X,如果没有这样的X,输出-1。

输入例子1:
564

输出例子1:
509
*/

思路如下：

对于一个数x, 是digit位
那么其对应的sum为x+x/10+x/100+...+(x/(10^(digit-1)))
那么一个sum题目现在最多是19位
那么可以枚举digit，对于每一个可能的digit根据sum，反过来求一个xx
比如sum/(111..1)一个digit个1得到的就是第一位
同理更新sum后得到第二位
要知道如果sum是D位的数，那么x一定是D位或者D-1位数，否则不可能

137174210616796
123456789555123

代码如下：

#include<stdio.h>
#include<iostream>

#define MAX_DIGIT 19

typedef long long LL;

using namespace std;

int GetDigit(LL num){
    if(num<0)
        return -1;
    if(num==0)
        return 1;
    int digit=0;
    while(num){
        digit++;
        num/=10;
    }
    return digit;
}

//digit只会是sumDigit或者sumDigit-1
LL GetCandidate(int digit, int sumDigit, LL sum)
{
    if(digit!=sumDigit && digit!=sumDigit-1)
        return -1;
    //设置一把“尺子”一个digit个1
    LL ruler=0;
    for(int i=0; i<digit; i++){
        ruler*=10;
        ruler+=1;
    }
    //若sum都小于ruler那么就没有必要尝试后面了要确保第一位至少不是0
    if(sum<ruler)
        return -1;
    //这里第一位算出来肯定不是0
    LL tempSum=sum;
    LL candidate=0;
    for(int i=0; i<digit; i++){
        LL currentDigit=tempSum/ruler;
        //currentDigit最多是一个两位数
        if(currentDigit>=10)
            currentDigit/=10;
        candidate*=10;
        candidate+=(LL)currentDigit;
        tempSum=tempSum-(LL)currentDigit*ruler;
        //更新ruler
        ruler/=10;
        //若新的tempSum的位数大于ruler的位数+1那么直接返回-1
        if(GetDigit(tempSum)>GetDigit(ruler)+1)
            return -1;
    }
    if(tempSum!=0)
        return -1;
    return candidate;
}

int main()
{
//    LL test=GetCandidate(GetDigit(137174210616796), GetDigit(137174210616796),137174210616796);
//    printf("%lld", test);
    LL sum, x=-1;
    scanf("%lld", &sum);
    //获取sum是多少位 至少是一位
    int sumDigit=GetDigit(sum);
    //遍历所有可能的digit
    for(int digit=max(1,sumDigit-1); digit<=sumDigit; digit++){
        long long xx=GetCandidate(digit, sumDigit, sum);
        if(xx==-1)
            continue;
        x=xx;
        break;
    }
    printf("%lld", x);
    return 0;
}