题目描述
$LZK$发明一个矩阵游戏,大家一起来玩玩吧,有一个$N$行$M$列的矩阵。第一行的数字是$1,2,...,M$,第二行的数字是$M+1,M+2,...,2 imes M$,以此类推,第$N$行的数字是$(N-1) imes M+1,(N-1) imes M+2,...,N imes M$。
例如$N=3,M=4$的矩阵是这样的:
1 2 3 4
5 6 7 8
9 10 11 12
对于身为智慧之神的$LZK$来说,这个矩阵过于无趣。于是他决定改造这个矩阵,改造会进行$K$次,每次改造会将矩阵的某一行或某一列乘上一个数字,你的任务是计算最终这个矩阵内所有数字的和,输出答案对${10}^9+7$取模。
输入格式
第一行包含三个正整$N$、$M$、$K$,表示矩阵的大小与改造次数。接下来的行,每行会是如下两种形式之一:
$R X Y$,表示将矩阵的第$X$行变为原来的$Y$倍。
$S X Y$,表示将矩阵的第$X$列变为原来的$Y$倍。
输出格式
输出一行一个整数,表示最终矩阵内所有元素的和对${10}^9+7$取模的结果。
样例
样例输入1:
3 4 4
R 2 4
S 4 1
R 3 2
R 2 0
样例输出1:
94
样例输入2:
2 4 4
S 2 0
S 2 3
R 1 5
S 1 3
样例输出2:
80
数据范围与提示
$40\%$的数据满足:$1leqslant N,Mleqslant 1,000$;
$80\%$的数据满足:$1leqslant N,Mleqslant 1,000,000,1leqslant Kleqslant 1,000$;
$100\%$的数据满足:$1leqslant N,Mleqslant 1,000,000,1leqslant Kleqslant 100,000$。
题解
$40\%$算法:
暴力求出每一个点的初始值,暴力更改,暴力统计答案。
时间复杂度:$Theta(N imes K)$。
期望得分:$40$分。
$100\%$算法:
显然,我们如果$Theta(N imes M)$求出每一个点的初始值是不能接受的,所以我们考虑用式子推出,点$(i,j)$的初始值就是$(i-1) imes M+j$。
考虑乘法交换律,先乘和后乘一样,所以我们可以预处理出来$prod R$和$prod S$,设其分别为$h[i]$和$l[i]$。
再来推式子,每一个点对答案的贡献就是:$((i-1) imes M+j) imes h[i] imes l[i]$。
把式子拆开:$(i-1) imes M imes h[i] imes l[i]+j imes h[i] imes l[i]$。
我们可以维护$sumh=sum limits_{i=1}^N h[i]$和$sum=sum limits_{i=1}^N (i-1) imes M imes h[i]$。
那么答案即为$sum limits_{i=1}^M (sum+i imes sumh) imes l[i]$。
具体实现看代码叭~
时间复杂度:$Theta(N+M)$。
期望得分:$100$分。
代码时刻
$40\%$算法:
#include<bits/stdc++.h>
using namespace std;
int n,m,k;
long long Map[5000][5000];
long long ans;
int main()
{
scanf("%d%d%d",&n,&m,&k);
int cnt=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
Map[i][j]=++cnt;
while(k--)
{
char ch[5];
long long x,y;
scanf("%s%lld%lld",ch+1,&x,&y);
if(ch[1]=='R')
for(int i=1;i<=m;i++)
Map[x][i]=(Map[x][i]*y)%1000000007;
else
for(int i=1;i<=n;i++)
Map[i][x]=(Map[i][x]*y)%1000000007;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
ans=(ans+Map[i][j])%1000000007;
printf("%lld",ans);
return 0;
}
$100\%$算法:
#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int h[1000001],l[1000001];
int sum,sumh,ans;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=1000000;i++)
h[i]=l[i]=1;
while(k--)
{
char ch[3];
int x,y;
scanf("%s%d%d",ch+1,&x,&y);
if(ch[1]=='R')h[x]=1LL*h[x]*y%1000000007;
else l[x]=1LL*l[x]*y%1000000007;
}
for(int i=1;i<=n;i++)
{
sum=(sum+(1LL*(i-1)*m+1)%1000000007*h[i]%1000000007)%1000000007;
sumh=(sumh+h[i])%1000000007;
}
for(int i=1;i<=m;i++)
{
ans=(ans+1LL*sum*l[i]%1000000007)%1000000007;
sum=(sum+sumh)%1000000007;
}
printf("%lld",ans);
return 0;
}
rp++