codeforce1195D1Submarine in the Rybinsk Sea (easy edition)

问题链接(http://codeforces.com/problemset/problem/1195/D1)

问题描述

定义了函数 f(a₁a₂…a_p−1a_p,b₁b₂…b_q−1b_q)=b₁b₂…a_p-1b_q−1a_pb_q。
在问题D1中给的数是等长的，所以有
f(a₁a₂…a_p−1a_p,b₁b₂…b_p−1b_p)=a₁b₁a₂b₂…a_p-1b_p−1a_pb_p。
要求$$\sum _{i=1}^n\sum _{j=1}^{n}f(a_i,a_j)$$，结果对 998244353取模。

问题分析

将上述合式展开，可以知道任意的整数i属于[1,n]对应的a_i都会与任意的整数j属于[1,n]对应的a_j进行一次f运算。也就是a₁分别与a₁、a₂、……、a_n进行一次f运算（a₁在前），其它的a_i也是。所以有f(a_i,a_j)就有f(a_j,a_i)。
可以发现，由于给的数是等长的，所以f(a₁a₂…a_p−1a_p,b₁b₂…b_p−1b_p)=a₁b₁a₂b₂…a_p-1b_p−1a_pb_p与f(b₁b₂…b_p−1b_p,a₁a₂…a_p−1a_p)=b₁a₁b₂a₂…b_p-1a_p−1b_pa_p的和可以看成是其中一个结果的a_i与另一个结果的b_i交换数字，也就是f(a₁a₂…a_p−1a_p,b₁b₂…b_p−1b_p)+f(b₁b₂…b_p−1b_p,a₁a₂…a_p−1a_p)=a₁a₁a₂a₂…a_p-1a_p−1a_pa_p+b₁b₁b₂b₂…b_p-1b_p−1b_pb_p。
再简单来说就是f(a_i,a_j)+f(a_j,a_i)=f(a_i,a_i)+f(a_j,a_j)。
因为可以改写成这样相同的式子共n个，那么我们求一次$$\sum _{i=1}^{n}f(a_i,a_i)$$，然后再乘个n就可以了。

代码如下

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const int mod=998244353;

ll change(string str){//求f(ai,ai)
	ll temp=0;
	int len=str.size();
	for(int i=0;i<len;i++){
		for(int j=0;j<2;j++)
		{
		temp*=10;
		temp+=str[i]-'0'; 
		temp%=mod;
		}
	}
	return temp%mod;
}

int main(){
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	int n,i;
	ll ans=0;
	string num;
	cin>>n;
	for(i=0;i<n;i++){
		cin>>num;
		ans+=change(num);
		ans%=mod;
	}
	ans*=n;
	cout<<ans%mod<<endl;
	return 0;
}