【矩阵乘法】[NOI2013]向量内积

题目描述

两个 $d$ 维向量 $A=[a_1,a_2,\dots,a_d]$ 与 $B=[b_1,b_2,\dots,b_d]$ 的内积为其相对应维度的权值的乘积和，即：

$$\begin{equation} \langle A, B\rangle = \sum_{i = 1}^{d} a_i b_i = a_1 b_1 + a_2 b_2 + \dots + a_d b_d \end{equation}$$

现在有 $n$ 个 $d$ 维向量 $x_1, x_2, \dots, x_n$，小喵喵想知道是否存在两个向量的内积为 $k$ 的倍数。请帮助她解决这个问题。

输入格式

第一行包含 $3$ 个正整数 $n,d,k$，分别表示向量的个数，维数以及待检测的倍数。

接下来 $n$ 行每行有 $d$ 个非负整数，其中第 $i$ 行的第 $j$ 个整数表示向量 $x_i$ 的第 $j$ 维权值 $x_{i,j}$。

输出格式

包含两个整数，用空格隔开。

如果存在两个向量 $x_p,x_q$ 的内积为 $k$ 的整数倍，则输出两个向量的编号 $p$ 与 $q$ （要求 $p \lt q$）。如果存在多组这样的向量组合，输出其中任意一组即可。

若不存在这样的向量组合，则输出两个 $-1$。

样例一

input

3 5 2
1 0 1 0 1
1 1 0 1 0
0 1 0 1 1

output

2 3

explanation

$\langle x_1, x_2 \rangle = 1$，$\langle x_1, x_3 \rangle = 1$，$\langle x_2, x_3 \rangle = 2$。

限制与约定

测试点编号	$n$	$d$	$k$	$x_{i, j}$
1	$2$	$20$	$2$	$\leq 10$
2	$5$	$20$	$2$	$\leq 10$
3	$10$	$20$	$3$	$\leq 10$
4	$20$	$20$	$2$	$\leq 100$
5	$50$	$20$	$3$	$\leq 100$
6	$50$	$50$	$2$	$\leq 1000$
7	$50$	$50$	$3$	$\leq 3000000$
8	$80$	$80$	$2$	$\leq 3000000$
9	$100$	$100$	$3$	$\leq 3000000$
10	$500$	$100$	$3$	$\leq 3000000$
11	$1000$	$100$	$2$	$\leq 3000000$
12	$1000$	$100$	$3$	$\leq 3000000$
13	$10000$	$100$	$2$	$\lt 10$
14	$10000$	$100$	$3$	$\lt 10$
15	$15000$	$100$	$2$	$\lt 10$
16	$18000$	$100$	$2$	$\lt 10$
17	$20000$	$100$	$2$	$\lt 10$
18	$50000$	$30$	$3$	$\lt 10$
19	$80000$	$30$	$3$	$\lt 10$
20	$100000$	$30$	$3$	$\lt 10$

时间限制：$5\texttt{s}$

空间限制：$256\texttt{MB}$

分析

注意:以下运算都在模k意义下

$k=2$

我们把这些向量看成一个$n\times d$的矩阵$A$

$$A=\begin{bmatrix} x_{1,1}&x_{1,2}&x_{1,3}&\cdots &x_{1,d} \\ x_{2,1}&x_{2,2}&x_{2,3}&\cdots& x_{2,d}\\ \vdots&\vdots&\vdots&\ddots&\vdots\\ x_{n,1}&x_{n,2}&x_{n,3}&\cdots& x_{n,d} \end{bmatrix}$$
$A$的转置矩阵$A^T= \begin{bmatrix} x_{1,1}&x_{2,1}&x_{3,1}&\cdots&x_{n,1}\\ x_{1,2}&x_{2,2}&x_{3,2}&\cdots&x_{n,2}\\ \vdots&\vdots&\vdots&\ddots&\vdots\\ x_{1,n}&x_{2,n}&x_{3,n}&\cdots&x_{nn}\\ \end{bmatrix}$

令$B=A \times A^T$，那么$b_{i,j}$就表示$x_i\cdot x_j$，如果$B$矩阵除了主对角线全部是1的话，显然无解。
但是求出$B$矩阵的时间显然是$O(n^2d)$，显然是无法通过本题的。
我们令$B'$为$b_{i,i}=x_i \cdot x_i$，其余元素为1的矩阵。
如果$B$=$B'$的话，就无解。
我们再随机找出一个列向量$C$ (其实就是全是1，因为0没用。。。)

$$B\times C=B' \times C\\ A\times A^T \times C=B' \times C\\ A \times(A^T \times C)=B' \times C$$

$D=A^T \times C$我们可以用$O(nd)$的时间求出来，$A \times D$也可以用$O(nd)$的时间求出来，$B' \times C$可以$O(n)$求出。
然后比较$A \times(A^T \times C)\ \ B' \times C$这两个列向量的每一个元素是否相同，如果第i行的元素不相同，说明$x_i$肯定是答案只要，枚举找出$x_j$即可。
尽管有出错的概率，但是这个算法还是比较可靠的。

$k=3$

我们发现

$$1 \times 1 \equiv 2\times 2 \equiv 1\pmod 3$$
所以，我们令 $B=(A*A^T)^2$其余矩阵定义不变，那么展开后会发现
$$A'= \begin{bmatrix} x_{1,1} \times x_{1,1}&x_{1,1}\times x_{1,2}&x_{1,1}\times x_{1,3}&\cdots &x_{1,1}\times x_{1,d} &x_{1,2} \times x_{1,1}&x_{1,2} \times x_{1,2}&\cdots& x_{1,d}\times x_{1,d}\\ x_{2,1} \times x_{2,1}&x_{2,1}\times x_{2,2}&x_{2,1}\times x_{2,3}&\cdots &x_{2,1}\times x_{2,d} &x_{2,2} \times x_{2,1}&x_{2,2} \times x_{2,2}&\cdots& x_{2,d}\times x_{2,d}\\ \vdots&\vdots&\vdots&\ddots&\vdots&\vdots&\vdots&\ddots&\vdots\\ x_{n,1} \times x_{n,1}&x_{n,1}\times x_{n,2}&x_{n,1}\times x_{n,3}&\cdots &x_{n,1}\times x_{n,d} &x_{n,2} \times x_{n,1}&x_{n,2} \times x_{n,2}&\cdots& x_{n,d}\times x_{n,d} \end{bmatrix} \\ B=A' \times A'^T$$
然后参考 $k=2$的算法即可，由于 $k=3$时矩阵太大，内存开不下，但是算的时候按照规则算就行了。

代码

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 100000
#define MAXM 100
int n,d,k,a[MAXN+10][MAXM+10],s[MAXN+10],c[MAXM*MAXM+10],g[MAXN+10];
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
int Get_id(int i,int j){
    return (i-1)*d+j;
}
void read(){
    Read(n),Read(d),Read(k);
    int i,j;
    for(i=1;i<=n;i++)
        for(j=1;j<=d;j++){
            Read(a[i][j]);
            a[i][j]%=k;
            s[i]+=a[i][j]*a[i][j];
        }
    for(i=1;i<=n;i++)
        s[i]%=k;
}
void solve2(){
    int i,j,k,sum;
    for(i=1;i<=d;i++)
        for(j=1;j<=n;j++)
            c[i]^=a[j][i];
    for(i=1;i<=n;i++)
        for(j=1;j<=d;j++)
            g[i]^=a[i][j]&c[j];
    for(i=1;i<=n;i++)
        if(g[i]!=(s[i]^((n-1)&1)))
            break;
    if(i>n){
        puts("-1 -1");
        return;
    }
    for(j=1;j<=n;j++)
        if(i!=j){
            sum=0;
            for(k=1;k<=d;k++)
                sum^=a[i][k]&a[j][k];
            if(!sum)
                break;
        }
    if(i>j)
        swap(i,j);
    printf("%d %d\n",i,j);
}
void solve3(){
    int i,j,k,sum;
    for(i=1;i<=d;i++)
        for(j=1;j<=d;j++)
            for(k=1;k<=n;k++)
                c[Get_id(i,j)]+=a[k][i]*a[k][j];
    for(i=d*d;i;i--)
        c[i]%=3;
    for(i=1;i<=n;i++)
        for(j=1;j<=d;j++)
            for(k=1;k<=d;k++)
                g[i]+=a[i][j]*a[i][k]*c[Get_id(j,k)];
    for(i=1;i<=n;i++)
        if(g[i]%3!=((n-1)+s[i]*s[i])%3)
            break;
    if(i>n){
        puts("-1 -1");
        return;
    }
    for(j=1;j<=n;j++)
        if(j!=i){
            sum=0;
            for(k=1;k<=d;k++)
                sum=(sum+a[i][k]*a[j][k])%3;
            if(!sum)
                break;
        }
    if(i>j)
        swap(i,j);
    printf("%d %d\n",i,j);
}
int main()
{
    read();
    if(k==2)
        solve2();
    else
        solve3();
}