結果

問題 No.2316 Freight Train
コンテスト
ユーザー watarimaycry2
提出日時 2023-06-08 22:36:51
言語 JavaScript
(node v23.5.0)
結果
AC  
実行時間 947 ms / 2,000 ms
コード長 4,977 bytes
コンパイル時間 120 ms
コンパイル使用メモリ 6,820 KB
実行使用メモリ 109,944 KB
最終ジャッジ日時 2024-12-31 06:56:17
合計ジャッジ時間 24,648 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

//Don't have to see. start------------------------------------------
var read = require('readline').createInterface({
	input: process.stdin, output: process.stdout
});
var obj; var inLine = []; var outputList = [];var retcode = new Set();
read.on('line', function(input){
	var tmp = input.split(' ');
	for(var i = 0; i < tmp.length; i++){
		inLine.push(tmp[i]);
		if(i == tmp.length - 1){
			retcode.add(inLine.length);
		}
	}
});
read.on('close', function(){
	obj = init(inLine);
	console.error('\n↑入力 ↓出力');
	Main();
	console.log(myconv(outputList, 9));
});
function makeClone(obj){return (obj instanceof Set) ? new Set(Array.from(obj)) : JSON.parse(JSON.stringify(obj));}
function nextArray(size, code){
	var ret = new Array(size);
	for(var i = 0; i < size; i++){
		if(code == 'int'){
			ret[i] = nextInt();
		}else if(code == 'long'){
			ret[i] = nextLong();
		}else if(code == 'double'){
			ret[i] = nextDouble();
		}else{
			ret[i] = next();
		}
	}
	return ret;
}
function nextIntArray(size){return nextArray(size, 'int');} function nextStrArray(size){return nextArray(size, 'str');} function nextLongArray(size){return nextArray(size, 'long');} function nextDoubleArray(size){return nextArray(size, 'double');}
function nextCharArray(){return myconv(next(),6);}
function next(){return obj.next();} function hasNext(){return obj.hasNext();} function nextInt(){return myconv(next(),1);} function nextLong(){return BigInt(next());} function nextDouble(){return parseFloat(next());}
function getCountMap(list){
	var map = {};
	for(var i = 0; i < list.length; i++){
		if(map[list[i]] == null){
			map[list[i]] = 0;
		}
		map[list[i]]++;
	}
	return map;
}
function init(input){  
	return {
		list : input, index : 0, max : input.length,
		hasNext : function(){return (this.index < this.max);},
		next : function(){if(this.hasNext()){return this.list[this.index++];}else{throw 'ArrayIndexOutOfBoundsException ‚There is no more input';}},
		isReturn : function(){return retcode.has(this.index);}
	};
}
function myout(s){outputList.push(s);}
function myerr(s){console.error('debug:' + require('util').inspect(s,false,null));}
function isReturn(){return obj.isReturn();}
//param "no" is
//unknown or outlier : return i. 1: parseInt.
//2: split space. 4: split space and parseInt.
//6: split 1 character. 7: split 1 character and parseInt.
//8: join space. 9: join nextline. 0: join no character.
function myconv(i,no){try{switch(no){case 1:return parseInt(i);case 2:return i.split(' ');case 4:return i.split(' ').map(Number);case 6:return i.split('');case 7:return i.split('').map(Number);case 8:return i.join(' ');case 9:return i.join('\n');case 0:return i.join('');default:return i;}}catch(e){return i;}}

//Don't have to see. end------------------------------------------
function Main(){
	var N = nextInt();
	var Q = nextInt();
	var list = nextIntArray(N);
	var uf = new UnionFind(N);
	for(var i = 0; i < N; i++){
		if(list[i] != -1){
			uf.doUnion(i, list[i] - 1);
		}
	}
	for(var i = 0; i < Q; i++){
		var A = nextInt() - 1;
		var B = nextInt() - 1;
		if(uf.isSame(A, B)){
			myout("Yes");
		}else{
			myout("No");
		}
	}
}
function UnionFind(n){
	var uf = {
		//全てのインデックスは「-X(親、絶対値はグループの大きさ)」「自分が属する親(≒根)」のいずれかを持つ。
		//最初はみんなグループの大きさ1の親
		'list' : new Array(n).fill(-1),
		'reset' : new Array(n),
		'group' : n,
		//同じ親を持つか
		'isSame' : function(L, R){
			return this.getRootIndex(L) == this.getRootIndex(R);
		},
		//自身の親インデックスをたどって根っこに着く
		//親にたどり終えたら子に帰っていくついでに、子たちに「一番上の親は誰か」を知らせる
		'getRootIndex' : function(index){
			var count = 0;
			while(this.list[index] >= 0){
				this.reset[count] = index;
				index = this.list[index];
				count++;
			}
			for(var i = 0; i < count; i++){
				this.list[this.reset[i]] = index;
			}
			return index;
		},
		//異なる親同士のまとまりを一つにする（同じ親ならスルー）
		//小さいグループの親が大きいグループの親の直下に着く
		//グループの大きさも更新する
		'doUnion' : function(L, R){
			var Lroot = this.getRootIndex(L);
			var Rroot = this.getRootIndex(R);
			if(Lroot != Rroot){
				this.group--;
				if(this.getSize(Lroot) >= this.getSize(Rroot)){
					var tmp = Lroot;
					Lroot = Rroot;
					Rroot = tmp;
				}
				this.list[Lroot] += this.list[Rroot];
				this.list[Rroot] = Lroot;
			}
		},
		//「-X(親、絶対値はグループの大きさ)」
		//なので、インデックスを指定→親を知る→親の持つグループの大きさがわかる。
		//ただしマイナス値なので、掛け算して返す。
		'getSize' : function(index){
			return -this.list[this.getRootIndex(index)];
		},
		'getGroupCount' : function(){
			return this.group;
		}
	}
	return uf;
}
0