发寿司
(测试用例总数:X,1.5 秒 (C/C++),2 秒 (Java)/内存限制:256 M,Stack 1 M)
丽雅开了一家寿司店,为了宣传店铺,她想在开业前准备举办一场活动,邀请N位当地居民免费品尝寿司。
为了增加活动的吸引力,她准备按照如下规则来发寿司。首先,让N名参与活动的人排成一队,每人抽取一张写着编号的纸。(编号可能相同。)如果某个人的编号比他前面以及后面的K个人的排序高(编号越小,排序越低),那么这个人拿到的寿司数量比排序低的人多。
丽雅准备按照上述规则来开展活动,同时希望发出去的寿司数量尽可能少。帮助丽雅求出此次活动需要准备的最少寿司数量。
假设当地有7名居民参加了活动,K为2。当7个人抽取的编号分别为 4、3、2、4、1、4、3时,第5个居民的排序为1,因为其他人的排序都比1高,所以他只能获得一个寿司。对于第3个居民,他前面两个人的排序都比他高,后面两个人中有一个居民抽取了更低的排序1,这个人获得了一个寿司,所以第3个居民会获得两个寿司。第7个居民也会获得两个寿司,因为他前面的两个人中有一个人的排序比他的更低,而那个排序更低的人获得了一个寿司。按这种方式计算,可以得出发给参与者的最少寿司数量为 19,如下所示。
1) 第 1 个居民:获得四个寿司
2) 第 2 个居民:获得三个寿司
3) 第 3 个居民:获得两个寿司
4) 第 4 个居民:获得四个寿司
5) 第 5 个居民:获得一个寿司
6) 第 6 个居民:获得三个寿司
7) 第 7 个居民:获得两个寿司
[限制条件]
1.参与者数量N为介于1到100,000之间的整数。
2.比较排序的人数K为介于1到100,000之间的整数。
3.每个参与者抽取的编号为介于1到N之间的整数。
(1是最低的编号,编号越大,排序越高。)
4.所有参与者至少会获得一个寿司。
[输入]
首先给出测试用例数量T,接着给出T种测试用例。每个测试用例的第一行,给出参与者数量N 和比较排序的人数K。第二行空格区分给出N名参与者抽取的编号。
[输出]
各测试用例输出一行。首先输出“#x”(x为测试用例的编号,从1开始),加一个空格后,输出丽雅需要准备的最少寿司数量。
[输入输出 示例]
(输入)
2
7 2
4 3 2 4 1 4 3
7 1
4 3 2 4 1 4 3
(输出)
#1 19
#2 12
思路分析:
根据题目意思,可知,需要按照排序大小计算,因此先对原数据进行排序,值正序,index逆序
然后求最长递增子序列,第3个用例分析可知,值相同的节点,前面的节点会影响后面的节点,因此需要延迟更新,即,遇到下一个值不同的节点时,再更新当前值的所有节点
用Indexed Tree求当前节点-K~当前节点+K个区间的LIS,加延迟更新
package tree; import java.io.BufferedReader; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStreamReader; import java.util.Arrays; import java.util.Comparator; import java.util.LinkedList; import java.util.Queue; import java.util.StringTokenizer; /** * 思路分析: * 根据题目意思可知,需要按照排序大小计算,因此先对原数据进行排序,值正序,index逆序 * 然后求最长递增子序列,分析用例可知,值相同的节点,前面的节点会影响后面的节点,因此需要延迟更新,即,遇到下一个值不同的节点时,再更新当前值的所有节点 * 用Indexed Tree求当前节点-K~当前节点+K个区间的LIS,加延迟更新 * @author XA-GDD * */ public class EQ_HandingOutSushi_0615 { static int T,N,K; static int _max_Nval = 100000; static int [][] srcArr = new int [_max_Nval][2]; static int [] lisIdxTree = new int[_max_Nval*4]; static int offset; static Queue<int []> queue = new LinkedList<int[]>(); //延迟更新用Queue,勿使用HashMap,效率太慢 static long ANS; public static void main(String[] args) throws IOException { System.setIn(new FileInputStream("D:\\workspace\\sw_pro\\test_case\\sample_input_0615.txt")); BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(br.readLine()); T = Integer.parseInt(st.nextToken()); for(int testCase = 1; testCase<=T;testCase++) { st = new StringTokenizer(br.readLine()); N = Integer.parseInt(st.nextToken()); K = Integer.parseInt(st.nextToken()); ANS=0L; for(int i=0;i<N;i++) { Arrays.fill(srcArr[i], 0); } Arrays.fill(lisIdxTree, 0); st = new StringTokenizer(br.readLine()); for(int i=0;i<N;i++) { srcArr[i][0] = i; srcArr[i][1] = Integer.parseInt(st.nextToken()); } //排序,value正序,index逆序 Arrays.sort(srcArr,0,N, new Comparator<int []>() { @Override public int compare(int[] o1, int[] o2) { if(o1[1]==o2[1]) { return o2[0] - o1[0]; } return o1[1] - o2[1]; } }); //初始化index tree int k=0; while((1<<k)<N) { k++; } offset = 1<<k; int preVal=0; for(int i=0;i<N;i++) { int currVal = srcArr[i][1]; if(currVal!=preVal) { //如果原数组当前节点和前一个节点不相同时,更新与前一个节点相同的所有节点的值 delayedUpdate(); //如果原数组当前节点和前一个节点不相同时,开始记录当前节点的值 queue.clear(); } int start = srcArr[i][0]-K>0?srcArr[i][0]-K:0; int end = srcArr[i][0]+K<offset-1?srcArr[i][0]+K:offset-1; int val = query(start+offset,end+offset); queue.add(new int[] {srcArr[i][0], val+1}); preVal = currVal; } delayedUpdate(); //更新最后一组值 queue.clear(); System.out.printf("#%d %d\n",testCase,ANS); } } static void delayedUpdate() { while(!queue.isEmpty()) { int obj [] = queue.poll(); update(offset+obj[0],obj[1]); ANS += (long)obj[1]; } } static void update(int index ,int val) { lisIdxTree[index] = val; index = index >>1; while(index>0) { lisIdxTree[index] = Math.max(lisIdxTree[index*2], lisIdxTree[index*2+1]); index = index >>1; } } static int query(int start ,int end) { int res=0; while(start<=end) { if(start%2==1) { res = Math.max(res, lisIdxTree[start]); } if(end%2==0) { res = Math.max(res, lisIdxTree[end]); } start = (start+1)>>1; end = (end-1)>>1; } return res; } }