題目分析:樹上啟發式合并,需要修改部分內部實現,如果可以想到樹啟的話,那么應該往子樹上去靠攏,當每個點作為子樹的根時,其可以作為 lca 然后去統計子樹中可以匹配的 ( u , v ) 點對,這個題目因為 a[ i ] != 0,換句話說,點 u , v , lca( u , v ) 一定是互不相同的三個點,極大程度上簡化了題目(因為三個點的形式一定是一個分叉的形狀,不可能是鏈狀的),換句話說,當 lca 確定后,可以枚舉每個點作為點 u,然后去統計“除了點 u 到點 lca 的這條路徑上的點之外,以?lca 為根節點的子樹中,所有 a[ v ] = a[ u ] ^ a[ lca ] 的點 v,然后對 v^u 加和”,但如果每個點都去枚舉的話,那么時間復雜度將會是 n * n 級別的,又因為如果每個點都枚舉一遍的話,換句話說對于一個有貢獻的 ( u , v ) 點對來說,點 u 和點 v 都會被枚舉一遍從而形成了浪費
下面講該如何實現,首先上面簡化的模型已經足夠可以完成公式,但本題實際上是要求記錄下標的異或和,比賽時沒有多想,直接對每個權值維護了一個?set,用于記錄當前權值下有多少個下標,維護答案的時候暴力枚舉即可,本以為時間復雜度是 nlog^2n 的,TLE 的原因是因為 set 的那一層 log,于是想到每次增加和刪除都是連續的一段,將 set 換成了 vector 便輕松將時間復雜度控制到了 nlogn 級別,交上去也真的 AC 了,賽后和潘學長還有zx學長討論過后才發現,自己實際上寫了個假算法。。因為如果暴力去維護下標的話,時間復雜度其實是 O( k * n * logn ),這里的 k 是,整棵樹中出現次數最多的數字的出現次數,如果整棵樹都是同一個數字的話,那么時間復雜度將退化成 O( n * n * logn ) 級別的。。然鵝感謝出題人,卑微菜雞在這里給你磕頭了,咚咚咚,多謝出題人沒有刻意去卡這種數據,導致在隨機數據下,常數 k 好像很小很小,甚至比 std 跑的還快。。
然后講一下正解吧,因為我們需要維護的是下標的異或和,而對于異或而言,非常重要的一個性質就是,拆位之后每一位都相互獨立,所以我們不妨對于下標的每一位都單獨跑一次樹啟,最后將答案加和即可,對于某一位來說,假設 u == 1,對于匹配到的 v 來說,只有 v == 0 的位置才具有貢獻,對于 u == 0 而言,同理只有 v == 1 才有貢獻,所以類比于上一段的思路,對于每個權值 val 來說,上一段的做法維護的是一個 vector,里面儲存著 a[ x ] == val 的 x,也就是下標,而本段的思路是,對于每個權值 val 來說,記錄一下所有下標,在二進制下第 i 位中共出現了多少個 0 和多少個 1,只是換了一下實現思路而已,這樣實現的時間復雜度是嚴格 O( nlogn * 20 ) 的,20 的意思是需要將下標拆成 20 位,因為 2^20 > 1e6 >= a[ i ]?