trigger UpdateSubgroupOnContact on Contact (before insert) {
    if(Trigger.isBefore){
        if(Trigger.isInsert){   
            
            // CHANGE: We will use a map instead, so we can remove this set
            // Set<Decimal> setOfSubgroupIds = new Set<Decimal>();
            
            // CHANGE: Create a Map of SubgroupId -> List of Contacts
            Map<String,List<Contact>> subIdContacts = new Map<String,List<Contact>>();
            // CHANGE: Temporary list for holding values
            List<Contact> tempContacts;
            
            for(Contact cnt : trigger.new){
                if(cnt.SubgroupCode__c != null && cnt.SubgroupId__c != null){
                    
                    // CHANGE: Add to the map by new entry or updating an existing entry
                    if(subIdContacts.contains(cnt.SubgroupId__c)) {
                        tempContacts = subIdContacts.get(cnt.SubgroupId__c);
                        tempContacts.add(cnt);
                        subIdContacts.put(cnt.SubgroupId__c, tempContacts);
                    } else {
                        tempContacts = new List<Contact>();
                        tempContacts.add(cnt);
                        subIdContacts.put(cnt.SubgroupId__c, tempContacts);
                    }
                    
                    // CHANGE: We have a map to track subgroup Ids so we will not use the set
                    // setOfSubgroupIds.add(cnt.SubgroupId__c);
                    
                    //system.debug('setof subgroup ids======'+setOfSubgroupIds);
                    //subGroupRecords = [Select Id,Account__c, Subgroup_Id__c, SubGroup_Code__c from SubGroup__c  where Subgroup_Id__c =:cnt.SubgroupId__c];
                    
                    // CHANGE: The below code needs to be moved outside of the for loop because of the query
                    /*
                    for(List<SubGroup__c> sbg: [Select Id,Account__c,Subgroup_Id__c, SubGroup_Code__c from SubGroup__c  where Subgroup_Id__c IN:setOfSubgroupIds]){
                        for(SubGroup__c s: sbg){
                            cnt.Subgroup__c = s.Account__c;
                            
                        }
                    }
                    */
                }     
            }
            
            // CHANGE: Now that this query is outside of the previous for loop, we will only run it once to grab all subgroups
            for(SubGroup__c sbg: [Select Id,Account__c,Subgroup_Id__c, SubGroup_Code__c from SubGroup__c  where Subgroup_Id__c IN :subIdContacts.keySet()]){
                // Because the subgroup has a SubgroupId field, we can find the contacts via the map we made
                for(Contact con : subIdContacts.get(sbg.SubGroup_Id__c)) {
                    // Then for all the contacts under one subgroup Id, we can add the same Account
                    con.Subgroup__c = sbg.Account__c;
                }
            }
        }
    }
}