[1800] CF1466E 式子变形 拆位

Author: EndlessCheng

main/1400-1499/1466E.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"bufio"
+	. "fmt"
+	"io"
+	"os"
+)
+
+// github.com/EndlessCheng/codeforces-go
+func CF1466E(_r io.Reader, _w io.Writer) {
+	in := bufio.NewReader(_r)
+	out := bufio.NewWriter(_w)
+	defer out.Flush()
+	const mod int64 = 1e9 + 7
+
+	var T, n int64
+	for Fscan(in, &T); T > 0; T-- {
+		Fscan(in, &n)
+		var p2, c [60]int64
+		a := make([]int64, n)
+		for i := range a {
+			Fscan(in, &a[i])
+			for j := range c {
+				c[j] += a[i] >> j & 1
+			}
+		}
+		for i := range c {
+			p2[i] = 1 << i % mod * n % mod
+			c[i] = 1 << i % mod * c[i] % mod
+		}
+		ans := int64(0)
+		for _, v := range a {
+			var x, y int64
+			for j, c := range c {
+				if v>>j&1 > 0 {
+					x += c
+					y += p2[j]
+				} else {
+					y += c
+				}
+			}
+			ans = (ans + x%mod*(y%mod)) % mod
+		}
+		Fprintln(out, ans)
+	}
+}
+
+func main() { CF1466E(os.Stdin, os.Stdout) }

main/1400-1499/1466E_test.go

@@ -0,0 +1,41 @@
+package main
+
+import (
+	"github.com/EndlessCheng/codeforces-go/main/testutil"
+	"testing"
+)
+
+// https://codeforces.com/problemset/problem/1466/E
+// https://codeforces.com/problemset/status/1466/problem/E
+func TestCF1466E(t *testing.T) {
+	// just copy from website
+	rawText := `
+inputCopy
+8
+2
+1 7
+3
+1 2 4
+4
+5 5 5 5
+5
+6 2 2 1 0
+1
+0
+1
+1
+6
+1 12 123 1234 12345 123456
+5
+536870912 536870911 1152921504606846975 1152921504606846974 1152921504606846973
+outputCopy
+128
+91
+1600
+505
+0
+1
+502811676
+264880351`
+	testutil.AssertEqualCase(t, rawText, 1, CF1466E)
+}