pskrunner14

import time
import numpy as np

def topk(a, k):
    # idx = a.argsort()[-k:][::-1] # slower (runs in O(n logn) time)
    idx = np.argpartition(a, -k)[-k:] # faster (runs in linear O(n) time)
    return a[idx], idx

def decode(n):
    return np.random.rand(n)