using System;
using System.Collections.Generic;
using System.Linq;

namespace AdventOfCode.Extensions;

public static class EnumerableExtensions
{
    public static IEnumerable<TValue[]> Combinations<TValue>(this IEnumerable<TValue> values, int count)
    {
        var pool = values.ToArray();
        var poolLength = pool.Length;
        if (count > poolLength)
            yield break;
        var indices = Enumerable.Range(0, count).ToArray();
        yield return GetCombination(indices, pool);
        while (true)
        {
            int idx;
            for(idx = count - 1; idx >= 0; idx--)
            {
                var isIndexBelowMax = indices[idx] != idx + poolLength - count;
                if (isIndexBelowMax) 
                    break;
            }
            if(idx < 0)
                yield break;
            
            indices[idx] += 1;
            for (var j = idx + 1; j < count; j++)
            {
                indices[j] = indices[j - 1] + 1;
            }

            yield return GetCombination(indices, pool);
        }

        static TValue[] GetCombination(int[] innerIndices, TValue[] innerPool) => 
            innerIndices.Select(i => innerPool[i]).ToArray();
    }
}