結果
| 問題 |
No.3123 Inversion
|
| ユーザー |
 GOTKAKO
|
| 提出日時 | 2025-04-20 11:34:30 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
RE
|
| 実行時間 | - |
| コード長 | 6,188 bytes |
| コンパイル時間 | 2,771 ms |
| コンパイル使用メモリ | 210,312 KB |
| 実行使用メモリ | 198,620 KB |
| 最終ジャッジ日時 | 2025-04-20 11:34:55 |
| 合計ジャッジ時間 | 24,127 ms |
|
ジャッジサーバーID （参考情報） |
judge4 / judge1 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 8 WA * 3 RE * 10 |
ソースコード
#include <bits/stdc++.h>
using namespace std;
long long mod = 998244353;
//入力が必ず-mod<a<modの時.
struct mint{
long long v = 0;
mint(){} mint(int a){v = a<0?a+mod:a;} mint(long long a){v = a<0?a+mod:a;}
mint(unsigned long long a){v = a;}
long long val(){return v;}
mint &operator=(const mint &b) = default;
mint operator-() const {return mint(0)-(*this);}
mint operator+(const mint b){return mint(v)+=b;}
mint operator-(const mint b){return mint(v)-=b;}
mint operator*(const mint b){return mint(v)*=b;}
mint operator/(const mint b){return mint(v)/=b;}
mint operator+=(const mint b){
v += b.v; if(v >= mod) v -= mod;
return *this;
}
mint operator-=(const mint b){
v -= b.v; if(v < 0) v += mod;
return *this;
}
mint operator*=(const mint b){v = v*b.v%mod; return *this;}
mint operator/=(mint b){
if(b == 0) assert(false);
int left = mod-2;
while(left){if(left&1) *this *= b; b *= b; left >>= 1;}
return *this;
}
mint operator++(){*this += 1; return *this;}
mint operator--(){*this -= 1; return *this;}
mint operator++(int){*this += 1; return *this;}
mint operator--(int){*this -= 1; return *this;}
bool operator==(const mint b){return v == b.v;}
bool operator!=(const mint b){return v != b.v;}
bool operator>(const mint b){return v > b.v;}
bool operator>=(const mint b){return v >= b.v;}
bool operator<(const mint b){return v < b.v;}
bool operator<=(const mint b){return v <= b.v;}
mint pow(long long n){
mint ret = 1,p = v;
if(n < 0) p = p.inv(),n = -n;
while(n){
if(n&1) ret *= p;
p *= p; n >>= 1;
}
return ret;
}
mint inv(){return mint(1)/v;}
};
class UnionFind{
private:
vector<int> par,siz;
public:
UnionFind(int N){
par.resize(N,-1);
siz.resize(N,1);
}
int root(int x){ //連結成分の代表頂点を返す.
if(par.at(x) == -1) return x;
else return par.at(x) = root(par.at(x));
}
bool unite(int u, int v){ //u,vを連結する 連結してた->false,した->trueを返す.
u = root(u),v = root(v);
if(u == v) return false;
if(siz.at(u) < siz.at(v)) swap(u,v); //Union by size.
par.at(v) = u;
siz.at(u) += siz.at(v);
return true;
}
bool issame(int u, int v){ //同じ連結成分ならtrue.
if(root(u) == root(v)) return true;
else return false;
}
int size(int pos){return siz.at(root(pos));} //posの連結成分の大きさを返す.
};
long long inv(vector<int> P){
int N = P.size();
long long ret = 0;
for(int i=0; i<N; i++) for(int k=i+1; k<N; k++) ret += P.at(i)>P.at(k);
return ret;
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
int T; cin >> T;
cin >> mod;
if(mod == 1){
while(T--) cout << "0\n";
return 0;
}
int n = 5000000;
int Limit = n; //Limit 必要なサイズ fac->x! facinv->1/x! inv->1/x.
vector<mint> fac(Limit+1,1); for(int i=1; i<=Limit; i++) fac.at(i) = fac.at(i-1)*i;
vector<mint> facinv(Limit+1); facinv.at(min((int)mod-1,Limit)) = fac.at(min((int)mod-1,Limit)).inv();
for(int i=min((int)mod-2,Limit-1); i>=0; i--) facinv.at(i) = facinv.at(i+1)*(i+1);
//vector<mint> inv(Limit+1); for(int i=1; i<=Limit; i++) inv.at(i) = fac.at(i-1)*facinv.at(i); //必要なら解放!.
auto nCr = [&](int n, int r) -> mint {
if(n < r || r < 0 || n < 0) return 0;
return fac.at(n)*facinv.at(r)*facinv.at(n-r);
};
//OEIS最高！→え～何が違うの
//連結成分サイズ2 逆順列=revとなる数 A037224.
//連結成分サイズ2 それ以外 A000898.
//連結成分サイズ8 A000899*8. n!-2Xn-Y[n/2]+2*Two[n/2]
vector<mint> Two(n/2+1),Eight(n+1),X(n+1),Y(n/2+1);
Two.at(0) = 1; Two.at(1) = 2%mod; X.at(0) = 1; X.at(1) = 1; X.at(2) = 2%mod;
for(int i=2; i<=n/2; i++) Two.at(i) = (Two.at(i-1)+Two.at(i-2)*(i-1))*2;
for(int i=3; i<=n; i++) X.at(i) = X.at(i-1)+X.at(i-2)*(i-1);
mint p2 = 1;
for(int i=0; i<=n/2; i++) Y.at(i) = fac.at(i)*p2,p2 += p2;
for(int i=1; i<=n; i++) Eight.at(i) = fac.at(i)-X.at(i)*2-Y.at(i/2)+Two.at(i/2)*2;
while(T--){
int N; cin >> N;
if(N == 1){cout << 1%mod << "\n"; continue;}
mint two = 0,answer = 0,eight = 0,four = 0;
if(N%4 == 0 || N%4 == 1) two = fac.at(N/4*2-1)*2*facinv.at(N/4-1);
two += Two.at(N/2); eight = Eight.at(N);
four = fac.at(N)-two-eight;
answer = two*2+four*4+eight*8;
answer.v %= mod;
cout << answer.v << "\n"; continue;
long long f = 1;
for(int i=1; i<=N; i++) f *= i;
UnionFind Z(f);
map<vector<int>,long long> name;
vector<int> P(N);
iota(P.begin(),P.end(),0);
do{
name[P] = name.size();
}while(next_permutation(P.begin(),P.end()));
iota(P.begin(),P.end(),0);
vector<int> Q(N);
int idx = -1;
do{
idx++;
reverse(P.begin(),P.end());
Z.unite(idx,name[P]);
reverse(P.begin(),P.end());
for(int i=0; i<N; i++) Q.at(P.at(i)) = i;
Z.unite(idx,name[Q]);
}while(next_permutation(P.begin(),P.end()));
map<long long,long long> ap;
for(auto [K,v] : name){
ap[Z.size(v)]++;
if(Z.size(v) == 2) continue;
//continue;
vector<int> S(N);
for(int i=0; i<N; i++) S[i] = K.at(i)+K.at(N-1-i);
for(auto p : K) cout << p+1 << " ";
cout << " ";
for(auto p : S) cout << p+1 << " ";
cout << " ";
//cout << " " << inv(K);
cout << " " << Z.size(v) << endl;
vector<bool> B(N);
}
//cout << N << endl;
long long ans = 0;
for(auto [k,v] : ap) cout << k << "->" << v << endl,ans += k*v;
cout << ans << endl;
cout << endl;
}
}