結果

問題 No.789 範囲の合計
ユーザー 37zigen
提出日時 2018-03-18 13:42:07
言語 Java
(openjdk 23)
結果
RE  
実行時間 -
コード長 1,485 bytes
コンパイル時間 2,418 ms
コンパイル使用メモリ 79,684 KB
実行使用メモリ 78,600 KB
最終ジャッジ日時 2024-06-25 06:03:17
合計ジャッジ時間 14,837 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 1 RE * 9 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

class Main{
	public static void main(String[] args){
		new Main().run();
	}
	
	void run(){
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int[][] a=new int[n][];
		ArrayList<Integer> list=new ArrayList<>();
		for(int i=0;i<n;++i){
			int q=sc.nextInt();
			if(q==0){
				a[i]=new int[2];
				a[i][0]=q;
				a[i][1]=sc.nextInt();
				list.add(a[i][1]);
			}else if(q==1){
				a[i]=new int[3];
				a[i][0]=q;
				a[i][1]=sc.nextInt();
				a[i][2]=sc.nextInt();
				list.add(a[i][1]);
				list.add(a[i][2]);
			}
		}
		Collections.sort(list);
		HashMap<Integer,Integer> map=new HashMap<>();
		int gen=0;
		for(int i=0;i<list.size();++i){
			int j=i;
			while(j+1<list.size()&&list.get(i)==list.get(j+1))++j;
			map.put(list.get(i),gen++);				
		}
		long ret=0;
		BIT bit=new BIT(gen);
		for(int i=0;i<n;++i){
			if(a[i][0]==0){
			    a[i][1]=map.get(a[i][1]);
			}else{
				a[i][1]=map.get(a[i][1]);
				a[i][2]=map.get(a[i][2]);
			}
		}
		for(int i=0;i<n;++i){
			if(a[i][0]==0){
				bit.add(a[i][1],1);
			}else{
				ret+=bit.sum(a[i][1],a[i][2]);
			}
		}
		System.out.println(ret);
	}
	
	class BIT{
		int n;
		long[] arr;
		public BIT(int n_){
			n=n_;
			arr=new long[n_+1];
		}
		
		void add(int k,long val){
			++k;
			while(k<=n){
				arr[k]+=val;
				k+=k&-k;
			}
		}

		long sum(int k){
			++k;
			long ret=0;
			while(k>0){
				ret+=arr[k];
				k-=k&-k;
			}
			return ret;
		}

		long sum(int l,int r){
			return sum(r)-sum(l-1);
		}
	}
}
0