結果
問題 | No.1683 Robot Guidance |
ユーザー |
![]() |
提出日時 | 2021-09-17 22:05:06 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 205 ms / 2,000 ms |
コード長 | 3,765 bytes |
コンパイル時間 | 3,112 ms |
コンパイル使用メモリ | 252,768 KB |
実行使用メモリ | 30,624 KB |
最終ジャッジ日時 | 2024-06-29 20:38:19 |
合計ジャッジ時間 | 10,816 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
#include <bits/stdc++.h> using namespace std; using Int = long long; const char newl = '\n'; template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;} template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;} template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);} template<typename T=int> vector<T> read(size_t n){ vector<T> ts(n); for(size_t i=0;i<n;i++) cin>>ts[i]; return ts; } template<typename T, T MOD = 1000000007> struct Mint{ inline static constexpr T mod = MOD; T v; Mint():v(0){} Mint(signed v):v(v){} Mint(long long t){v=t%MOD;if(v<0) v+=MOD;} Mint pow(long long k){ Mint res(1),tmp(v); while(k){ if(k&1) res*=tmp; tmp*=tmp; k>>=1; } return res; } static Mint add_identity(){return Mint(0);} static Mint mul_identity(){return Mint(1);} Mint inv(){return pow(MOD-2);} Mint& operator+=(Mint a){v+=a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator-=(Mint a){v+=MOD-a.v;if(v>=MOD)v-=MOD;return *this;} Mint& operator*=(Mint a){v=1LL*v*a.v%MOD;return *this;} Mint& operator/=(Mint a){return (*this)*=a.inv();} Mint operator+(Mint a) const{return Mint(v)+=a;} Mint operator-(Mint a) const{return Mint(v)-=a;} Mint operator*(Mint a) const{return Mint(v)*=a;} Mint operator/(Mint a) const{return Mint(v)/=a;} Mint operator+() const{return *this;} Mint operator-() const{return v?Mint(MOD-v):Mint(v);} bool operator==(const Mint a)const{return v==a.v;} bool operator!=(const Mint a)const{return v!=a.v;} static Mint comb(long long n,int k){ Mint num(1),dom(1); for(int i=0;i<k;i++){ num*=Mint(n-i); dom*=Mint(i+1); } return num/dom; } }; template<typename T, T MOD> ostream& operator<<(ostream &os,Mint<T, MOD> m){os<<m.v;return os;} template<typename M_> class Enumeration{ using M = M_; protected: inline static vector<M> fact,finv,invs; public: static void init(int n){ n=min<decltype(M::mod)>(n,M::mod-1); int m=fact.size(); if(n<m) return; fact.resize(n+1,1); finv.resize(n+1,1); invs.resize(n+1,1); if(m==0) m=1; for(int i=m;i<=n;i++) fact[i]=fact[i-1]*M(i); finv[n]=M(1)/fact[n]; for(int i=n;i>=m;i--) finv[i-1]=finv[i]*M(i); for(int i=m;i<=n;i++) invs[i]=finv[i]*fact[i-1]; } static M Fact(int n){ init(n); return fact[n]; } static M Finv(int n){ init(n); return finv[n]; } static M Invs(int n){ init(n); return invs[n]; } static M C(int n,int k){ if(n<k or k<0) return M(0); init(n); return fact[n]*finv[n-k]*finv[k]; } static M P(int n,int k){ if(n<k or k<0) return M(0); init(n); return fact[n]*finv[n-k]; } // put n identical balls into k distinct boxes static M H(int n,int k){ if(n<0 or k<0) return M(0); if(!n and !k) return M(1); init(n+k); return C(n+k-1,n); } }; //INSERT ABOVE HERE signed main(){ cin.tie(0); ios::sync_with_stdio(0); int a,b,x,y; cin>>a>>b>>x>>y; int p=b/2+1; int q=(b+1)/2; using M = Mint<int>; using E = Enumeration<M>; const int MAX = 1e6+100; vector<M> ss(MAX),ts(MAX); for(int i=0;i<MAX;i++){ { // s + t = i // s - t = -x int s=(i-x)/2; int t=(i+x)/2; int neg=(p+1)/2,pos=p-neg; if(s+t==i and s-t==-x and min(s,t)>=0) ss[i]=E::H(s,pos)*E::H(t,neg); } { // s + t = i // s - t = -y int s=(i-y)/2; int t=(i+y)/2; int neg=(q+1)/2,pos=q-neg; if(s+t==i and s-t==-y and min(s,t)>=0) ts[i]=E::H(s,pos)*E::H(t,neg); } //if(i<10) cout<<i<<':'<<ss[i]<<' '<<ts[i]<<endl; } M ans{0}; for(int i=0;i<=a;i++) ans+=ss[i]*ts[a-i]; cout<<ans<<newl; return 0; }