间隔至少K天,分配任务

package org.example;

import java.util.*;
public class Main{
    /**
     * 任务类
     * 【无需改动】
     */
    static class Task {
        //任务id
        Long taskId;
        //货柜名称
        String counterName;
        //配送任务类型=子任务类型之和
        Long taskType;

        public Task() {
        }

        public Long getTaskId() {
            return taskId;
        }

        public void setTaskId(Long taskId) {
            this.taskId = taskId;
        }

        public Long getTaskType() {
            return taskType;
        }

        public void setTaskType(Long taskType) {
            this.taskType = taskType;
        }

        public String getCounterName() {
            return counterName;
        }

        public void setCounterName(String counterName) {
            this.counterName = counterName;
        }
    }

    /**
     * 将输入的字符串任务转换为任务类的对象
     * 【无需改动】
     *
     * @param taskStr
     * @return
     */
    private static Task convertToTask(String taskStr) {
        String cleanTask = taskStr.substring(1, taskStr.length() - 1);
        String[] vars = cleanTask.split(",");
        String[] nameStr = vars[0].split(":");
        String[] taskIdStr = vars[1].split(":");
        String[] typeStr = vars[2].split(":");
        Task task = new Task();
        task.setCounterName(nameStr[1]);
        task.setTaskId(Long.parseLong(taskIdStr[1]));
        task.setTaskType(Long.parseLong(typeStr[1]));
        return task;
    }

    /**
     * ACM模式输入输出处理【无需改动】
     *
     * @param args
     */
    public static void main(String[] args) {
        //ACM模式输入
        Scanner in = new Scanner(System.in);
        List<Task> strArr = new ArrayList<>();
        int i = 0;
        int k = 0;
        while (in.hasNext()) {
            if (i == 0) {
                k = Integer.parseInt(in.nextLine());
            } else {
                strArr.add(convertToTask(in.nextLine()));
            }
            i++;
        }
        //题目逻辑计算
        int[] res = planTasks(strArr, k);
        //ACM模式输出
        System.out.println(Arrays.toString(res));
    }

    /**
     * 任务规划评估
     *
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * @param tasks 要规划的任务列表
     * @param k     间隔的天数
     * @return 一个数组a[], 比如[10,2,3] a[0]即10为做完最少的天数, a[1]即2为空闲的天数,a[2]即3为2k天内完成的校期子任务数量,
     */
    public static int[] planTasks(List<Task> tasks, int k) {
        //todo 请实现此函数
        Collections.sort(tasks,(a,b)->{
                if(a.getCounterName().equals(b.getCounterName())){
                    if(((a.getTaskType()&2)==2)&&((b.getTaskType()&2)==2)){
                        return (int)(a.getTaskId()-b.getTaskId());
                    }else{
                        if((a.getTaskType()&2)==2){
                            return -1;
                        }else if((b.getTaskType()&2)==2){
                            return 1;
                        }else{
                            return (int)(a.getTaskId()-b.getTaskId());
                        }
                    }
                }else{
                    return a.getCounterName().compareTo(b.getCounterName());
                }


        });
        Map<String,LinkedList<Integer>> map=new HashMap<>();
        for(int i=0;i<tasks.size();i++){
            Task cur=tasks.get(i);
            if(!map.containsKey(cur.getCounterName())){
                map.put(cur.getCounterName(),new LinkedList<>());
            }
            map.get(cur.getCounterName()).addLast(i);
        }
        int[] ret=new int[3];
        Map<String,Integer> lastMap=new HashMap<>();
        for(String key:map.keySet()){
            lastMap.put(key,-2);
        }
        int day=1;
        PriorityQueue<String> queue=new PriorityQueue<>((a,b)->{
            int n1=map.get(a).size();
            int n2=map.get(b).size();
            return n2-n1;
        });
        for(String key:map.keySet()){
            queue.add(key);
        }
        while(queue.size()>0){
            String cur=queue.poll();
            while((day-lastMap.get(cur)<3)){
                if(queue.size()==0){
                    cur=null;
                    break;
                }else {
                    cur = queue.poll();
                }
            }

            if(cur==null){
                ret[1]++;
            }else{
                int id=map.get(cur).removeFirst();
                lastMap.put(tasks.get(id).getCounterName(),day);
                if(day<=2*k){
                    if((tasks.get(id).getTaskType()&2)==2){
                        ret[2]++;
                    }
                }
            }
            day++;
            queue.clear();
            for(String key:map.keySet()){
                if(map.get(key).size()>0){
                    queue.add(key);
                }
            }

        }
        ret[0]=day-1;
        return ret;
    }
}
/*
2
{"counterName":"A","taskId":1,"taskType":3}
{"counterName":"A","taskId":2,"taskType":21}
{"counterName":"A","taskId":3,"taskType":3}
{"counterName":"B","taskId":4,"taskType":10}
{"counterName":"C","taskId":5,"taskType":11}
{"counterName":"C","taskId":6,"taskType":4}
7 1 4
 */
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值