yukicoder : No.1558 Derby Live

割と典型寄り。
https://yukicoder.me/problems/no/1558

問題

N頭の馬がM個の区間を順に走る。

最初の区間には1番～N番の馬が順に突入する。
以下のクエリを順次処理せよ。

• ある区間Dについて、i番目に突入した馬が何番目に出たという情報が与えられる。
• 指定された区間を出た馬の番号順を答える。
• 連続した区間[L,R]が与えられる。L番の区間にi番目に入った馬が、R番の区間をT[i]番目に出たとする。sum(|i-T[i]|)を答えよ。

解法

区間毎に馬の順番がどうシャッフルされるか、長さNの配列を持とう。
あとはそれをSegTreeの要領で隣接区間を合成していくだけ。

template<class V,int NV> class SegTree_1 {
public:
	vector<V> val;
	V def;
	// static V const def=1LL<<60; min
	V comp(V l,V r){
		vector<int> S(18);
		int i;
		FOR(i,18) S[i]=r[l[i]];
		return S;
	};
	
	SegTree_1(){
		def.resize(18);
		int i;
		FOR(i,18) def[i]=i;
		val.resize(NV*2);
		FOR(i,NV*2) val[i]=def;
	};
	V getval(int x,int y,int l=0,int r=NV,int k=1) { // x<=i<y
		if(r<=x || y<=l) return def;
		if(x<=l && r<=y) return val[k];
		return comp(getval(x,y,l,(l+r)/2,k*2),getval(x,y,(l+r)/2,r,k*2+1));
	}
	void update(int entry, V v) {
		entry += NV;
		val[entry]=v;
		while(entry>1) entry>>=1, val[entry]=comp(val[entry*2],val[entry*2+1]);
	}
};

SegTree_1<vector<int>,1<<16> st;
int N,M,Q;

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N>>M>>Q;
	while(Q--) {
		cin>>i;
		if(i==1) {
			cin>>x;
			vector<int> V(18);
			FOR(i,18) V[i]=i;
			FOR(i,N) {
				cin>>V[i];
				V[i]--;
			}
			st.update(x,V);
		}
		else if(i==2) {
			cin>>x;
			auto a=st.getval(1,x+1);
			vector<int> R(N);
			FOR(i,N) R[a[i]]=i+1;
			FORR(r,R) cout<<r<<" ";
			cout<<endl;
		}
		else {
			cin>>x>>y;
			auto a=st.getval(x,y+1);
			int ret=0;
			FOR(i,N) ret+=abs(i-a[i]);
			cout<<ret<<endl;
		}
		
	}
}

まとめ

type2の設問要るのかな。