POJ 2155 : Matrix
問題
N*N()の配列が与えられる。各要素は、1か0しか値をとることができない。ふたつのクエリに対する処理を実行せよ。
- (x,y)の要素が1なのか0なのか求めよ
- (x1,y1,x2,y2)の範囲をフリップせよ
解法
二次元のBITを使い、範囲操作をするために4つの二次元BITを用意する。以前の問題(POJ3109)を二次元に拡張させる。境界条件は、一次元の場合とほとんど変わらないのでつまずかなかった。
計算量
N : 与えれる要素
T : クエリの数
X : テスト数
コード
POJ 2886 : Who Gets the Most Candies?
問題
N()人の名前と値が与えられる。初めにK番目の人(1から始まる)からスタートし、選ばれた人は抜ける。抜ける際に、p番目の人はpの約数だけ飴がもらえる。次の人は、その人の値分だけ時計回りした番号の人になる。マイナスの場合は逆回転となる。飴の最大数とその名前を求めよ。
解法
Binary Indexed Tree(BIT)、二分探索、エラトステネスの篩っぽい約数の個数を求める前処理で解ける。約数の数を求める前処理はkomiyamさんの記事を参考にした。他の箇所もいくつか参考にしたので、同じようなコードになっている。
BITを用いることで、途中で人が抜けた場合でも次の順番の人を計算することができる。初めに、0~Nまでに+1をする。その後、抜けていく番号の部分に-1をする。こうすることで、で次の人が計算可能になる。
例を用いて説明する。6人いるとする。この時のBITの各要素は以下となる。
その後、2番目と5番目の人が抜けたとする。
こうすることで、抜けた人の分を考慮したインデックスをsumを用いることで求めることができる。抜けた人の分を考慮したインデックスから本当のインデックスを求めるには、二分探索を用いれば良い。
約数の個数を前処理
約数の個数はナイーブに実装すると、ある値の約数の個数を求める場合の計算量はなので、となる。このアルゴリズムを用いると、大雑把に見積もってとなる。しかし、これは少なく見積もっている。
アルゴリズムの大まかな原理を説明する。
合成数2700を考える。2700は以下の素因数を持っている。
2,2,3,3,3,5,5
この素因数を用いた積の組み合わせの合計が約数の個数となる。素因数を用いているので積が同じになることはない。
2の時のパターンは
1 2 4
3の時のパターンは
1 3 9 27
5の時のパターンは
1 5 25
となり、これを計算すると3*4*3となり36となる。
計算量
N : 要素数
コード
POJ 3264 : Balanced Lineup
問題
要素数N()の数列が与えれる。[a,b]区間の最大値と最小値の差分を求めなさい。
解法
セグメント木を使う。最大値を保持する配列と最小値を保持する配列の2つを持つ。
計算量
更新:
クエリ: