注意在弹出栈内元素时要将 $u$ 也弹出。

非

while (st[top - 1] != u) {
	int v = st[top --];
	ans[acnt] ++;
	color[v] = acnt;
	vis[v] = 0;
}

而是

while (1) {
	int v = st[top --];
	ans[acnt] ++;
	color[v] = acnt;
	vis[v] = 0;
	if (v == u) break;
}

区别：第二个代码将 $u$ 点也更新了。