2023 csp 矩阵运算

题目背景

Softmax(�×���)×� 是 Transformer 中注意力模块的核心算式，其中 �、� 和 � 均是 � 行 � 列的矩阵，�� 表示矩阵 � 的转置，× 表示矩阵乘法。

问题描述

为了方便计算，顿顿同学将 Softmax 简化为了点乘一个大小为 � 的一维向量 �：
(�⋅(�×��))×�
点乘即对应位相乘，记 �(�) 为向量 � 的第 � 个元素，即将 (�×��) 第 � 行中的每个元素都与 �(�) 相乘。

现给出矩阵 �、� 和 � 和向量 �，试计算顿顿按简化的算式计算的结果。

输入格式

从标准输入读入数据。

输入的第一行包含空格分隔的两个正整数 � 和 �，表示矩阵的大小。

接下来依次输入矩阵 �、� 和 �。每个矩阵输入 � 行，每行包含空格分隔的 � 个整数，其中第 � 行的第 � 个数对应矩阵的第 � 行、第 � 列。

最后一行输入 � 个整数，表示向量 �。

输出格式

输出到标准输出中。

输出共 � 行，每行包含空格分隔的 � 个整数，表示计算的结果。

样例输入

3 2
1 2
3 4
5 6
10 10
-20 -20
30 30
6 5
4 3
2 1
4 0 -5

Data

样例输出

480 240
0 0
-2200 -1100

Data

子任务

70 的测试数据满足：�≤100 且 �≤10；输入矩阵、向量中的元素均为整数，且绝对值均不超过 30。

全部的测试数据满足：�≤104 且 �≤20；输入矩阵、向量中的元素均为整数，且绝对值均不超过 1000。

提示

请谨慎评估矩阵乘法运算后的数值范围，并使用适当数据类型存储矩阵中的整数。

f

做题方法

1.按顺序计算--70分

2.先合并K*V --100分

原因：d的复杂度远小于n

在这里仅仅展示最优解答：

运用到的知识点：

1.矩阵转置

2.矩阵乘法

3.给定的点乘

4.各种浓缩简化书写

5.map划定数组，单独map划定ans数组

#include<bits/stdc++.h>
using namespace std;
int n,d;
//long long Q[1000][100],K[1000][100],V[1000][100],W[1000][100],ans[1000][100];
map<int ,map<int ,int> > Q,K,V,W;
int main()
{
	cin>>n>>d;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=d;j++)
	cin>>Q[i][j];
	
	//K的转置 
	for(int i=1;i<=n;i++)
	for(int j=1;j<=d;j++)
	cin>>K[j][i];
	
	for(int i=1;i<=n;i++)
	for(int j=1;j<=d;j++)
	cin>>V[i][j];
	
	for(int i=1;i<=n;i++)
	cin>>W[1][i];
	map<long long, map<long long, long long> > ans;
	//K*V 23  32   --22
	for(int i=1;i<=d;i++)
	{
		for(int j=1;j<=d;j++)
		{
			int x=0;
			{
				for(int k=1;k<=n;k++)
				{
				
					x+=K[i][k]*V[k][j];
					ans[i][j]=x;
		    	}
				
			}
			
		}
	}
    //	W*Q*F  3 2  22   32
    for(int i=1;i<=n;i++)
    {
    	for(int j=1;j<=d;j++)
    	{
    		long long x=0;
    		for(int k=1;k<=d;k++)
    		
    			x+=Q[i][k]*ans[k][j];
    			x=x*W[1][i];
    			cout<<x<<" ";
			
		}
		cout<<endl;
	}

	
	
	
	
}

晴天小课堂开课啦

定义二维数组，双map嵌套

例如：

map<int ,map<int ,int> > Q,K,V,W;

 map<long long, map<long long, long long> > ans;

上一堂课定义字符串数组，单map

map<string,int>status_map