codechef 3D Queries(MGCH3D)

資瓷點這裏閱讀QvQ

題外話

這個題是cc 9月challenge的最後一題，當時沒仔細考慮，以爲這是個近似計算的題。。。賽後看了題解才猛然醒悟

codechef 3D Queries(MGCH3D)

Description

求∑i!=j|A(Xi−Xj)+B(Yi−Yj)+C(Zi−Zj)+D|N(N−1)(Xi−Xj)4+(Yi−Yj)4+(Zi−Zj)4√
2≤N≤777777,1≤X,Y,Z,A,B,C≤77 , Q(1≤Q≤77) 組詢問,每次給A,B,C,D , 求上式的值。

Solution

x,y,z 座標都很小(1≤x,y,z≤77 )，差的取值也很小，不妨構造多項式A[N2×x+N×y+z]+=1 , B[N2×(77−x)+N×(77−y)+(77−z)]+=1 。令N=77∗2
我們發現A×B=N2×(77+x[i]−x[j])+N×(77+y[i]−y[j])+(77+z[i]−z[j])
顯然x[i]−x[j],y[i]−y[j],z[i]−z[j] 我們可以通過，於是這題FFT一下就做完了= =

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
#define F first
#define S second
typedef long long LL;
typedef pair<int, int> pii;
const int N = 153;
const double PI = acos(-1.0);
double D[80][80][80];
namespace FFT {
  static const int W = 1 << 22;
  static const double PI = acos(-1.0);
  struct Complex {
    double x, y;
    Complex(double _x = 0, double _y = 0) : x(_x), y(_y) {}
    Complex operator + (const Complex &t) const {return Complex(x + t.x, y + t.y);}
    Complex& operator += (const Complex &t) {x += t.x, y += t.y;return *this;}
    Complex operator - (const Complex &t) const {return Complex(x-t.x,y-t.y);}
    Complex& operator -= (const Complex &t) {x -= t.x, y -= t.y;return *this;}
    Complex operator * (const Complex &t) const {return Complex(x * t.x - y * t.y, x * t.y + y * t.x);}
    Complex operator / (const double &t) const {return Complex(x / t,y / t);}
    Complex& operator /= (const double &t) {x /= t, y /= t;return *this;}
    double real() {return x;}
    double imag() {return y;}
  };
  void fft(Complex a[], int n, int rev) {// rev=-1, reverse
      for (int i = 1, j = 0, k; i < n; ++i) {
          for (k = n >> 1; k > (j ^= k); k >>= 1);
          if (i < j) swap(a[i], a[j]);
      }
      for (int s = 1, ss = 2; s < n; s <<= 1,ss <<= 1) {
          Complex wn(cos(2 * PI * rev / ss), sin(2 * PI * rev / ss)), w;
          for (int i = 0, j ; i < n; i += ss) {
              for (j = i, w = 1; j < i + s; ++j, w = w * wn) {
                  Complex t = w * a[j + s];
                  a[j + s] = a[j] - t; 
                  a[j] = a[j] + t;
              }
          }
      }
      if (rev == -1) for (int i = 0; i < n; ++i) a[i] /= n;
  }
}
FFT::Complex A[FFT::W], B[FFT::W];
inline double sqrr(double x) {
    return x * x * x * x;
}
int main() {
    int n, q;
    scanf("%d%d", &n, &q);
    for (int i = 1, x, y, z; i <= n; ++i) {
        scanf("%d%d%d", &x, &y, &z);
        --x, --y, --z;
        int t1 = N * N * x + N * y + z, t2 = N * N * (76 - x) + N * (76 - y) + (76 - z);
        A[t1].x += 1, B[t2].x += 1;
    }
    for (int i = 0; i <= 77; ++i)
        for (int j = 0; j <= 77; ++j)
            for (int k = 0; k <= 77; ++k)
                D[i][j][k] = sqrt(sqrr(i) + sqrr(j) + sqrr(k));
    FFT::fft(A, FFT::W, 1);
    FFT::fft(B, FFT::W, 1);
    for (int i = 0; i < FFT::W; ++i)    A[i] = A[i] * B[i];
    FFT::fft(A, FFT::W, -1);
    while (q--) {
        double ans = 0;
        int a, b, c, d;
        scanf("%d%d%d%d", &a, &b, &c, &d);
        for (int i = 0; i < FFT::W; ++i) {
            int cnt = int(A[i].real() + 0.5);
            if (!cnt)   continue;
            int x = i / (N * N) - 76, y = i / N % N - 76, z = i % N - 76;
            if (!x && !y && !z) continue;
            ans += fabs(a * x + b * y + c * z + d) * cnt / D[abs(x)][abs(y)][abs(z)];
        }
        ans /= n, ans /= n - 1;
        printf("%.10lf\n", ans);
    }
    return 0;
}

總結

自己還是太年輕，一看範圍小就想着想忽略近似計算和亂搞，犯了方向性的錯誤= =。。還是要多加思考QvQ