[NOIP模拟测试]:矩阵游戏(数学)

题目描述

LZK发明一个矩阵游戏,大家一起来玩玩吧,有一个$N$行$M$列的矩阵。第一行的数字是$1,2,...,M$,第二行的数字是$M+1,M+2,...,2\times M$,以此类推,第$N$行的数字是$(N-1)\times M+1,(N-1)\times M+2,...,N\times 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%$的数据满足:$1\leqslant N,M\leqslant 1,000$;
$80%$的数据满足:$1\leqslant N,M\leqslant 1,000,000,1\leqslant K\leqslant 1,000$;
$100%$的数据满足:$1\leqslant N,M\leqslant 1,000,000,1\leqslant K\leqslant 100,000$。


题解

$40%$算法:

暴力求出每一个点的初始值,暴力更改,暴力统计答案。

时间复杂度:$\Theta(N\times K)$。

期望得分:$40$分。

$100%$算法:

显然,我们如果$\Theta(N\times M)$求出每一个点的初始值是不能接受的,所以我们考虑用式子推出,点$(i,j)$的初始值就是$(i-1)\times M+j$。

考虑乘法交换律,先乘和后乘一样,所以我们可以预处理出来$\prod R$和$\prod S$,设其分别为$h[i]$和$l[i]$。

再来推式子,每一个点对答案的贡献就是:$((i-1)\times M+j)\times h[i]\times l[i]$。

把式子拆开:$(i-1)\times M\times h[i]\times l[i]+j\times h[i]\times l[i]$。

我们可以维护$sumh=\sum \limits_{i=1}^N h[i]$和$sum=\sum \limits_{i=1}^N (i-1)\times M\times h[i]$。

那么答案即为$\sum \limits_{i=1}^M (sum+i\times sumh)\times 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++

猜你喜欢

转载自www.cnblogs.com/wzc521/p/11309291.html