結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
ユーザー threepipes_sthreepipes_s
提出日時 2016-10-28 19:31:38
言語 Java19
(openjdk 21)
結果
AC  
実行時間 548 ms / 4,000 ms
コード長 8,111 bytes
コンパイル時間 3,018 ms
コンパイル使用メモリ 88,812 KB
実行使用メモリ 74,376 KB
最終ジャッジ日時 2023-08-15 20:46:48
合計ジャッジ時間 19,742 ms
ジャッジサーバーID
(参考情報)
judge14 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 523 ms
74,376 KB
testcase_01 AC 484 ms
72,408 KB
testcase_02 AC 346 ms
64,148 KB
testcase_03 AC 429 ms
61,868 KB
testcase_04 AC 443 ms
61,720 KB
testcase_05 AC 384 ms
66,492 KB
testcase_06 AC 548 ms
74,068 KB
testcase_07 AC 400 ms
72,740 KB
testcase_08 AC 383 ms
65,760 KB
testcase_09 AC 375 ms
61,644 KB
testcase_10 AC 355 ms
61,576 KB
testcase_11 AC 380 ms
61,520 KB
testcase_12 AC 354 ms
61,576 KB
testcase_13 AC 307 ms
61,328 KB
testcase_14 AC 358 ms
61,544 KB
testcase_15 AC 255 ms
61,192 KB
testcase_16 AC 383 ms
61,440 KB
testcase_17 AC 436 ms
61,480 KB
testcase_18 AC 393 ms
61,928 KB
testcase_19 AC 285 ms
61,256 KB
testcase_20 AC 306 ms
61,424 KB
testcase_21 AC 299 ms
61,456 KB
testcase_22 AC 393 ms
61,352 KB
testcase_23 AC 329 ms
61,784 KB
testcase_24 AC 282 ms
61,188 KB
testcase_25 AC 366 ms
61,552 KB
testcase_26 AC 294 ms
61,680 KB
testcase_27 AC 380 ms
61,912 KB
testcase_28 AC 304 ms
61,228 KB
testcase_29 AC 342 ms
61,612 KB
testcase_30 AC 45 ms
49,628 KB
testcase_31 AC 42 ms
49,728 KB
testcase_32 AC 409 ms
61,740 KB
testcase_33 AC 422 ms
63,200 KB
testcase_34 AC 358 ms
61,928 KB
testcase_35 AC 48 ms
49,608 KB
testcase_36 AC 45 ms
49,484 KB
testcase_37 AC 50 ms
49,680 KB
testcase_38 AC 51 ms
49,636 KB
testcase_39 AC 84 ms
52,692 KB
testcase_40 AC 75 ms
52,620 KB
testcase_41 AC 134 ms
56,712 KB
testcase_42 AC 121 ms
57,020 KB
testcase_43 AC 128 ms
56,748 KB
testcase_44 AC 138 ms
56,768 KB
testcase_45 AC 90 ms
52,432 KB
testcase_46 AC 170 ms
56,740 KB
testcase_47 AC 169 ms
56,760 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.Closeable;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.PriorityQueue;
import java.util.Random;
import java.util.Scanner;
import java.util.TreeMap;

public class Main {
	static ContestScanner in;static Writer out;static StringBuilder sb=new StringBuilder();
	public static void main(String[] args)
	{try{in=new ContestScanner();out=new Writer();Main solve=new Main();solve.solve();
	in.close();out.flush();out.close();}catch(IOException e){e.printStackTrace();}}
	static void dump(int[]a){sb.setLength(0);for(int i=0;i<a.length;i++)
	sb.append(a[i]).append("\t");out.println(sb.toString().trim());}
	static void dump(int[]a,int n){for(int i=0;i<a.length;i++)out.printf("%"+n+"d ",a[i]);out.println();}
	static void dump(long[]a){for(int i=0;i<a.length;i++)out.print(a[i]+" ");out.println();}
	static long pow(long a,int n){long r=1;while(n>0){if((n&1)==1)r*=a;a*=a;n>>=1;}return r;}
	static String itob(int a,int l){return String.format("%"+l+"s",Integer.toBinaryString(a)).replace(' ','0');}
	static void sort(int[]a){m_sort(a,0,a.length,new int[a.length]);}
	static void sort(int[]a,int l){m_sort(a,0,l,new int[l]);}
	static void sort(int[]a,int l,int[]buf){m_sort(a,0,l,buf);}
	static void sort(int[]a,int s,int l,int[]buf){m_sort(a,s,l,buf);}
	static void m_sort(int[]a,int s,int sz,int[]b)
	{if(sz<7){for(int i=s;i<s+sz;i++)for(int j=i;j>s&&a[j-1]>a[j];j--)swap(a, j, j-1);return;}
	m_sort(a,s,sz/2,b);m_sort(a,s+sz/2,sz-sz/2,b);int x=s;int l=s,r=s+sz/2;final int le=s+sz/2,re=s+sz;
	while(l<le&&r<re){if(a[l]>a[r])b[x++]=a[r++];else b[x++]=a[l++];}
	while(r<re)b[x++]=a[r++];while(l<le)b[x++]=a[l++];for(int i=s;i<s+sz;i++)a[i]=b[i];
	} /* qsort(3.5s)<<msort(9.5s)<<<shuffle+qsort(17s)<Arrays.sort(Integer)(20s) */
	static void sort(long[]a){m_sort(a,0,a.length,new long[a.length]);}
	static void sort(long[]a,int l){m_sort(a,0,l,new long[l]);}
	static void sort(long[]a,int l,long[]buf){m_sort(a,0,l,buf);}
	static void sort(long[]a,int s,int l,long[]buf){m_sort(a,s,l,buf);}
	static void m_sort(long[]a,int s,int sz,long[]b)
	{if(sz<7){for(int i=s;i<s+sz;i++)for(int j=i;j>s&&a[j-1]>a[j];j--)swap(a, j, j-1);return;}
	m_sort(a,s,sz/2,b);m_sort(a,s+sz/2,sz-sz/2,b);int x=s;int l=s,r=s+sz/2;final int le=s+sz/2,re=s+sz;
	while(l<le&&r<re){if(a[l]>a[r])b[x++]=a[r++];else b[x++]=a[l++];}
	while(r<re)b[x++]=a[r++];while(l<le)b[x++]=a[l++];for(int i=s;i<s+sz;i++)a[i]=b[i];}
	static void swap(long[]a,int i,int j){final long t=a[i];a[i]=a[j];a[j]=t;}
	static void swap(int[]a,int i,int j){final int t=a[i];a[i]=a[j];a[j]=t;}
	static int binarySearchSmallerMax(int[]a,int v)// get maximum index which a[idx]<=v
	{int l=-1,r=a.length-1,s=0;while(l<=r){int m=(l+r)/2;if(a[m]>v)r=m-1;else{l=m+1;s=m;}}return s;}
	static int binarySearchSmallerMax(int[]a,int v,int l,int r)
	{int s=-1;while(l<=r){int m=(l+r)/2;if(a[m]>v)r=m-1;else{l=m+1;s=m;}}return s;}
	List<Integer>[]randomTree(int n,Random r){List<Integer>[]g=graph(n);
	for(int i=1;i<n;i++){int p=r.nextInt(i);g[p].add(i);g[i].add(p);}return g;}
	@SuppressWarnings("unchecked")
	static List<Integer>[]graph(int n){List<Integer>[]g=new List[n];
	for(int i=0;i<n;i++)g[i]=new ArrayList<>();return g;}
	void solve() throws NumberFormatException, IOException{
		int n = in.nextInt();
		int k = in.nextInt();
		int[] map = new int[100001];
		Arrays.fill(map, -1);
		List<Univ> univ = new ArrayList<>();
		int idx = 0;
		int[] s = new int[n];
		int[] p = new int[n];
		int[] u = new int[n];
		for(int i=0; i<n; i++){
			s[i] = in.nextInt();
			p[i] = in.nextInt();
			u[i] = in.nextInt();
			Team t = new Team(s[i], p[i], i);
			if(map[u[i]]==-1){
				Univ un = new Univ(idx);
				univ.add(un);
				un.qu.add(t);
				map[u[i]] = idx++;
			}else{
				Univ un = univ.get(map[u[i]]);
				un.qu.add(t);
			}
		}
		PriorityQueue<Univ> uq = new PriorityQueue<>();
		uq.addAll(univ);
		for(int i=0; i<k; i++){
			Univ un = uq.poll();
			Team t = un.qu.poll();
			un.chosen++;
			if(!un.qu.isEmpty()) uq.add(un);
			out.println(t.id);
		}
	}
}

class Univ implements Comparable<Univ>{
	PriorityQueue<Team> qu = new PriorityQueue<Team>();
	int chosen, id;
	public Univ(int id) {
		this.id = id;
	}
	@Override
	public int compareTo(Univ o) {
		if(qu.isEmpty()) return 1;
		Team me = qu.peek();
		Team en = o.qu.peek();
		if(me.s!=en.s) return en.s-me.s;
		if(chosen!=o.chosen) return chosen-o.chosen;
		return me.p-en.p;
	}
}

class Team implements Comparable<Team>{
	int s, p, id;
	public Team(int s, int p, int id) {
		this.s = s;
		this.p = p;
		this.id = id;
	}
	@Override
	public int compareTo(Team o) {
		if(s!=o.s)return o.s-s;
		return p-o.p;
	}
}

@SuppressWarnings("serial")
class MultiSet<T> extends HashMap<T, Integer>{
	@Override public Integer get(Object key){return containsKey(key)?super.get(key):0;}
	public void add(T key,int v){put(key,get(key)+v);}
	public void add(T key){put(key,get(key)+1);}
	public void sub(T key){final int v=get(key);if(v==1)remove(key);else put(key,v-1);}
	public MultiSet<T> merge(MultiSet<T> set)
	{MultiSet<T>s,l;if(this.size()<set.size()){s=this;l=set;}else{s=set;l=this;}
	for(Entry<T,Integer>e:s.entrySet())l.add(e.getKey(),e.getValue());return l;}
}
@SuppressWarnings("serial")
class OrderedMultiSet<T> extends TreeMap<T, Integer>{
	@Override public Integer get(Object key){return containsKey(key)?super.get(key):0;}
	public void add(T key,int v){put(key,get(key)+v);}
	public void add(T key){put(key,get(key)+1);}
	public void sub(T key){final int v=get(key);if(v==1)remove(key);else put(key,v-1);}
	public OrderedMultiSet<T> merge(OrderedMultiSet<T> set)
	{OrderedMultiSet<T>s,l;if(this.size()<set.size()){s=this;l=set;}else{s=set;l=this;}
	while(!s.isEmpty()){l.add(s.firstEntry().getKey(),s.pollFirstEntry().getValue());}return l;}
}
class Pair implements Comparable<Pair>{
	int a,b;final int hash;Pair(int a,int b){this.a=a;this.b=b;hash=(a<<16|a>>16)^b;}
	public boolean equals(Object obj){Pair o=(Pair)(obj);return a==o.a&&b==o.b;}
	public int hashCode(){return hash;}
	public int compareTo(Pair o){if(a!=o.a)return a<o.a?-1:1;else if(b!=o.b)return b<o.b?-1:1;return 0;}
	@Override
	public String toString() {
		return String.format("(%d, %d)", a, b);
	}
}
class Timer{
	long time;public void set(){time=System.currentTimeMillis();}
	public long stop(){return time=System.currentTimeMillis()-time;}
	public void print(){System.out.println("Time: "+(System.currentTimeMillis()-time)+"ms");}
	@Override public String toString(){return"Time: "+time+"ms";}
}
class Writer extends PrintWriter{
	public Writer(String filename)throws IOException
	{super(new BufferedWriter(new FileWriter(filename)));}
	public Writer()throws IOException{super(System.out);}
}
class ContestScanner implements Closeable{
	private BufferedReader in;private int c=-2;
	public ContestScanner()throws IOException 
	{in=new BufferedReader(new InputStreamReader(System.in));}
	public ContestScanner(String filename)throws IOException
	{in=new BufferedReader(new InputStreamReader(new FileInputStream(filename)));}
	public String nextToken()throws IOException {
		StringBuilder sb=new StringBuilder();
		while((c=in.read())!=-1&&Character.isWhitespace(c));
		while(c!=-1&&!Character.isWhitespace(c)){sb.append((char)c);c=in.read();}
		return sb.toString();
	}
	public String readLine()throws IOException{
		StringBuilder sb=new StringBuilder();if(c==-2)c=in.read();
		while(c!=-1&&c!='\n'&&c!='\r'){sb.append((char)c);c=in.read();}
		return sb.toString();
	}
	public long nextLong()throws IOException,NumberFormatException
	{return Long.parseLong(nextToken());}
	public int nextInt()throws NumberFormatException,IOException
	{return(int)nextLong();}
	public double nextDouble()throws NumberFormatException,IOException 
	{return Double.parseDouble(nextToken());}
	public void close() throws IOException {in.close();}
}
0