[计蒜客16958] Colored Graph [2017 ACM-ICPC 亚洲区（乌鲁木齐赛区）网络赛 I]

这道题有个同学先暴力跑答案，然后对答案矩阵观察找规律A掉这题真的厉害Orz

题意

n点完全图，对边黑白染色，要求使得三边颜色相等的三元组最少，输出最少数量和染色方案。

题解

所有的三元组为$C_n^3$，假设一开始所有的边都是白色的，现在将一些边染黑，那么不在答案中的三角形必然是两白一黑或两黑一白。设一个点连出去的黑边为$d_i$个，那么包含这个点且不在答案中的三元组个数为$d_i\cdot (n-1-d_i)$，思考一下发现，无论是两白一黑还是两黑一白都会被这个式子统计两遍，于是，答案为

$$minimize \ C_n^3-\frac{\sum{d_i\cdot (n-1-d_i)}}{2}$$
则 $$maxmize \sum{d_i\cdot (n-1-d_i)}$$
则当 $d_i$接近于 $\frac{n-1}2$时最优。
下面分情况讨论：

当$n$为偶数时：

$n-1$为奇数，对每个点$d_i=\lfloor \frac{n-1}{2}\rfloor=\frac{n-2}{2}$或$\lceil\frac{n-1}{2}\rceil=\frac{n}{2}$都是最优的，我们取$d_i=\frac{n}{2}$，那么将点集分为两部分，每个点向另一个点集所有点连边即可满足条件。

当$n$为奇数时：

当$\frac{n-1}{2}$为偶数时：

每个点向它之后的$\frac{n-1}{4}$个点连边即可

当$\frac{n-1}{2}$为奇数时：

因为如果$d_i=\frac{n-1}{2}$则点的总度数为奇数，这是不合法的。因此需要使一个点为$d_m=\frac{n-1}{2}+1=\frac{n+1}{2}$。这个点定为第$m=\frac{n+1}{2}$个点。让前$\frac{d_m}{2}$个点和最后$\frac{d_m}{2}$个点向点$m$连边，这样，点$m$的度数满足了，而点$1,2,\dots,\frac{d_m}{2},n-\frac{d_m}{2}+1,n-\frac{d_m}{2}+2,\dots,n$的度数为$1$，令点$\frac{d_m}{2}+1,\frac{d_m}{2}+2,\dots,\frac{n+1}{2}-1$中每个点$j$都和点$n-j+1$相连，这样$1,\dots,\frac{n+1}{2}-1$和$\frac{n+1}{2}+1,\dots,n$度数都是$1$了，而我们希望他们度数为$d_i=\frac{n-1}{2}$，那么让前$\frac{n+1}{2}-1=\frac{n-1}{2}$个点连为团，后$\frac{n-1}{2}$个点连为团正好可以满足条件。

代码

/// by ztx
/// blog.csdn.net/hzoi_ztx
#include <bits/stdc++.h>
#define Rep(i,l,r) for(i=(l);i<=(r);i++)
#define rep(i,l,r) for(i=(l);i< (r);i++)
#define Rev(i,r,l) for(i=(r);i>=(l);i--)
#define rev(i,r,l) for(i=(r);i> (l);i--)
#define Each(i,v)  for(i=v.begin();i!=v.end();i++)
#define r(x)   read(x)
typedef long long ll ;
typedef double lf ;
int CH , NEG ;
template <typename TP>inline void read(TP& ret) {
    ret = NEG = 0 ; while (CH=getchar() , CH<'!') ;
    if (CH == '-') NEG = true , CH = getchar() ;
    while (ret = ret*10+CH-'0' , CH=getchar() , CH>'!') ;
    if (NEG) ret = -ret ;
}

#define  kN  510LL

int ans[kN][kN],cnt;

int n;

inline void out() {
    int i, j;
    printf("%d\n",cnt);
    Rep (i,1,n) {
        Rep (j,1,n) printf("%d ",ans[i][j]);
        puts("");
    }
}

inline void calc_1() {
    int mid = n/2, i, j;
    Rep (i,1,mid) {
        Rep (j,1,mid) ans[i][j] = ans[j][i] = ans[i+mid][j+mid] = ans[j+mid][i+mid] = 1;
        Rep (j,1,mid) ans[i][j+mid] = ans[j+mid][i] = 2;
    }
    Rep (i,1,n) ans[i][i] = 0;
    cnt = n*(n-1)*(n-2)/6;
    cnt -= mid*(mid-1)*n/2;
    out();
}


inline void calc_2() {
    int mid = (n-1)/2, i, j;
    Rep (i,1,n) Rep (j,1,n) ans[i][j] = 1;
    Rep (j,1,(mid/2))
        Rep (i,1,n)
            ans[i][(i+j-1)%n+1] = ans[(i+j-1)%n+1][i] = 2;
    Rep (i,1,n) ans[i][i] = 0;
    cnt = n*(n-1)*(n-2)/6;
    cnt -= mid*mid*n/2;
    out();
}

inline void calc_3() {
    int mid = (n-1)/2, i, j, k, p;
    Rep (i,1,n) Rep (j,1,n) ans[i][j] = 1;
    k = (mid+1)/2;
    p = (n+1)/2;
    Rep (i,1,k) ans[i][p] = ans[p][i] = ans[n-i+1][p] = ans[p][n-i+1] = 2;
    Rep (i,k+1,p-1) ans[i][n+1-i] = ans[n+1-i][i] = 2;
    Rep (i,1,p-1) Rep (j,1,p-1) ans[i][j] = ans[j][i] = ans[n+1-i][n+1-j] = ans[n+1-j][n+1-i] = 2;
    Rep (i,1,n) ans[i][i] = 0;
    cnt = n*(n-1)*(n-2)/6;
    cnt -= (mid*mid*(n-1)+(mid+1)*(mid-1))/2;
    out();
}

inline void work() {
    r(n);
    if (n&1) {
        if (((n-1)/2)&1) calc_3();
        else calc_2();
    } else calc_1();
}

int main() {
    int T;
    r(T);
    while (T-->0) work();
    END: getchar(), getchar();
    return 0;
}