Server Stress Scoring

Here is the full interview question "Server Stress Scoring - Stripe OA Interview | ShowOffer" from Stripe on ShowOffer.io:

Server Stress Scoring - Stripe OA Interview | ShowOffer

You are given a list of servers, each with a stress score. The stress score is an integer representing the current stress level of the server. Your task is to calculate the total stress score for each server, taking into account the stress levels of its neighbors.

Input:

• A list of integers servers, where servers[i] represents the stress level of the server at index i.
• An integer k, representing the number of neighbors to consider on each side of a server.

Output:

• A list of integers, where each integer represents the total stress score for the corresponding server.

Constraints:

• 1 <= len(servers) <= 10^5
• 0 <= k < len(servers)

Examples:

1. Input:

   • servers = [1, 2, 3, 4, 5]
   • k = 1
   • neighbors = [1, 1, 2, 2, 1]
   • weights = [0.5, 1.0]
   • threshold = 10

   Output:

   • [1.0, 2.5, 5.5, 6.5, 5.0]

   Explanation:

   • Server 0: Stress score = 1.0 (no neighbors)
   • Server 1: Stress score = 1.0 (server 0) + 2.0 (server 1) + 0.5 (server 2) = 3.5
   • Server 2: Stress score = 2.0 (server 1) + 3.0 (server 2) + 1.0 (server 3) + 0.5 (server 4) = 7.5
   • Server 3: Stress score = 3.0 (server 2) + 4.0 (server 3) + 1.0 (server 4) = 8.0
   • Server 4: Stress score = 4.0 (server 3) + 5.0 (server 4) = 9.0

2. Input:

   • servers = [10, 20, 30, 40, 50]
   • k = 2
   • neighbors = [2, 2, 3, 3, 2]
   • weights = [0.5, 0.5, 1.0]
   • threshold = 15

   Output:

   • [10.0, 25.0, 40.0, 50.0, 60.0]

   Explanation:

   • Server 0: Stress score = 10.0 (no neighbors)
   • Server 1: Stress score = 10.0 (server 0) + 20.0 (server 1) + 15.0 (server 2) = 45.0
   • Server 2: Stress score = 20.0 (server 1) + 30.0 (server 2) + 20.0 (server 3) + 15.0 (server 4) = 85.0
   • Server 3: Stress score = 30.0 (server 2) + 40.0 (server 3) + 30.0 (server 4) = 100.0
   • Server 4: Stress score = 40.0 (server 3) + 50.0 (server 4) + 25.0 (server 0) = 115.0

Hints:

1. Iterate through the servers and calculate the stress score for each server by considering its neighbors.
2. Use a sliding window approach to efficiently calculate the stress score for each server.
3. Handle edge cases where the server is near the beginning or end of the list.

Solution:

def server_stress_scoring(servers, k, neighbors, weights, threshold):
    n = len(servers)
    stress_scores = [0] * n

    for i in range(n):
        total_stress = servers[i]
        current_weight = 1.0

        for j in range(1, neighbors[i] + 1):
            if i - j >= 0:
                total_stress += servers[i - j] * weights[j - 1]
                current_weight += weights[j - 1]
            if i + j <