【补题日记】[2022牛客暑期多校2]I-let fat tension

最新推荐文章于 2024-05-30 20:15:33 发布

cls1277

最新推荐文章于 2024-05-30 20:15:33 发布

阅读量155

点赞数 1

分类专栏：补题日记文章标签：算法

本文链接：https://blog.csdn.net/cls1277/article/details/126006101

版权

补题日记专栏收录该内容

37 篇文章 0 订阅

订阅专栏

Pro

https://ac.nowcoder.com/acm/contest/33187/I

Sol

其实是个简单的计算题，不过比赛的时候榜被带歪了所以没人看这个题。

接下来化简式子（增加的下标o的含义为Y向量的每个分量的下标）： $Y_{i}^{new}=\sum_{j=1}^{n}\sum_{p=1}^{k}\frac{X_{ip}·X_{jp}}{|X_i|·|X_j|}Y_{i}$
$Y_{i,o}^{new}=\sum_{j=1}^{n}\sum_{p=1}^{k}\frac{X_{ip}·X_{jp}}{|X_i|·|X_j|}Y_{i,o}$
$Y_{i,o}^{new}=\sum_{j=1}^{n}\sum_{p=1}^{k}\frac{X_{ip}}{|X_i|}·\frac{X_{jp}}{|X_j|}Y_{i,o}$
$Y_{i,o}^{new}=\sum_{p=1}^{k}(\frac{X_{ip}}{|X_i|}·\sum_{j=1}^{n}\frac{X_{jp}}{|X_j|}Y_{i,o})$

令 $Z_{p,o}=\sum_{j=1}^{n}\frac{X_{jp}}{|X_j|}Y_{i,o}$ ，则
$Y_{i,o}^{new}=\sum_{p=1}^{k}(\frac{X_{ip}}{|X_i|}·Z_{p,o})$

可以输入完后直接预处理出 $\frac{X_{i,p}}{|X_i|}$ 和 $\frac{X_{j,p}}{|X_j|}$ ，可以看出来计算 $Z_{p,o}$ 和 $Y_{i,o}^{new}$ 的时间复杂度均为 $O (nk d)$

Code

//By cls1277
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define Fo(i,a,b) for(LL i=(a); i<=(b); i++)
#define Ro(i,b,a) for(LL i=(b); i>=(a); i--)
#define Eo(i,x,_) for(LL i=head[x]; i; i=_[i].next)
#define Ms(a,b) memset((a),(b),sizeof(a))
#define endl '\n'

const LL maxn = 1e4+5;
LL n, k, d;
double x[maxn][55], y[maxn][55], z[maxn][55], ans[maxn][55];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    #ifdef DEBUG
    freopen("data.txt","r",stdin);
    #endif
    cin>>n>>k>>d;
    Fo(i,1,n) {
        double len = 0;
        Fo(j,1,k) {
            cin>>x[i][j];
            len += x[i][j]*x[i][j];
        }
        len = sqrt(len);
        Fo(j,1,k) x[i][j] /= len;
    }
    Fo(i,1,n)
        Fo(j,1,d)
            cin>>y[i][j];
    Fo(i,1,n)
        Fo(j,1,k)
            Fo(l,1,d)
                z[j][l] += x[i][j]*y[i][l];
    Fo(i,1,n)
        Fo(j,1,d)
            Fo(l,1,k)
                ans[i][j] += x[i][l]*z[l][j];
    Fo(i,1,n) {
        Fo(j,1,d)
            cout<<fixed<<setprecision(8)<<ans[i][j]<<' ';
        cout<<endl;
    }
    return 0;
}