Codeforces Round #574 (Div. 2) :D2. Submarine in the Rybinsk Sea (hard edition) (枚举 + 思维)

题目链接:https://codeforc.es/contest/1195/problem/D2

题目大意:有n个数字,定义 f ( a , b ) f(a,b) f(a,b):若 a a a b b b一样长,则按数位从左到右 a a a 先放, b b b后放,如此交替放下去。若 a a a b b b 不一样长,则从左到右先放长的数位,到剩余数位和较短的那个数一样长时,再和上面一样依次放。然后就会得出一个长度为 l e n ( a ) + l e n ( b ) len(a) + len(b) len(a)+len(b)的数字。
例子:
Codeforces Round #574 (Div. 2) :D2. Submarine in the Rybinsk Sea (hard edition) (枚举 + 思维)_第1张图片
题目让你求 ∑ i = 1 n ∑ j = 1 n f ( a [ i ] , a [ j ] ) \sum_{i=1}^{n}\sum_{j=1}^{n}f(a[i],a[j]) i=1nj=1nf(a[i],a[j])

题解:若所有数字都一样长,每个数字要么放左边要么放右边,直接按位计算每一个数字的每一位的贡献和,计算所有数字的贡献并加起来。
若数字不一样长,一个数字和不同长度的数字拼起来得到的贡献不同,但和某一长度的数字拼起来的贡献都相同,由于数字最多10位,枚举所有的数字,对每一个数字,枚举数字长度,计算该数字和所有这种长度数字拼接起来时该数字的贡献,所有数字的贡献加起来得到答案。

#include
using namespace std;
const int maxn = 1e5 + 10;
const int mod = 998244353;
typedef long long ll;
int num[maxn];
int n,a[maxn];
int getlen(int x) {
	int l = 0;
	while(x) {
		l++;
		x /= 10;
	}
	return l;
}
ll solve(int x,int len,int sz) {
	ll tmp = 1;
	int cur = 0;
	ll ans = 0; 
	while(x) {
		cur++;
		int p = x % 10;
		x /= 10;
		if(cur <= len) {
			ans = (ans + ((p * tmp % mod + (p * tmp % mod) * 10 % mod) % mod) * sz % mod) % mod;
			tmp = tmp * 100 % mod;
		}
		else {
			ans = (ans + (((2 * p) % mod) * tmp % mod) * sz % mod) % mod;
			tmp = tmp * 10 % mod;
		}
		
	}
	return ans;
}
int main() {
	scanf("%d",&n);
	for(int i = 1; i <= n; i++) {
		scanf("%d",&a[i]);
		num[getlen(a[i])]++;
	}
	ll res = 0;
	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= 10; j++)
			if(num[j])
				res = (res + solve(a[i],j,num[j])) % mod;
	}
	printf("%lld\n",res);
	return 0;
}

你可能感兴趣的:(枚举,思维,枚举)