牛客练习赛65 多重序列 数学 快速乘 数学

链接：https://ac.nowcoder.com/acm/contest/5961/B
来源：牛客网

给出n个组，第i组有m个数，分别为$a_{i,j}$一组数的权值表示为该组数所有数的乘积，找出权值最大的组，输出权值对mod取模后的值

对于每组数据给出一个k，保证$a_{i,j}$是k的非负整数次幂
输入描述:

第一行4个数n,m,k,mod，意义见题目描述
接下来n行，每行m个数，第i行第j个数表示$a_{i,j}$

输出描述:

一个数，表示最大的权值对mod取模的结果

示例1
输入
复制

3 3 2 100
2 8 4
16 4 1
8 1 32

输出
复制

56

说明

三组权值分别为64,64,256，最大值为256

备注:
$1\le n,m\le 2000,1\le k\le 100,1\le ai,j,mod\le 10121\le n,m\le 2000,$
$1\le k\le 100,1\le a_{i,j},mod\le 10^{12}1\le n,m\le 2000,1\le k\le 100,1\le ai,j,mod\le 1012$


题目中有一句话 : 对于每组数据给出一个k，保证$a_{i,j}$是k的非负整数次幂
即,给出的所有$A_{i,j}=K^b$
即 :

• 第$i$行的数为$A_{i,1},A_{i,2},A_{i,3},...A_{i,m}=K^{b1},K^{b2},K^{b3}....K^{bm}$
• 只需要比较$b$的和即可比较所有序列的乘积大小
• 注意用快速乘(龟速乘)求答案

代码如下

int mod;
read(n, m, K, mod);
if(K == 1) {                       // K为 1 的时候特判
	printf("%lld\n", 1ll%mod);
	return 0;
}
int maxid = 1, maxsum = 0;
for(int i=1; i<=n; i++) 
	for(int j=1; j<=m; j++) {
		read(mtx[i][j]);
		int p = mtx[i][j];
		int& sum = mtx[i][0];

		while(p % K == 0)          //题目给定num=K^b 
			sum ++, p /= K;        //不断累除统计次数
		if(sum > maxsum) {
			maxsum = sum;
			maxid = i;
		}
	}

int ans = 1;
for(int i=1; i<=m; i++) {
	ans = ksc(ans, mtx[maxid][i], mod);
}
cout << ans << endl;

完整代码

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>

#define MAXN ((int)1e5+7)
#define ll long long 
#define int long long
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)

using namespace std;

#define show(x...) \
	do { \
		cout << "\033[31;1m " << #x << " -> "; \
		err(x); \
	} while (0)

void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }

namespace FastIO {

	char print_f[105];
	void read() { }
	void print() { putchar('\n'); }

	template <typename T, typename... T2>
		inline void read(T &x, T2 &... oth) {
			x = 0;
			char ch = getchar();
			ll f = 1;
			while (!isdigit(ch)) {
				if (ch == '-') f *= -1; 
				ch = getchar();
			}
			while (isdigit(ch)) {
				x = x * 10 + ch - 48;
				ch = getchar();
			}
			x *= f;
			read(oth...);
		}
	template <typename T, typename... T2>
		inline void print(T x, T2... oth) {
			ll p3=-1;
			if(x<0) putchar('-'), x=-x;
			do{
				print_f[++p3] = x%10 + 48;
			} while(x/=10);
			while(p3>=0) putchar(print_f[p3--]);
			putchar(' ');
			print(oth...);
		}
} // namespace FastIO
using FastIO::print;
using FastIO::read;

int n, m, Q, K;
int mtx[2048][2048];

inline ll ksc(ll x,ll y,ll p){//计算x乘y的积
	ll res=0;//加法初始化
	while(y){
		if(y&1)res=(res+x)%p;//模仿二进制
		x=(x<<1)%p; y>>=1;//将x不断乘2达到二进制
	}return res;
}

signed main() {
#ifdef debug
	freopen("test", "r", stdin);
	// freopen("out_main", "w", stdout);
	clock_t stime = clock();
#endif
	int mod;
	read(n, m, K, mod);
	if(K == 1) {                       // K为 1 的时候特判
		printf("%lld\n", 1ll%mod);
		return 0;
	}
	int maxid = 1, maxsum = 0;
	for(int i=1; i<=n; i++) 
		for(int j=1; j<=m; j++) {
			read(mtx[i][j]);
			int p = mtx[i][j];
			int& sum = mtx[i][0];

			while(p % K == 0)          //题目给定num=K^b 
				sum ++, p /= K;        //不断累除统计次数
			if(sum > maxsum) {
				maxsum = sum;
				maxid = i;
			}
		}
	
	int ans = 1;
	for(int i=1; i<=m; i++) {
		ans = ksc(ans, mtx[maxid][i], mod);
	}
	cout << ans << endl;




#ifdef debug
	clock_t etime = clock();
	printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endif 
	return 0;
}