如何重新计算GridLayoutManager的列数？附近似可行实现代码

我帮你梳理下这段GridLayoutManager的列数适配代码，先说说原有实现里的几个潜在问题，再给出修正后的版本，确保逻辑更严谨：

原有代码的小问题 #

• 循环边界漏判：calculateSpan里的while (currentPosition != 0)会漏掉对索引0元素的判断，比如当前元素是第3个（position=2），且前两个元素都和它首字母相同时，循环只会统计到position=1的元素，漏掉position=0的，导致统计的同组元素数量少一个。
• 最后一个元素处理太粗糙：直接固定返回1，但如果最后一个元素所在的组还没凑满3列，应该计算剩余跨度来填满一行，而不是硬占1列。
• 组判断逻辑不严谨：仅通过相邻元素首字母判断是否为同一组，但如果当前元素是列表最后一个，就直接返回1，没考虑它所在组的整体情况。

修正后的完整代码 #

GridLayoutManager layoutManager = new GridLayoutManager(this, 3);
layoutManager.setSpanSizeLookup(new GridLayoutManager.SpanSizeLookup() {
    @Override
    public int getSpanSize(int position) {
        SchedulePlanGlobal currentElem = mElements.get(position);
        char currentRoomChar = currentElem.getRoomName().charAt(0);
        
        // 判断当前元素是不是组内最后一个：要么是列表末尾，要么下一个元素首字母不同
        boolean isLastInGroup = (position == mElements.size() - 1) 
                || (mElements.get(position + 1).getRoomName().charAt(0) != currentRoomChar);
        
        if (!isLastInGroup) {
            // 不是组内最后一个，固定占1列
            return 1;
        } else {
            // 是组内最后一个，计算需要占多少列填满当前行
            return calculateSpanToFillRow(position, currentRoomChar);
        }
    }

    private int calculateSpanToFillRow(int position, char targetRoomChar) {
        // 统计当前元素往前连续相同首字母的元素总数（包含当前元素）
        int sameGroupCount = 1;
        int currentPos = position - 1;
        // 修正循环边界：currentPos >=0 才能统计到第一个元素
        while (currentPos >= 0) {
            if (mElements.get(currentPos).getRoomName().charAt(0) == targetRoomChar) {
                sameGroupCount++;
                currentPos--;
            } else {
                break;
            }
        }
        
        // 处理同组元素超过3个的情况：每3个占满一行，最后不足3个的填满剩余列
        int remainder = sameGroupCount % 3;
        if (remainder == 0) {
            // 刚好是3的倍数，当前元素是组内第3的倍数个，占1列（和前面两个凑满一行）
            return 1;
        } else {
            // 剩余不足3个，计算需要占的列数填满当前行
            return 3 - (sameGroupCount - remainder - 1);
        }
    }
});

关键改进说明 #

• 修复循环统计问题：把循环条件改成currentPos >=0，确保能统计到列表第一个元素，准确计算同组元素的总数量。
• 完善多元素组的布局：当同组元素超过3个时，会自动分成多行，每3个占满一行，最后不足3个的元素会自动调整跨度填满一行（比如5个同组元素，前3个各占1列，后2个分别占2列和1列）。
• 逻辑更清晰：用isLastInGroup统一判断当前元素的角色，避免零散的条件判断。

验证场景参考 #

• 同组1个元素：直接占3列，单独一行
• 同组2个元素：第一个占1列，第二个占2列，刚好填满一行
• 同组3个元素：每个占1列，填满一行
• 同组4个元素：前3个占一行，第4个占3列单独一行
• 同组5个元素：前3个占一行，后2个占2列+1列填满一行

这样调整后，就能保证相同房间首字母的连续元素整齐排列，自动适配GridLayoutManager的3列布局啦。

内容的提问来源于stack exchange，提问作者Lechucico