Untitled

 avatar
unknown
python
20 days ago
1.7 kB
6
Indexable
#include <bits/stdc++.h>
using namespace std;

int logic1(int n, int k, int arr[])
{
    int st = 0, en = n - 1;
    sort(arr + st, arr + en);
    return arr[k - 1];
}

int correct(int n, int k, int arr[])
{
    for (int i = 0; i < n; i++)
    {
        int cnt = 0;
        for (int j = 0; j < n; j++)
        {
            if (arr[j] < arr[i])
                cnt++;
        }
        if (cnt == (k - 1))
        {
            return arr[i];
        }
        else
        {
            continue;
        }
    }
}

int get_random(int a, int b)
{
    return a + rand() % (b - a + 1);
}

void generate()
{
    int t = 100;
    while (t--)
    {
        int n, k;
        n = get_random(1, 6);
        k = get_random(1, n);

        int arr[n];
        set<int> st;
        while (st.size() < n)
        {
            int x = get_random(1, 100);
            st.insert(x);
        }
        int index = 0;
        for (auto v : st)
        {
            arr[index++] = v;
        }
        random_shuffle(arr, arr + n);

        cout << "Input generated: \n";
        cout << n << " " << k << endl;
        for (int i = 0; i < n; i++)
        {
            cout << arr[i] << " ";
        }
        cout << endl
             << endl;

        int code_op = logic1(n, k, arr);
        int correct_op = correct(n, k, arr);
        if (code_op != correct_op)
        {
            cout << "FAILING!\n";
            return;
        }
    }
}

void solve()
{
    int n, k;
    cin >> n >> k;
    int arr[n];
    for (int i = 0; i < n; i++)
    {
        cin >> arr[i];
    }
    cout << logic1(n, k, arr) << endl;
}

int main()
{
    // generate();
    int t;
    cin >> t;
    while (t--)
        solve();
}
Editor is loading...
Leave a Comment