using System;

using System.Collections;

using System.Collections.Generic;

​

public class Program

{

    public static void Main()

    {

        Console.WriteLine("Hello World");

    }

    public class Solution {

        public int[] KWeakestRows(int[][] mat, int k) {

            Dictionary<int, List<int>> dic = FillDic(mat);

            List<int> sorted = new List<int>(dic.Keys);

            sorted.Sort();

            List<int> res = new List<int>();

            for (int i=0; i<sorted.Count; i++)

            {

                foreach (int val in dic[sorted[i]]) {

                    if(k > 0) {

                        res.Add(val);

                        k--;

                    }

                }

            }

​

            return res.ToArray();

        }

​

​

        private Dictionary<int, List<int>> FillDic (int [][] mat) {

            Dictionary<int, List<int>> res = new Dictionary<int, List<int>>();

            for(int i=0; i<mat.Length; i++) {

                int st = 0; 

                for(int j=0; j<mat[i].Length; j++) 

                {

                    st += mat[i][j];

                }

                if(!res.ContainsKey(st))

                    res.Add(st, new List<int>());

                res[st].Add(i);

            }

            return res;

        }

    }

}