結果

問題 No.59 鉄道の旅
コンテスト
ユーザー 37zigen
提出日時 2016-05-16 01:16:08
言語 Java
(openjdk 23)
結果
AC  
実行時間 715 ms / 5,000 ms
コード長 983 bytes
コンパイル時間 2,650 ms
コンパイル使用メモリ 77,120 KB
実行使用メモリ 52,612 KB
最終ジャッジ日時 2024-12-24 22:19:50
合計ジャッジ時間 8,698 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

package yukicoder;
import java.util.Scanner;
public class Main{
	public static void main(String[] args){
		new Main().solve();
	}
	void solve(){
		Scanner sc=new Scanner(System.in);
		int n=sc.nextInt();
		int k=sc.nextInt();
		FenwickTree ft=new FenwickTree(1_000_000);
		for(int i = 0; i < n; i++){
			int w=sc.nextInt();
			if(w>0){
				if(ft.sum(w, 1_000_000)>=k)continue;
				else{
					ft.add(w, 1);
				}
			}else if(w<0){
				if(ft.sum(-w, -w)>0){
					ft.add(-w, -1);
				}
			}
		}
		System.out.println(ft.sum(1_000_000));
	}
	class FenwickTree{
		int[] bit;
		int n;

		public FenwickTree(int n){
			this.n = n;
			bit = new int[n + 1];
		}

		//xにiを加える。
		void add(int i, int x)
		{
			while(i <= n){
				bit[i] += x;
				i += i & -i;
			}
		}
		//iまでの和を求める。
		int sum(int i)
		{
			int sum = 0;
			while(i > 0){
				sum += bit[i];
				i -= i & -i;
			}
			return sum;
		}
		int sum(int l, int r)
		{
			return sum(r) - sum(l - 1);
		}
	}
}
0