/********************************************************** Yukicoder No.940 ワープ ε=ε=ε=ε=ε=│;p>д<│ https://yukicoder.me/problems/no/940 <解法> n=X+Y+Z とおく。 nより十分に大きいNをとり、 T=(1+x+x^2+..+x^(N-1))(1+y+y^2+..+y^(N-1))(1+z+z^2+..+z^(N-1))-1 =(1-x^N)/(1-x) (1-y^N)/(1-y) (1-z^N)/(1-z) -1 とおくと、 F=Σ(m=1..n)T^m の x^X y^Y z^Z の係数を求めればよい。 ※理由:x^i y^j z^k というのが、1回の移動でx方向にi、y方向にj、z方向にk 進むことを表していて、 m回移動したときに(X,Y,Z)に到達する場合の数が T^m の x^X y^Y z^Z の係数と等しいので。 F=Σ(m=1..n)T^m の x^X y^Y z^Z の係数を求めるときには FをxでX回、yでY回、zでZ回偏微分してx=y=z=0 を入れた値を X!Y!Z! で割ればよい。 この計算を行うときには、Nを十分大きくとっておけば、 T=(1-x^N)/(1-x) (1-y^N)/(1-y) (1-z^N)/(1-z) -1 のかわりに T=1/(1-x)(1-y)(1-z) -1 として計算してもよいことがわかる。 S=(1-x)^(-1) (1-y)^(-1) (1-z)^(-1) とおくと、 F=Σ(m=1..n)(S-1)^m =(S-1)(1-(S-1)^n)/(1-(S-1)) これは、(S-1)^nを二項定理で展開した後、多項式の掛け算、割り算をすれば、Σ(m=1..∞)Am S^m の形に書ける。 (係数Am を m<=n の範囲のみ計算することにすれば O(n)で求まる) あとは、各 S^m についてx^X y^Y z^Z の係数を求めるには S^m=(1-x)^(-m) (1-y)^(-m) (1-z)^(-m) を xでX回、yでY回、zでZ回偏微分してx=y=z=0 を入れればよいが、 これは m(m+1)..(m+X-1) と m(m+1)..(m+Y-1) と m(m+1)..(m+Z-1) の積になる。 **********************************************************/ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #pragma warning(disable:4996) typedef long long ll; #define MIN(a, b) ((a)>(b)? (b): (a)) #define MAX(a, b) ((a)<(b)? (b): (a)) #define LINF 9223300000000000000 #define INF 2140000000 const long long MOD = 1000000007; //const long long MOD = 998244353; using namespace std; int main(int argc, char* argv[]) { int n,K; scanf("%d%d", &n, &K); int m=n/K; if(K%2==0 && m%2) { printf("No\n"); return 0; } else if(K%2==0 || m%2==0) { printf("Yes\n"); vector > z(K); int i,j; int cnt=0; for(i=0; i > z(K); int i,j; int cnt=0; for(i=0; i > v; int p; for(p=0; p