试题 G: 外卖店优先级
时间限制: 1.0s 内存限制: 512.0MB 本题总分:20 分
【问题描述】
“饱了么”外卖系统中维护着 N 家外卖店,编号 1 ∼ N。每家外卖店都有 一个优先级,初始时 (0 时刻) 优先级都为 0。 每经过 1 个时间单位,如果外卖店没有订单,则优先级会减少 1,最低减 到 0;而如果外卖店有订单,则优先级不减反加,每有一单优先级加 2。 如果某家外卖店某时刻优先级大于 5,则会被系统加入优先缓存中;如果 优先级小于等于 3,则会被清除出优先缓存。 给定 T 时刻以内的 M 条订单信息,请你计算 T 时刻时有多少外卖店在优 先缓存中。
【输入格式】
第一行包含 3 个整数 N、M 和 T。 以下 M 行每行包含两个整数 ts 和 id,表示 ts 时刻编号 id 的外卖店收到 一个订单。
【输出格式】
输出一个整数代表答案。
【样例输入】
2 6 6 1 1 5 2 3 1 6 2 2 1 6 2
【样例输出】
1
【样例解释】
6 时刻时,1 号店优先级降到 3,被移除出优先缓存;2 号店优先级升到 6, 加入优先缓存。所以是有 1 家店 (2 号) 在优先缓存中。
【评测用例规模与约定】
对于 80% 的评测用例,1≤ N,M,T ≤10000。 对于所有评测用例,1≤ N,M,T ≤100000,1≤ts≤T,1≤id ≤ N。
题解
import java.util.Scanner;
import java.util.TreeSet;
public class Main {
public static void main(String[] args) {
int N,M,T;
Scanner sc = new Scanner(System.in);
N = sc.nextInt();
M = sc.nextInt();
T = sc.nextInt();
int[][] array = new int[N+1][T+1];
TreeSet<Integer> set = new TreeSet<>();
while(M-->0)
{
int a = sc.nextInt(),b = sc.nextInt();
array[b][a] ++;
}
int ans = 0;
for(int i=1;i<=N;i++)
{
int times = 0;
int flag = 0;
for(int j=1;j<=T;j++)
{
if(array[i][j]==0) times = times>0?times-=1:0;
else
times+= 2*array[i][j];
if(times>5)
set.add(i);
if(times<=3 && set.contains(i))
set.remove(i);
}
}
System.out.println(set.size());
}
}