Amazon OA | SDE 1 | March 2025 | Set - 1

Question

Answer 1

def inventory_process(robots):

    survival = []

    for i in range(len(robots)):

        robots_copy = robots[:]

        repeat = True

        while repeat:

            for j in range(len(robots)):

               

                if i != j:

                    if robots_copy[i] >= robots_copy[j] and robots_copy[j] > 0:

                        robots_copy[i] += robots_copy[j]

                        robots_copy[j] = 0

            robots_at_i_backup = robots_copy[i]

            robots_copy[i] = 0

            if all(x == 0 for x in robots_copy):

                survival.append(i + 1)

                break

            else:

                robots_copy[i] = robots_at_i_backup

            for k in range(len(robots_copy)):

                if k != i and robots_copy[i] >= robots_copy[k] and robots_copy[k] > 0:

                    repeat = True

                    break

                else:

                    repeat = False

    return survival

print(inventory_process([1, 2, 3, 4, 5]))

print(inventory_process([1, 6, 2, 7, 2]))

Answer 2

#include <iostream>

#include <vector>

#include <algorithm>

using namespace std;

vector<int> solve(vector<int>& arr, int n){

    vector<pair<int,int>> sorted;

    

    for(int i = 0; i < n; i++){

        sorted.push_back({arr[i], i+1});

    }

    

    sort(sorted.begin(), sorted.end());

    

    vector<int> prefix;

    

    int c = 0;

    

    for(auto it : sorted){

        c += it.first;

        prefix.push_back(c);

    }

    

    int i = n-2;

    

    while(i >= 0 && prefix[i] >= sorted[i+1].first) i--;

    

    i++;

    vector<int> ans;

    

    while(i < n){

        ans.push_back(sorted[i].second);

        i++;

    }

    

    return ans;    

}

int main()

{

    int n;

    cin >> n;

    vector<int> arr;

    

    for(int i = 0; i < n; i++){

        int t;

        cin >> t;

        arr.push_back(t);

    }

    

    vector<int> ans = solve(arr, n);

    

    for(auto it : ans){

        cout << it << endl;

    }

    

    return 0;

}