結果

問題 No.696 square1001 and Permutation 5
ユーザー beet
提出日時 2019-12-17 20:36:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,600 ms / 10,000 ms
コード長 17,448 bytes
コンパイル時間 3,380 ms
コンパイル使用メモリ 227,080 KB
最終ジャッジ日時 2025-01-08 11:56:54
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#ifndef call_from_test
#include<bits/stdc++.h>
using namespace std;
#define call_from_test
#ifndef call_from_test
#include<bits/stdc++.h>
using namespace std;
#endif
//BEGIN CUT HERE
namespace FFT{
using dbl = double;
struct num{
dbl x,y;
num(){x=y=0;}
num(dbl x,dbl y):x(x),y(y){}
};
inline num operator+(num a,num b){
return num(a.x+b.x,a.y+b.y);
}
inline num operator-(num a,num b){
return num(a.x-b.x,a.y-b.y);
}
inline num operator*(num a,num b){
return num(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
}
inline num conj(num a){
return num(a.x,-a.y);
}
int base=1;
vector<num> rts={{0,0},{1,0}};
vector<int> rev={0,1};
const dbl PI=asinl(1)*2;
void ensure_base(int nbase){
if(nbase<=base) return;
rev.resize(1<<nbase);
for(int i=0;i<(1<<nbase);i++)
rev[i]=(rev[i>>1]>>1)+((i&1)<<(nbase-1));
rts.resize(1<<nbase);
while(base<nbase){
dbl angle=2*PI/(1<<(base+1));
for(int i=1<<(base-1);i<(1<<base);i++){
rts[i<<1]=rts[i];
dbl angle_i=angle*(2*i+1-(1<<base));
rts[(i<<1)+1]=num(cos(angle_i),sin(angle_i));
}
base++;
}
}
void fft(vector<num> &as){
int n=as.size();
assert((n&(n-1))==0);
int zeros=__builtin_ctz(n);
ensure_base(zeros);
int shift=base-zeros;
for(int i=0;i<n;i++)
if(i<(rev[i]>>shift))
swap(as[i],as[rev[i]>>shift]);
for(int k=1;k<n;k<<=1){
for(int i=0;i<n;i+=2*k){
for(int j=0;j<k;j++){
num z=as[i+j+k]*rts[j+k];
as[i+j+k]=as[i+j]-z;
as[i+j]=as[i+j]+z;
}
}
}
}
vector<long long> multiply(vector<int> &as,vector<int> &bs){
int need=as.size()+bs.size()-1;
int nbase=0;
while((1<<nbase)<need) nbase++;
ensure_base(nbase);
int sz=1<<nbase;
vector<num> fa(sz);
for(int i=0;i<sz;i++){
int x=(i<(int)as.size()?as[i]:0);
int y=(i<(int)bs.size()?bs[i]:0);
fa[i]=num(x,y);
}
fft(fa);
num r(0,-0.25/sz);
for(int i=0;i<=(sz>>1);i++){
int j=(sz-i)&(sz-1);
num z=(fa[j]*fa[j]-conj(fa[i]*fa[i]))*r;
if(i!=j)
fa[j]=(fa[i]*fa[i]-conj(fa[j]*fa[j]))*r;
fa[i]=z;
}
fft(fa);
vector<long long> res(need);
for(int i=0;i<need;i++)
res[i]=round(fa[i].x);
return res;
}
};
//END CUT HERE
#ifndef call_from_test
signed main(){
int n;
scanf("%d",&n);
vector<int> as(n+1,0),bs(n+1,0);
for(int i=1;i<=n;i++) scanf("%d %d",&as[i],&bs[i]);
auto cs=FFT::multiply(as,bs);
for(int i=1;i<=n*2;i++) printf("%lld\n",cs[i]);
return 0;
}
/*
verified on 2019/11/26
https://atcoder.jp/contests/atc001/tasks/fft_c
*/
#endif
#undef call_from_test
#endif
//BEGIN CUT HERE
struct bigint {
using ll = long long;
using vll = vector<ll>;
constexpr static ll base = 1000000000;
constexpr static ll base_digits = 9;
vll a;
ll sign;
bigint():sign(1){}
bigint(ll v){*this=v;}
bigint(const string &s){read(s);}
static bigint add_identity(){return bigint(0);}
static bigint mul_identity(){return bigint(1);}
void operator=(ll v){
sign=1;
if(v<0) sign=-1,v=-v;
for(;v>0;v=v/base) a.push_back(v%base);
}
bigint operator+(const bigint &v) const{
if(sign==v.sign){
bigint res=v;
for(ll i=0,carry=0;i<(ll)max(a.size(),v.a.size())||carry;++i){
if(i==(ll)res.a.size()) res.a.push_back(0);
res.a[i]+=carry+(i<(ll)a.size()?a[i]:0);
carry=res.a[i]>=base;
if(carry) res.a[i]-=base;
}
return res;
}
return *this -(-v);
}
bigint operator-(const bigint &v) const{
if(sign==v.sign){
if(abs()>=v.abs()){
bigint res=*this;
for(ll i=0,carry=0;i<(ll)v.a.size()||carry;++i){
res.a[i]-=carry+(i<(ll)v.a.size()?v.a[i]:0);
carry=res.a[i]<0;
if(carry) res.a[i]+=base;
}
res.trim();
return res;
}
return -(v-*this);
}
return *this+(-v);
}
void operator*=(ll v){
if(v<0) sign=-sign,v=-v;
for(ll i=0,carry=0;i<(ll)a.size()|| carry;++i){
if(i ==(ll)a.size()) a.push_back(0);
ll cur=a[i] *(ll)v+carry;
carry=(ll)(cur/base);
a[i]=(ll)(cur%base);
// asm("divl %%ecx" : "=a"(carry),"=d"(a[i]) : "A"(cur),"c"(base));
}
trim();
}
bigint operator*(ll v) const{
bigint res=*this;
res *=v;
return res;
}
friend pair<bigint,bigint> divmod(const bigint &a1,const bigint &b1){
ll norm=base/(b1.a.back()+1);
bigint a=a1.abs()*norm;
bigint b=b1.abs()*norm;
bigint q,r;
q.a.resize(a.a.size());
for(ll i=a.a.size()-1;i>=0;i--){
r *=base;
r+=a.a[i];
ll s1=r.a.size()<=b.a.size() ? 0 : r.a[b.a.size()];
ll s2=r.a.size()<=b.a.size()-1 ? 0 : r.a[b.a.size()-1];
ll d=((ll)base*s1+s2)/b.a.back();
r-=b*d;
while(r<0) r+=b,--d;
q.a[i]=d;
}
q.sign=a1.sign*b1.sign;
r.sign=a1.sign;
q.trim();
r.trim();
return make_pair(q,r/norm);
}
bigint operator/(const bigint &v) const{
return divmod(*this,v).first;
}
bigint operator%(const bigint &v) const{
return divmod(*this,v).second;
}
void operator/=(ll v){
if(v<0) sign=-sign,v=-v;
for(ll i=(ll)a.size()-1,rem=0;i>=0;--i){
ll cur=a[i]+rem *(ll)base;
a[i]=(ll)(cur/v);
rem=(ll)(cur%v);
}
trim();
}
bigint operator/(ll v) const{
bigint res=*this;
res/=v;
return res;
}
ll operator%(ll v) const{
if(v<0) v=-v;
ll m=0;
for(ll i=a.size()-1;i>=0;--i) m=(a[i]+m *(ll)base)%v;
return m*sign;
}
void operator+=(const bigint &v){
*this=*this+v;
}
void operator-=(const bigint &v){
*this=*this-v;
}
void operator*=(const bigint &v){
*this=*this*v;
}
void operator/=(const bigint &v){
*this=*this/v;
}
bool operator<(const bigint &v) const{
if(sign!=v.sign) return sign<v.sign;
if(a.size()!=v.a.size()) return a.size()*sign<v.a.size()*v.sign;
for(ll i=a.size()-1;i>=0;i--)
if(a[i]!=v.a[i]) return a[i]*sign<v.a[i]*sign;
return false;
}
bool operator>(const bigint &v) const{
return v<*this;
}
bool operator<=(const bigint &v) const{
return !(v<*this);
}
bool operator>=(const bigint &v) const{
return !(*this<v);
}
bool operator==(const bigint &v) const{
return !(*this<v)&&!(v<*this);
}
bool operator!=(const bigint &v) const{
return *this<v|| v<*this;
}
void trim(){
while(!a.empty()&&!a.back()) a.pop_back();
if(a.empty()) sign=1;
}
bool isZero() const{
return a.empty()||(a.size()==1&&!a[0]);
}
bigint operator-() const{
bigint res=*this;
res.sign=-sign;
return res;
}
bigint abs() const{
bigint res=*this;
res.sign*=res.sign;
return res;
}
ll longValue() const{
ll res=0;
for(ll i=a.size()-1;i>=0;i--) res=res*base+a[i];
return res*sign;
}
friend bigint gcd(const bigint &a,const bigint &b){
return b.isZero() ? a : gcd(b,a%b);
}
friend bigint lcm(const bigint &a,const bigint &b){
return a/gcd(a,b)*b;
}
void read(const string &s){
sign=1;
a.clear();
ll pos=0;
while(pos<(ll)s.size()&&(s[pos]=='-'|| s[pos]=='+')){
if(s[pos]=='-') sign=-sign;
++pos;
}
for(ll i=s.size()-1;i>=pos;i-=base_digits){
ll x=0;
for(ll j=max(pos,i-base_digits+1);j<=i;j++) x=x*10+s[j]-'0';
a.push_back(x);
}
trim();
}
friend istream &operator>>(istream &stream,bigint &v){
string s;
stream>>s;
v.read(s);
return stream;
}
friend ostream &operator<<(ostream &stream,const bigint &v){
if(v.sign==-1) stream<<'-';
stream<<(v.a.empty()?0:v.a.back());
for(ll i=(ll)v.a.size()-2;i>=0;--i)
stream<<setw(base_digits)<<setfill('0')<<v.a[i];
return stream;
}
static vll convert_base(const vll &a,ll old_digits,ll new_digits){
vll p(max(old_digits,new_digits)+1);
p[0]=1;
for(ll i=1;i<(ll)p.size();i++) p[i]=p[i-1]*10;
vll res;
ll cur=0;
ll cur_digits=0;
for(ll i=0;i<(ll)a.size();i++){
cur+=a[i]*p[cur_digits];
cur_digits+=old_digits;
while(cur_digits>=new_digits){
res.push_back(signed(cur%p[new_digits]));
cur/=p[new_digits];
cur_digits-=new_digits;
}
}
res.push_back((signed)cur);
while(!res.empty()&&!res.back()) res.pop_back();
return res;
}
static vll karatsubaMultiply(const vll &a,const vll &b){
ll n=a.size();
vll res(n+n);
if(n<=32){
for(ll i=0;i<n;i++)
for(ll j=0;j<n;j++)
res[i+j]+=a[i]*b[j];
return res;
}
ll k=n>>1;
vll a1(a.begin(),a.begin()+k);
vll a2(a.begin()+k,a.end());
vll b1(b.begin(),b.begin()+k);
vll b2(b.begin()+k,b.end());
vll a1b1=karatsubaMultiply(a1,b1);
vll a2b2=karatsubaMultiply(a2,b2);
for(ll i=0;i<k;i++) a2[i]+=a1[i];
for(ll i=0;i<k;i++) b2[i]+=b1[i];
vll r=karatsubaMultiply(a2,b2);
for(ll i=0;i<(ll)a1b1.size();i++) r[i]-=a1b1[i];
for(ll i=0;i<(ll)a2b2.size();i++) r[i]-=a2b2[i];
for(ll i=0;i<(ll)r.size();i++) res[i+k]+=r[i];
for(ll i=0;i<(ll)a1b1.size();i++) res[i]+=a1b1[i];
for(ll i=0;i<(ll)a2b2.size();i++) res[i+n]+=a2b2[i];
return res;
}
bigint operator*(const bigint &v) const{
constexpr static ll nbase = 10000;
constexpr static ll nbase_digits = 4;
vll a=convert_base(this->a,base_digits,nbase_digits);
vll b=convert_base(v.a,base_digits,nbase_digits);
//*/
while(a.size()<b.size()) a.push_back(0);
while(b.size()<a.size()) b.push_back(0);
while(a.size() &(a.size()-1)) a.push_back(0),b.push_back(0);
vll c=karatsubaMultiply(a,b);
/*/
if(a.empty()) a.push_back(0);
if(b.empty()) b.push_back(0);
vll c=FFT::multiply(a,b);
//*/
bigint res;
res.sign=sign*v.sign;
for(ll i=0,carry=0;i<(ll)c.size();i++){
ll cur=c[i]+carry;
res.a.push_back((ll)(cur%nbase));
carry=(ll)(cur/nbase);
if(i+1==(int)c.size()&&carry>0) c.push_back(0);
}
res.a=convert_base(res.a,nbase_digits,base_digits);
res.trim();
return res;
}
};
//END CUT HERE
#ifndef call_from_test
#define call_from_test
#ifndef call_from_test
#include<bits/stdc++.h>
using namespace std;
#endif
//BEGIN CUT HERE
template<typename T>
struct BIT{
int n;
vector<T> bit;
//1-indexed
BIT(int n_):n(n_+1),bit(n,0){}
T sum(int i){
T s(0);
for(int x=i;x>0;x-=(x&-x))
s+=bit[x];
return s;
}
void add(int i,T a){
if(i==0) return;
for(int x=i;x<=n;x+=(x&-x))
bit[x]+=a;
}
int lower_bound(int w){
if(w<=0) return 0;
int x=0,r=1;
while(r<n) r<<=1;
for(int k=r;k>0;k>>=1){
if(x+k<=n&&bit[x+k]<w){
w-=bit[x+k];
x+=k;
}
}
return x+1;
}
// [l, r)
T query(int l,int r){
return sum(r-1)-sum(l-1);
}
T sum0(int i){
return sum(i+1);
}
void add0(int i,T a){
add(i+1,a);
}
T query0(int l,int r){
return sum(r)-sum(l);
}
};
//END CUT HERE
#ifndef call_from_test
#define call_from_test
#include "../tools/fastio.cpp"
#undef call_from_test
// lower_bound
signed ARC033_C(){
int q;
cin>>q;
BIT<int> bit(2e5);
while(q--){
int t,x;
cin>>t>>x;
if(t==1) bit.add(x,1);
if(t==2){
int k=bit.lower_bound(x);
bit.add(k,-1);
cout<<k<<endl;
}
}
return 0;
}
/*
verified on 2019/11/11
https://atcoder.jp/contests/arc033/tasks/arc033_3
*/
signed main(){
ARC033_C();
return 0;
}
#endif
#ifndef call_from_test
#include<bits/stdc++.h>
using namespace std;
#endif
//BEGIN CUT HERE
template<typename R, size_t N>
struct SquareMatrix{
typedef array<R, N> arr;
typedef array<arr, N> mat;
mat dat;
SquareMatrix(){
for(size_t i=0;i<N;i++)
for(size_t j=0;j<N;j++)
dat[i][j]=R::add_identity();
}
bool operator==(const SquareMatrix& a) const{
return dat==a.dat;
}
size_t size() const{return N;}
arr& operator[](size_t k){return dat[k];}
const arr& operator[](size_t k) const {return dat[k];}
static SquareMatrix add_identity(){return SquareMatrix();}
static SquareMatrix mul_identity(){
SquareMatrix res;
for(size_t i=0;i<N;i++) res[i][i]=R::mul_identity();
return res;
}
SquareMatrix operator*(const SquareMatrix &B) const{
SquareMatrix res;
for(size_t i=0;i<N;i++)
for(size_t j=0;j<N;j++)
for(size_t k=0;k<N;k++)
res[i][j]=res[i][j]+(dat[i][k]*B[k][j]);
return res;
}
SquareMatrix operator+(const SquareMatrix &B) const{
SquareMatrix res;
for(size_t i=0;i<N;i++)
for(size_t j=0;j<N;j++)
res[i][j]=dat[i][j]+B[i][j];
return res;
}
SquareMatrix pow(long long n) const{
SquareMatrix a=*this,res=mul_identity();
while(n){
if(n&1) res=res*a;
a=a*a;
n>>=1;
}
return res;
}
};
//END CUT HERE
#ifndef call_from_test
#define call_from_test
#include "../tools/fastio.cpp"
#include "../tools/compress.cpp"
#include "../mod/mint.cpp"
#include "../segtree/basic/ushi.cpp"
#undef call_from_test
//INSERT ABOVE HERE
signed YAHOO2019_FINAL_D(){
using ll = long long;
using M = Mint<int>;
using MM = SquareMatrix<M, 2>;
int n,q;
cin>>n>>q;
vector<ll> ts(q),ls(q),rs(q),ps(q);
for(ll i=0;i<q;i++){
cin>>ts[i];
if(ts[i]==1) cin>>ps[i];
if(ts[i]==2) cin>>ls[i]>>rs[i];
}
vector<ll> vs;
for(ll i=0;i<q;i++){
if(ts[i]==1){
vs.emplace_back(ps[i]);
vs.emplace_back(ps[i]+1);
vs.emplace_back(ps[i]+2);
vs.emplace_back(ps[i]+3);
}
if(ts[i]==2){
vs.emplace_back(ls[i]);
vs.emplace_back(ls[i]+1);
vs.emplace_back(rs[i]);
vs.emplace_back(rs[i]+1);
}
}
vs.emplace_back(0);
vs.emplace_back(n+1);
vs=compress(vs);
auto ms=dict(vs);
MM A;
A[0][0]=M(1);A[0][1]=M(1);
A[1][0]=M(1);A[1][1]=M(0);
vector<MM> vt(vs.size()-1,A);
for(ll i=0;i+1<(ll)vs.size();i++){
vt[i]=A.pow(vs[i+1]-vs[i]);
}
MM I=MM::mul_identity();
auto f=[&](MM a,MM b){return b*a;};
SegmentTree<MM> seg(f,I);
seg.build(vt);
vector<int> used(vs.size(),0);
for(ll i=0;i<q;i++){
if(ts[i]==1){
ll k=ms[ps[i]];
used[k]^=1;
MM B;
for(ll j=k;j<=k+2;j++){
if(used[j]){
B[0][0]=M(0);B[0][1]=M(0);
B[1][0]=M(1);B[1][1]=M(0);
}else{
B[0][0]=M(!used[j-1]);B[0][1]=M(!used[j-2]);
B[1][0]=M(1);B[1][1]=M(0);
}
seg.set_val(j,B);
}
}
if(ts[i]==2){
ll l=ms[ls[i]],r=ms[rs[i]];
if(used[l]){
cout<<0<<"\n";
continue;
}
auto B=seg.query(l+1,r+1);
cout<<B[0][0].v<<"\n";
}
}
cout<<flush;
return 0;
}
/*
verified on 2019/10/29
https://atcoder.jp/contests/yahoo-procon2019-final/tasks/yahoo_procon2019_final_d
*/
signed DDCC2019_FINAL_D(){
string s;
cin>>s;
struct M{
uint32_t v;
M(){*this=add_identity();}
M(uint32_t v):v(v){}
M operator+(const M &a)const{return M(v+a.v);}
M operator*(const M &a)const{return M(v*a.v);}
static M add_identity(){return M(0);}
static M mul_identity(){return M(1);}
};
using SM = SquareMatrix<M, 6>;
auto f=[](SM a,SM b){return a*b;};
SM ti=SM::mul_identity();
SegmentTree<SM> seg(f,ti);
vector<SM> vt(s.size(),ti);
for(int i=0;i<(int)s.size();i++){
if(s[i]=='D') vt[i][0][1]=1;
if(s[i]=='I') vt[i][1][2]=1;
if(s[i]=='S') vt[i][2][3]=1;
if(s[i]=='C') vt[i][3][4]=1;
if(s[i]=='O') vt[i][4][5]=1;
}
seg.build(vt);
int q;
cin>>q;
for(int i=0;i<q;i++){
int l,r;
cin>>l>>r;
l--;
cout<<seg.query(l,r)[0][5].v<<"\n";
}
cout<<flush;
return 0;
}
/*
verified on 2019/10/29
https://atcoder.jp/contests/ddcc2019-final/tasks/ddcc2019_final_d
*/
signed main(){
//YAHOO2019_FINAL_D();
//DDCC2019_FINAL_D();
return 0;
}
#endif
#undef call_from_test
//INSERT ABOVE HERE
using ll = long long;
signed YUKI_129(){
bigint MOD = 1000000000;
ll n,m;
cin>>n>>m;
n/=1000;
n%=m;
bigint res(1);
for(int i=0;i<n;i++){
res*=m-i;
res/=i+1;
}
cout<<res%MOD<<endl;
return 0;
}
/*
verified on 2019/10/17
https://yukicoder.me/problems/no/129
*/
signed YUKI_303(){
ll l;
cin>>l;
if(l==2){
cout<<3<<endl;
cout<<"INF"<<endl;
return 0;
}
cout<<l<<endl;
using M = SquareMatrix<bigint, 2>;
M A;
A[0][0]=1;A[0][1]=1;
A[1][0]=1;A[1][1]=0;
if(l&1) cout<<A.pow(l)[1][0]<<endl;
else{
auto B=A.pow(l/2);
auto X=(B*B)[1][0];
auto Y=B[1][0];
cout<<X-Y*Y<<endl;
}
return 0;
}
/*
verified on 2019/10/17
https://yukicoder.me/problems/no/303
*/
const int MAX = 1e5+100;
ll dp[MAX];
pair<bigint, bigint> dfs(int l,int r){
if(l+1>=r) return make_pair(bigint(dp[l]),bigint(l+1));
int m=(l+r)>>1;
auto v=dfs(l,m);
auto t=dfs(m,r);
t.first*=v.second;
t.first+=v.first;
t.second*=v.second;
return t;
}
signed YUKI_696(){
int n;
cin>>n;
vector<int> p(n);
for(int i=0;i<n;i++) cin>>p[i];
BIT<ll> bit(n+1);
for(int i=0;i<n;i++) bit.add(p[i],1);
for(int i=0;i<n;i++){
dp[n-(i+1)]=bit.sum(p[i]-1);
bit.add(p[i],-1);
}
cout<<bigint(dfs(0,n).first+1)<<endl;
return 0;
}
/*
verified on 2019/10/17
https://yukicoder.me/problems/no/696
*/
signed main(){
//YUKI_129();
//YUKI_303();
YUKI_696();
return 0;
}
#endif
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0