[HNOI2001]矩阵乘积

题目描述

输入输出格式

输入格式：

第1行为：x y （第1行为两个正整数：x,y分别表示输出结果所在的行和列）

第2行为：m n o p（第2行给出的正整数表明A为m×n矩阵，B为n×o矩阵，C为o×p矩阵）

第3行为：i j a（第3行以后的每一行有三个整数分别是矩阵的三元组表示法中的一个元素的值，每个矩阵之间有一个空行。表示的顺序是矩阵A、B和C）

… … … … … …

注：1≤m，n，o，p≤6000，三元数组的总个数不大于6000。数据之间用空格分开。

输出格式：

为 的第x行第y列元素的值。

输入输出样例

输入样例#1：

1 2                       
3 4 2 3
1 1 3
1 4 5
2 2 1
3 1 2

1 2 2
2 1 1
3 1 2
3 2 4

1 2 2
1 3 3
2 1 1
2 2 2

输出样例#1：

12
只需要(x,y)的值
ans(x,y)=∑B[x][k]*c[k][y]
B[i][k]=∑a[i][p]*b[p][k]
所以可以只求一部分的值
输入部分用一种水法，还好数据弱
如果前面输入(i,j)，后面输入(a,b)
(a,b)<(i,j) 则为另一矩阵

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int m,n,o,p,x,y,A[6001],B[6001];
int main()
{int a,b,c,i,j,z;
  cin>>x>>y;
  cin>>m>>n>>o>>p;
  scanf("%d%d%d",&a,&b,&z);
  while (1)
    {
      if (a==x) A[b]=z;
      scanf("%d%d%d",&i,&j,&c);
      if (i<a||(i==a&&j<=b)) break;
      a=i;b=j;z=c;
    }
  a=i;b=j;z=c;
  while (1)
    {
      B[b]+=A[a]*z;
      scanf("%d%d%d",&i,&j,&c);
      if (i<a||(i==a&&j<=b)) break;
      a=i;b=j;z=c;
    }
  memcpy(A,B,sizeof(A));
  memset(B,0,sizeof(B));
  a=i;b=j;z=c;
  while (1)
    {
      if (b==y) B[b]+=A[a]*z;
      scanf("%d%d%d",&i,&j,&c);
      if (i<a||(i==a&&j<=b)) break;
      a=i;b=j;z=c;
    }
  cout<<B[y]<<endl;
}