Sum of Digits of String After Convert

You are given a string s consisting of lowercase English letters, and an integer k.

First, convert s into an integer by replacing each letter with its position in the alphabet (i.e., replace 'a' with 1, 'b' with 2, ..., 'z' with 26). Then, transform the integer by replacing it with the sum of its digits. Repeat the transform operation k times in total.

For example, if s = "zbax" and k = 2, then the resulting integer would be 8 by the following operations:

• Convert: "zbax" ➝ "(26)(2)(1)(24)" ➝ "262124" ➝ 262124
• Transform #1: 262124 ➝ 2 + 6 + 2 + 1 + 2 + 4 ➝ 17
• Transform #2: 17 ➝ 1 + 7 ➝ 8

Return the resulting integer after performing the operations described above.

Example 1:

Input: s = "iiii", k = 1
Output: 36
Explanation: The operations are as follows:
- Convert: "iiii" ➝ "(9)(9)(9)(9)" ➝ "9999" ➝ 9999
- Transform #1: 9999 ➝ 9 + 9 + 9 + 9 ➝ 36
Thus the resulting integer is 36.

Example 2:

Input: s = "leetcode", k = 2
Output: 6
Explanation: The operations are as follows:
- Convert: "leetcode" ➝ "(12)(5)(5)(20)(3)(15)(4)(5)" ➝ "12552031545" ➝ 12552031545
- Transform #1: 12552031545 ➝ 1 + 2 + 5 + 5 + 2 + 0 + 3 + 1 + 5 + 4 + 5 ➝ 33
- Transform #2: 33 ➝ 3 + 3 ➝ 6
Thus the resulting integer is 6.

Example 3:

Input: s = "zbax", k = 2
Output: 8

Approach

Java

public class SumofFDigits {
    public static void main(String[] args) {
        String s = "iiii";
        int k = 1;

        System.out.println(getLucky(s, k));

    }

    static int getLucky(String s, int k) {

        int sum = 0;
        for (int i = 0; i < s.length(); i++) {
            int num = s.charAt(i) - 'a' + 1;
            while (num > 0) {
                sum += num % 10;
                num = num / 10;
            }
        }
        k--;
        while (k > 0 && sum > 0) {
            int newSum = 0;

            while (sum > 0) {
                newSum += sum % 10;
                sum = sum / 10;
            }
            sum = newSum;
            k--;
        }
        return sum;
    }

}

C++

#include <bits/stdc++.h>
using namespace std;

int getLucky(string s, int k)
{

    int sum = 0;
    for (int i = 0; i < s.size(); i++)
    {
        int num = s[i] - 'a' + 1;
        while (num > 0)
        {
            sum += num % 10;
            num = num / 10;
        }
    }
    k--;
    while (k > 0 && sum > 0)
    {
        int newSum = 0;

        while (sum > 0)
        {
            newSum += sum % 10;
            sum = sum / 10;
        }
        sum = newSum;
        k--;
    }
    return sum;
}

int main()
{
    string s = "iiii";
    int k = 1;

    cout << getLucky(s, k) << "\n";

    return 0;
}