結果

問題 No.1193 Penguin Sequence
コンテスト
ユーザー HIR180
提出日時 2020-08-22 14:37:29
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 205 ms / 2,000 ms
コード長 2,676 bytes
コンパイル時間 3,467 ms
コンパイル使用メモリ 248,116 KB
最終ジャッジ日時 2025-01-13 08:49:17
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:102:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  102 |         scanf("%d", &n);
      |         ~~~~~^~~~~~~~~~
main.cpp:103:25: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  103 |         repn(i, n) scanf("%d", &a[i]);
      |                    ~~~~~^~~~~~~~~~~~~

ソースコード

diff # 

//Let's join Kaede Takagaki Fan Club !!
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
typedef pair<int,P> P1;
typedef pair<P,P> P2;
#define pu push
#define pb push_back
#define mp make_pair
#define eps 1e-7
#define INF 1000000000
#define fi first
#define sc second
#define rep(i,x) for(int i=0;i<x;i++)
#define repn(i,x) for(int i=1;i<=x;i++)
#define SORT(x) sort(x.begin(),x.end())
#define ERASE(x) x.erase(unique(x.begin(),x.end()),x.end())
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin())
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin())
#define all(x) x.begin(),x.end()
template<class T>
void dmp(T a){
	rep(i,a.size()) cout << a[i] << " ";
	cout << endl;
}
template<class T>
bool chmax(T&a, T b){
	if(a < b){
		a = b;
		return 1;
	}
	return 0;
}
template<class T>
bool chmin(T&a, T b){
	if(a > b){
		a = b;
		return 1;
	}
	return 0;
}
template<class T>
void g(T &a){
	cin >> a;
}
template<class T>
void o(const T &a,bool space=false){
	cout << a << (space?' ':'\n');
}
//ios::sync_with_stdio(false);
const ll mod = 998244353;
template<class T>
void add(T&a,T b){
	a+=b;
	if(a >= mod) a-=mod;
}
ll modpow(ll x,ll n){
	ll res=1;
	while(n>0){
		if(n&1) res=res*x%mod;
		x=x*x%mod;
		n>>=1;
	}
	return res;
}
ll F[200005],R[200005];
void make(){
	F[0] = 1;
	for(int i=1;i<200005;i++) F[i] = F[i-1]*i%mod;
	for(int i=0;i<200005;i++) R[i] = modpow(F[i],mod-2);
}
ll C(int a,int b){
	return F[a]*R[b]%mod*R[a-b]%mod;
}

int n, a[200005];
vector<int>vi;
struct BIT
{
	int bit[(1<<20)+5];
	int f(int x)
	{
		return x&-x;
	}
	void add(int i,int x)
	{
		for(int s=i;s<=(1<<20);s+=f(s)) bit[s]+=x;
	}
	int sum(int i)
	{
		int res=0;
		for(int s=i;s>0;s-=f(s)) res+=bit[s];
		return res;
	}
}bit;
int main(){
	scanf("%d", &n);
	repn(i, n) scanf("%d", &a[i]);
	repn(i, n) vi.pb(a[i]);
	SORT(vi); ERASE(vi);
	ll R = 0, S = 0;
	repn(i, n){
		a[i] = POSL(vi, a[i]) + 1;
		R += i-1-bit.sum(a[i]);
		S += i-1-bit.sum(a[i])+bit.sum(a[i]-1);
		bit.add(a[i], 1);
	}
	make();
	ll ans = 1;
	repn(i, n) ans = ans * C(n, i) % mod;
	ll c1 = 0;
	c1 = 1LL*n*(n+1)/2%mod;
	c1 = c1*c1%mod;
	repn(i, n) c1 -= 1LL*i*i%mod;
	c1 = (c1%mod+mod)%mod;
	c1 = c1 * ((mod+1)/2) % mod;
	c1 = c1 * modpow(n, mod-2) % mod * modpow(n, mod-2) % mod;
	ll c2 = 0;
	for(int i=1;i<=n;i++){
		c2 += 1LL*i*(i-1)%mod * modpow(n, mod-2) % mod * modpow(n-1, mod-2) % mod;
	}
	c2 %= mod;
	c1 = c1 * ans % mod;
	c2 = c2 * ans % mod;
	ll ret = S%mod*c1%mod;
	ret += R%mod*c2%mod;
	cout << (ret%mod) << endl;
}
0