码迷,mamicode.com
首页 > 其他好文 > 详细

矩阵化一维

时间:2019-10-30 13:31:31      阅读:107      评论:0      收藏:0      [点我收藏+]

标签:e431   矩阵乘法   i++   nbsp   scan   算法   sed   高中   自己   

矩阵化一维

  前缀和思想:

技术图片
for(ri i=1;i<=n;i++)
    for(ri j=1;j<=m;j++)
    {
        scanf("%lld",&arr[i][j]);
        arr[i][j]=arr[i-1][j]+arr[i][j-1]-arr[i-1][j-1]+arr[i][j];
    }
View Code

应用:

  题目:

技术图片
小 F 很喜欢数学,但是到了高中以后数学总是考不好。

有一天,他在数学课上发起了呆;他想起了过去的一年。一年前,当他初识算法竞赛的 时候,觉得整个世界都焕然一新。这世界上怎么会有这么多奇妙的东西?曾经自己觉得难以 解决的问题,被一个又一个算法轻松解决。

小 F 当时暗自觉得,与自己的幼稚相比起来,还有好多要学习的呢。

一年过去了,想想都还有点恍惚。

他至今还能记得,某天晚上听着入阵曲,激动地睡不着觉,写题写到鸡鸣时分都兴奋不 已。也许,这就是热血吧。



也就是在那个时候,小 F 学会了矩阵乘法。让两个矩阵乘几次就能算出斐波那契数列的 第 10^{100}10 
100
  项,真是奇妙无比呢。

不过,小 F 现在可不想手算矩阵乘法——他觉得好麻烦。取而代之的,是一个简单的小 问题。他写写画画,画出了一个 n \times mn×m 的矩阵,每个格子里都有一个不超过 kk 的正整数。

小 F 想问问你,这个矩阵里有多少个不同的子矩形中的数字之和是 kk 的倍数? 如果把一个子矩形用它的左上角和右下角描述为 (x_1,y_1,x_2,y_2)(x 
1
?     ,y 
1
?     ,x 
2
?     ,y 
2
?     ),其中x_1 \le x_2,y_1 \le y_2x 
1
?     ≤x 
2
?     ,y 
1
?     ≤y 
2
?     ; 那么,我们认为两个子矩形是不同的,当且仅当他们以 (x_1,y_1,x_2,y_2)(x 
1
?     ,y 
1
?     ,x 
2
?     ,y 
2
?     ) 表示时不同;也就是 说,只要两个矩形以 (x_1,y_1,x_2,y_2)(x 
1
?     ,y 
1
?     ,x 
2
?     ,y 
2
?     ) 表示时相同,就认为这两个矩形是同一个矩形,你应该 在你的答案里只算一次。

输入格式

从标准输入中读入数据。

输入第一行,包含三个正整数 n,m,kn,m,k。

输入接下来 nn 行,每行包含 mm 个正整数,第 ii 行第 jj 列表示矩阵中第 ii 行第 jj 列 中所填的正整数 a_{i,j}a 
i,j
?     。

输出格式

输出到标准输出中。

输入一行一个非负整数,表示你的答案。

输入输出样例

输入 #1 复制
2 3 2 
1 2 1 
2 1 2
输出 #1 复制
6 
View Code

  代码

技术图片
#include <bits/stdc++.h>
using namespace std;
const int M = 10005;
const int N = 100005;
#define ri register int
int n,m,mod;
long long ans,arr[520][520],num[1000001],a[520];
int main(){
    scanf("%d%d%d",&n,&m,&mod);
    for(ri i=1;i<=n;i++)
    for(ri j=1;j<=m;j++)
    {
        scanf("%lld",&arr[i][j]);
        arr[i][j]=arr[i-1][j]+arr[i][j-1]-arr[i-1][j-1]+arr[i][j];
    }
    for(ri i=0;i<n;i++)
    for(ri j=i+1;j<=n;j++)
    {
          num[0]=1;
       for(ri k=1;k<=m;k++)
       {
              a[k]=(arr[j][k]-arr[i][k]+(mod<<1))%mod;
              ans+=num[a[k]];
               num[a[k]]++;
       }
       for(ri k=1;k<=m;k++)
       { 
          num[a[k]]=0;
       }
    }
    printf("%lld\n",ans);
}
View Code

 

  6

矩阵化一维

标签:e431   矩阵乘法   i++   nbsp   scan   算法   sed   高中   自己   

原文地址:https://www.cnblogs.com/Lamboofhome/p/11764230.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!