Commit 864245a

authored

Merge pull request #186 from atcoder/opt/dsu

fix dsu.leader to avoid multiple assert checks

2 parents 2df885d + ad80e62 commit 864245aCopy full SHA for 864245a

File tree

-2

lines changed

-2

lines changed

Lines changed: 6 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -35,8 +35,7 @@ struct dsu {`
`35`	`35`
`36`	`36`	`int leader(int a) {`
`37`	`37`	`assert(0 <= a && a < _n);`
`38`		`- if (parent_or_size[a] < 0) return a;`
`39`		`- return parent_or_size[a] = leader(parent_or_size[a]);`
	`38`	`+ return _leader(a);`
`40`	`39`	`}`
`41`	`40`
`42`	`41`	`int size(int a) {`
`@@ -69,6 +68,11 @@ struct dsu {`
`69`	`68`	`// root node: -1 * component size`
`70`	`69`	`// otherwise: parent`
`71`	`70`	`std::vector<int> parent_or_size;`
	`71`	`+`
	`72`	`+ int _leader(int a) {`
	`73`	`+ if (parent_or_size[a] < 0) return a;`
	`74`	`+ return parent_or_size[a] = _leader(parent_or_size[a]);`
	`75`	`+ }`
`72`	`76`	`};`
`73`	`77`
`74`	`78`	`} // namespace atcoder`

Comments

(0)