Added tasks 3556-3663

Author: javadev

src/main/kotlin/g3501_3600/s3556_sum_of_largest_prime_substrings/Solution.kt

@@ -0,0 +1,44 @@
+package g3501_3600.s3556_sum_of_largest_prime_substrings
+
+// #Medium #2025_05_25_Time_21_ms_(100.00%)_Space_42.82_MB_(100.00%)
+
+class Solution {
+    fun sumOfLargestPrimes(s: String): Long {
+        val primeSet: MutableSet<Long> = HashSet<Long>()
+        val n = s.length
+        for (i in 0..<n) {
+            var temp: Long = 0
+            for (j in i..<n) {
+                temp = temp * 10 + (s[j].code - '0'.code)
+                if (isPrime(temp)) {
+                    primeSet.add(temp)
+                }
+            }
+        }
+        val primes: MutableList<Long> = ArrayList<Long>(primeSet)
+        primes.sort()
+        val m = primes.size
+        if (m < 3) {
+            var sum: Long = 0
+            for (p in primes) {
+                sum += p
+            }
+            return sum
+        }
+        return primes[m - 1] + primes[m - 2] + primes[m - 3]
+    }
+
+    private fun isPrime(n: Long): Boolean {
+        if (n < 2) {
+            return false
+        }
+        var i: Long = 2
+        while (i * i <= n) {
+            if (n % i == 0L) {
+                return false
+            }
+            ++i
+        }
+        return true
+    }
+}

src/main/kotlin/g3501_3600/s3556_sum_of_largest_prime_substrings/readme.md

@@ -0,0 +1,36 @@
+3556\. Sum of Largest Prime Substrings
+
+Medium
+
+Given a string `s`, find the sum of the **3 largest unique prime numbers** that can be formed using any of its ****substring****.
+
+Return the **sum** of the three largest unique prime numbers that can be formed. If fewer than three exist, return the sum of **all** available primes. If no prime numbers can be formed, return 0.
+
+**Note:** Each prime number should be counted only **once**, even if it appears in **multiple** substrings. Additionally, when converting a substring to an integer, any leading zeros are ignored.
+
+**Example 1:**
+
+**Input:** s = "12234"
+
+**Output:** 1469
+
+**Explanation:**
+
+*   The unique prime numbers formed from the substrings of `"12234"` are 2, 3, 23, 223, and 1223.
+*   The 3 largest primes are 1223, 223, and 23. Their sum is 1469.
+
+**Example 2:**
+
+**Input:** s = "111"
+
+**Output:** 11
+
+**Explanation:**
+
+*   The unique prime number formed from the substrings of `"111"` is 11.
+*   Since there is only one prime number, the sum is 11.
+
+**Constraints:**
+
+*   `1 <= s.length <= 10`
+*   `s` consists of only digits.

src/main/kotlin/g3501_3600/s3557_find_maximum_number_of_non_intersecting_substrings/Solution.kt

@@ -0,0 +1,28 @@
+package g3501_3600.s3557_find_maximum_number_of_non_intersecting_substrings
+
+import java.util.LinkedList
+import kotlin.math.max
+
+// #Medium #2025_05_25_Time_103_ms_(100.00%)_Space_54.87_MB_(100.00%)
+
+class Solution {
+    fun maxSubstrings(s: String): Int {
+        val last: Array<LinkedList<Int>> = Array(26) { LinkedList() }
+        val n = s.length
+        val dp = IntArray(n + 1)
+        for (i in 0..<n) {
+            val c = s[i].code - 'a'.code
+            dp[i + 1] = dp[i]
+            for (j in last[c]) {
+                if (i - j + 1 >= 4) {
+                    dp[i + 1] = max(dp[i + 1], dp[j] + 1)
+                }
+            }
+            last[c].addLast(i)
+            if (last[c].size > 4) {
+                last[c].removeFirst()
+            }
+        }
+        return dp[n]
+    }
+}

src/main/kotlin/g3501_3600/s3557_find_maximum_number_of_non_intersecting_substrings/readme.md

@@ -0,0 +1,32 @@
+3557\. Find Maximum Number of Non Intersecting Substrings
+
+Medium
+
+You are given a string `word`.
+
+Return the **maximum** number of non-intersecting ****substring**** of word that are at **least** four characters long and start and end with the same letter.
+
+**Example 1:**
+
+**Input:** word = "abcdeafdef"
+
+**Output:** 2
+
+**Explanation:**
+
+The two substrings are `"abcdea"` and `"fdef"`.
+
+**Example 2:**
+
+**Input:** word = "bcdaaaab"
+
+**Output:** 1
+
+**Explanation:**
+
+The only substring is `"aaaa"`. Note that we cannot **also** choose `"bcdaaaab"` since it intersects with the other substring.
+
+**Constraints:**
+
+*   <code>1 <= word.length <= 2 * 10<sup>5</sup></code>
+*   `word` consists only of lowercase English letters.

src/main/kotlin/g3501_3600/s3558_number_of_ways_to_assign_edge_weights_i/Solution.kt

@@ -0,0 +1,65 @@
+package g3501_3600.s3558_number_of_ways_to_assign_edge_weights_i
+
+import java.util.*
+
+// #Medium #2025_05_25_Time_118_ms_(100.00%)_Space_122.30_MB_(100.00%)
+
+class Solution {
+    fun assignEdgeWeights(edges: Array<IntArray>): Int {
+        val n = edges.size + 1
+        val adj: MutableList<MutableList<Int?>?> = ArrayList<MutableList<Int?>?>()
+        for (i in 0..n) {
+            adj.add(ArrayList<Int?>())
+        }
+        for (i in edges) {
+            adj.get(i[0])!!.add(i[1])
+            adj.get(i[1])!!.add(i[0])
+        }
+        val l = IntArray(n + 1)
+        var max = 0
+        Arrays.fill(l, -1)
+        val q: Queue<IntArray?> = LinkedList<IntArray?>()
+        q.offer(intArrayOf(1, 0))
+        l[1] = 0
+        while (!q.isEmpty()) {
+            val curr = q.peek()!![0]
+            val level = q.peek()!![1]
+            if (l[max] < l[curr]) {
+                max = curr
+            }
+            q.remove()
+            for (next in adj.get(curr)!!) {
+                if (l[next!!] != -1) {
+                    continue
+                }
+                q.offer(intArrayOf(next, level + 1))
+                l[next] = level + 1
+            }
+        }
+        val dp: Array<IntArray> = Array<IntArray>(l[max]) { IntArray(2) }
+        for (i in dp) {
+            Arrays.fill(i, -1)
+        }
+        return solve(0, 0, dp)
+    }
+
+    private fun solve(ind: Int, odd: Int, dp: Array<IntArray>): Int {
+        if (ind == dp.size) {
+            if (odd == 1) {
+                return 1
+            } else {
+                return 0
+            }
+        }
+        if (dp[ind][odd] != -1) {
+            return dp[ind][odd]
+        }
+        dp[ind][odd] =
+            (solve(ind + 1, odd, dp) % MOD + solve(ind + 1, (odd + 1) % 2, dp) % MOD) % MOD
+        return dp[ind][odd]
+    }
+
+    companion object {
+        private val MOD = 1e9.toInt() + 7
+    }
+}

src/main/kotlin/g3501_3600/s3558_number_of_ways_to_assign_edge_weights_i/readme.md

@@ -0,0 +1,50 @@
+3558\. Number of Ways to Assign Edge Weights I
+
+Medium
+
+There is an undirected tree with `n` nodes labeled from 1 to `n`, rooted at node 1. The tree is represented by a 2D integer array `edges` of length `n - 1`, where <code>edges[i] = [u<sub>i</sub>, v<sub>i</sub>]</code> indicates that there is an edge between nodes <code>u<sub>i</sub></code> and <code>v<sub>i</sub></code>.
+
+Initially, all edges have a weight of 0. You must assign each edge a weight of either **1** or **2**.
+
+The **cost** of a path between any two nodes `u` and `v` is the total weight of all edges in the path connecting them.
+
+Select any one node `x` at the **maximum** depth. Return the number of ways to assign edge weights in the path from node 1 to `x` such that its total cost is **odd**.
+
+Since the answer may be large, return it **modulo** <code>10<sup>9</sup> + 7</code>.
+
+**Note:** Ignore all edges **not** in the path from node 1 to `x`.
+
+**Example 1:**
+
+![](https://assets.leetcode.com/uploads/2025/03/23/screenshot-2025-03-24-at-060006.png)
+
+**Input:** edges = [[1,2]]
+
+**Output:** 1
+
+**Explanation:**
+
+*   The path from Node 1 to Node 2 consists of one edge (`1 → 2`).
+*   Assigning weight 1 makes the cost odd, while 2 makes it even. Thus, the number of valid assignments is 1.
+
+**Example 2:**
+
+![](https://assets.leetcode.com/uploads/2025/03/23/screenshot-2025-03-24-at-055820.png)
+
+**Input:** edges = [[1,2],[1,3],[3,4],[3,5]]
+
+**Output:** 2
+
+**Explanation:**
+
+*   The maximum depth is 2, with nodes 4 and 5 at the same depth. Either node can be selected for processing.
+*   For example, the path from Node 1 to Node 4 consists of two edges (`1 → 3` and `3 → 4`).
+*   Assigning weights (1,2) or (2,1) results in an odd cost. Thus, the number of valid assignments is 2.
+
+**Constraints:**
+
+*   <code>2 <= n <= 10<sup>5</sup></code>
+*   `edges.length == n - 1`
+*   <code>edges[i] == [u<sub>i</sub>, v<sub>i</sub>]</code>
+*   <code>1 <= u<sub>i</sub>, v<sub>i</sub> <= n</code>
+*   `edges` represents a valid tree.

src/main/kotlin/g3501_3600/s3559_number_of_ways_to_assign_edge_weights_ii/Solution.kt

@@ -0,0 +1,98 @@
+package g3501_3600.s3559_number_of_ways_to_assign_edge_weights_ii
+
+import kotlin.math.ceil
+import kotlin.math.ln
+
+// #Hard #2025_05_25_Time_135_ms_(100.00%)_Space_119.27_MB_(100.00%)
+
+class Solution {
+    private var adj: MutableList<MutableList<Int?>?>? = null
+    private lateinit var level: IntArray
+    private lateinit var jumps: Array<IntArray?>
+
+    private fun mark(node: Int, par: Int) {
+        for (neigh in adj!!.get(node)!!) {
+            if (neigh == par) {
+                continue
+            }
+            level[neigh!!] = level[node] + 1
+            jumps[neigh]!![0] = node
+            mark(neigh, node)
+        }
+    }
+
+    fun lift(u: Int, diff: Int): Int {
+        var u = u
+        var diff = diff
+        while (diff > 0) {
+            val rightmost = diff xor (diff and (diff - 1))
+            val jump = (ln(rightmost.toDouble()) / ln(2.0)).toInt()
+            u = jumps[u]!![jump]
+            diff -= rightmost
+        }
+        return u
+    }
+
+    private fun findLca(u: Int, v: Int): Int {
+        var u = u
+        var v = v
+        if (level[u] > level[v]) {
+            val temp = u
+            u = v
+            v = temp
+        }
+        v = lift(v, level[v] - level[u])
+        if (u == v) {
+            return u
+        }
+        for (i in jumps[0]!!.indices.reversed()) {
+            if (jumps[u]!![i] != jumps[v]!![i]) {
+                u = jumps[u]!![i]
+                v = jumps[v]!![i]
+            }
+        }
+        return jumps[u]!![0]
+    }
+
+    private fun findDist(a: Int, b: Int): Int {
+        return level[a] + level[b] - 2 * level[findLca(a, b)]
+    }
+
+    fun assignEdgeWeights(edges: Array<IntArray>, queries: Array<IntArray>): IntArray {
+        val n = edges.size + 1
+        adj = ArrayList<MutableList<Int?>?>()
+        level = IntArray(n)
+        for (i in 0..<n) {
+            adj!!.add(ArrayList<Int?>())
+        }
+        for (i in edges) {
+            adj!!.get(i[0] - 1)!!.add(i[1] - 1)
+            adj!!.get(i[1] - 1)!!.add(i[0] - 1)
+        }
+        val m = (ceil(ln(n - 1.0) / ln(2.0))).toInt() + 1
+        jumps = Array<IntArray?>(n) { IntArray(m) }
+        mark(0, -1)
+        for (j in 1..<m) {
+            for (i in 0..<n) {
+                val p = jumps[i]!![j - 1]
+                jumps[i]!![j] = jumps[p]!![j - 1]
+            }
+        }
+        val pow = IntArray(n + 1)
+        pow[0] = 1
+        for (i in 1..n) {
+            pow[i] = (pow[i - 1] * 2) % MOD
+        }
+        val q = queries.size
+        val ans = IntArray(q)
+        for (i in 0..<q) {
+            val d = findDist(queries[i][0] - 1, queries[i][1] - 1)
+            ans[i] = if (d > 0) pow[d - 1] else 0
+        }
+        return ans
+    }
+
+    companion object {
+        private const val MOD = 1000000007
+    }
+}