hihocoder 1989 幸运数之和

题目描述

如果一个十进制数只包含数字0和1，我们就称它是"幸运数"。例如0、1、10、11、100、1110等都是幸运数。
给定一个整数N，请你计算N一共有多少种方法表示成两个非负幸运数的和。注意交换两个数的顺序视为同一种方法。
具体参见原题链接

思路分析

由于幸运数只包括0和1，所以不存在进位的可能，每一位数字需要进行独立拆分。
对每一位数字有以下几种情况：

1. 如果该位数字大于2，则无法拆分成两个数字之和。
2. 如果该位数字为2，则只能拆分成1+1。
3. 如果该位数字为1，可以拆分成0+1和1+0(对每一位拆分顺序不同，可以组合出不同的结果)。
4. 如果该位数字为0，只能拆分成0+0。

通过以上分析，其实只需要计算有多少位为1即可。
存在一种特殊情况，从高到低第一位1需要特殊考虑，如果拆分成两种情况存在重复计数，所以只能拆分成一种。
由于输入输出均是大数，长整型也无法表示，需转成字符串处理。

代码实现

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

using namespace std;

int main() {
    string s;
    cin>>s;
    int cnt = 0; // 记录1出现的次数
    for(int i=0;i<s.size();i++) {
        if(s[i]>'2') {cout<<0<<endl; return 0;}
        if(s[i]=='1') cnt++;
    }
    if(cnt>0) cnt--; // 去掉首位1的计数
    if(cnt==0) {cout<<1<<endl; return 0;}
    // 主要难点在于数字会越界，需要转成字符串表示
    string res = "1";
    while(cnt--) {
        int d = 0;
        for(int i=res.size()-1;i>=0;i--) {
            d += (res[i]-'0')*2;
            res[i] = d%10 + '0';
            d /= 10;
        }
        if(d>0) res.insert(res.begin(), d+'0');
    }
    cout<<res<<endl;
    return 0;
}