dsa-cpp/66.cpp at main · HirenGajjar/dsa-cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#include <iostream>
#include <vector>
using namespace std;
// vector<int> plus_one_brute_force(vector<int> vec)

// {
//     vector<int> answer;
//     int size = vec.size();
//     int decimal = 0;
//     int power = 1;

//     for (int i = size - 1; i >= 0; i--)
//     {
//         decimal += (power * vec[i]);
//         power *= 10;
//     }
//     decimal++;

//     while (decimal > 0)
//     {
//         answer.insert(answer.begin(), decimal % 10);
//         decimal /= 10;
//     }
//     return answer;
// }

vector<int> plus_one(vector<int> vec)
{

    if (vec.size() == 1 && vec[vec.size() - 1] == 9)
    {
        vec.pop_back();
        vec.push_back(1);
        vec.push_back(0);
    }

    else
    {
        for (int i = vec.size() - 1; i >= 0; i--)
        {

            if (vec[i] != 9)
            {
                vec[i]++;
                break;
            }
            else if (vec[i] == 9)
            {
                vec[i] = 0;
            }
        }
        if (vec[0] == 0)
        {
            vec[0] = 1;
            vec.push_back(0);
        }
    }
    return vec;
}
int main()
{
    vector<int> vec = {0};
    // vector<int> answer = plus_one_brute_force(vec);
    // for (int i : answer)
    // {
    //     cout << i << " ";
    // }
    vector<int> answer = plus_one(vec);
    for (int i : answer)
    {
        cout << i << " ";
    }
    cout << endl;

    return 0;
}