递归淘汰List集合头部数据,获取最终集合的起始坐标

简介: 递归淘汰List集合头部数据,获取最终集合的起始坐标

需求分析:

场景:调用三方chatGPT服务,通过传递历史聊天记录来获得联系上下文的回答

需要解决的问题:传递的历史聊天记录有限制,所以每次就需要去判断是否超出限制,如果超出限制就要淘汰掉一些最早的聊天记录去传值。

数据结构(简化):定义一个集合,集合中的对象有两个属性,一个是聊天内容,一个是聊天内容的字数

分析:如果传递的数据>设定的最大值,这个时候我们把两者的差计算出来定义为A,然后去聊天记录的集合的开始去判断A是否小于集合中第一个元素设定的字数,如果小于那把这个元素从集合中移除就行了,如果不小于,那就需要拿到集合的前两个元素设定的字数加起来,再和A去判断,以此类推

使用递归算法获取淘汰集合的临界坐标:

温馨提示:注释都放到代码里了哦~

/**
     * 淘汰老的聊天记录
     * @param  num 是集合总字数和设定限制的差值
     * @param  s 是最后要返回的临界值,也就是s前面的坐标都要被淘汰
     */
    int weedOut(int num, int s, List<ChatMessage> chatMessageList) {
        int sum = 0; // 集合聊天记录最前面部分tokens相加
        for (int i = 0; i < s; i++) {
            ChatMessage chatMessages = chatMessageList.get(i);
            // 计算要淘汰的元素的字数
            sum += chatMessages.getTokens();
        }
        // 判断要淘汰的元素的字数,如果不符合条件  下标继续下移  s + 1
        if (sum > num) {
            return s; // 最后要这个集合下标为s后面的
        } else {
            return weedOut(num, s + 1, chatMessageList);
        }
    }

结合业务的完整代码供大家交流:

/**
 * @author : jiagang
 * @date : Created in 2023/7/12 10:48
 */
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class ChatTest {
    @Autowired
    private ChatMessageService chatMessageService;
    @Test
    public void test1(){
        List<ChatMessage> chatMessageList = chatMessageService.chatMessageList(ChatMessage.builder().sessionId("f3667fec-ff71-f1e9-108a-658f527378a3").build());
        JSONArray jsonArray = new JSONArray(); // 最终结果
        if (!CollectionUtils.isEmpty(chatMessageList)) {
            // 计算聊天信息tokens
            int sum = chatMessageList.stream().mapToInt(ChatMessage::getTokens).sum(); // 301 个   99    202
            if (sum > VoiceChatConstant.TOKENS) {
                // 获取淘汰集合坐标的临界点
                int index = weedOut(sum - VoiceChatConstant.TOKENS, 1, chatMessageList);
                System.out.println("获取到的临界点下标---->" + index);
                for (int i = index; i < chatMessageList.size();i ++){
                    ChatMessage chatMessage = chatMessageList.get(i);
                    JSONObject jsonObject = new JSONObject();
                    //jsonObject.put("id",chatMessage.getId());
                    jsonObject.put("role",chatMessage.getSenderCode().equals(VoiceChatConstant.CHAT_CODE) ? VoiceChatConstant.CHAT_CODE : VoiceChatConstant.USER_CODE);
                    jsonObject.put("content",chatMessage.getMessage());
                    jsonArray.add(jsonObject);
                }
            } else {
                chatMessageList.forEach(c ->{
                    JSONObject jsonObject = new JSONObject();
                    jsonObject.put("role", c.getSenderCode().equals(VoiceChatConstant.CHAT_CODE) ? VoiceChatConstant.CHAT_CODE : VoiceChatConstant.USER_CODE);
                    jsonObject.put("content", c.getMessage());
                    jsonArray.add(jsonObject);
                });
            }
        }
        System.out.println(jsonArray);
    }
    /**
     * 淘汰老的聊天记录
     * @param  num 是集合总字数和设定限制的差值
     * @param  s 是最后要返回的临界值,也就是s前面的坐标都要被淘汰
     */
    int weedOut(int num, int s, List<ChatMessage> chatMessageList) {
        int sum = 0; // 集合聊天记录最前面部分tokens相加
        for (int i = 0; i < s; i++) {
            ChatMessage chatMessages = chatMessageList.get(i);
            // 计算要淘汰的元素的字数
            sum += chatMessages.getTokens();
        }
        // 判断要淘汰的元素的字数,如果不符合条件  下标继续下移  s + 1
        if (sum > num) {
            return s; // 最后要这个集合下标为s后面的
        } else {
            return weedOut(num, s + 1, chatMessageList);
        }
    }
}

分割线-----------------------------------------------------------------------------------------------------------------------------------------------------------

创作不易,三连支持一下吧 👍

最后的最后送大家一句话

白驹过隙,沧海桑田

相关文章
|
3月前
|
存储 安全 Java
【Java集合类面试二十五】、有哪些线程安全的List?
线程安全的List包括Vector、Collections.SynchronizedList和CopyOnWriteArrayList,其中CopyOnWriteArrayList通过复制底层数组实现写操作,提供了最优的线程安全性能。
|
3月前
|
安全
List集合特有功能
List集合特有功能
37 2
|
3月前
|
Java
【Java集合类面试二十三】、List和Set有什么区别?
List和Set的主要区别在于List是一个有序且允许元素重复的集合,而Set是一个无序且元素不重复的集合。
|
3月前
|
存储 Java
Java学习笔记 List集合的定义、集合的遍历、迭代器的使用
Java学习笔记 List集合的定义、集合的遍历、迭代器的使用
|
7天前
|
安全 Java 程序员
深入Java集合框架:解密List的Fail-Fast与Fail-Safe机制
本文介绍了 Java 中 List 的遍历和删除操作,重点讨论了快速失败(fail-fast)和安全失败(fail-safe)机制。通过普通 for 循环、迭代器和 foreach 循环的对比,详细解释了各种方法的优缺点及适用场景,特别是在多线程环境下的表现。最后推荐了适合高并发场景的 fail-safe 容器,如 CopyOnWriteArrayList 和 ConcurrentHashMap。
34 5
|
23天前
|
存储 分布式计算 NoSQL
大数据-40 Redis 类型集合 string list set sorted hash 指令列表 执行结果 附截图
大数据-40 Redis 类型集合 string list set sorted hash 指令列表 执行结果 附截图
23 3
|
2月前
|
NoSQL Java Redis
List集合按照由小到大排序或者由大到小排序
List集合按照由小到大排序或者由大到小排序
18 3
|
2月前
|
Python
Python量化炒股的获取数据函数— get_billboard_list()
Python量化炒股的获取数据函数— get_billboard_list()
33 0
|
3月前
|
存储 安全 Java
java集合框架复习----(2)List
这篇文章是关于Java集合框架中List集合的详细复习,包括List的特点、常用方法、迭代器的使用,以及ArrayList、Vector和LinkedList三种实现类的比较和泛型在Java中的使用示例。
java集合框架复习----(2)List
|
3月前
|
存储 安全 Java
java集合框架复习----(4)Map、List、set
这篇文章是Java集合框架的复习总结,重点介绍了Map集合的特点和HashMap的使用,以及Collections工具类的使用示例,同时回顾了List、Set和Map集合的概念和特点,以及Collection工具类的作用。
java集合框架复习----(4)Map、List、set