結果

問題 No.1419 Power Moves
ユーザー publflpublfl
提出日時 2021-03-05 22:43:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 2,109 bytes
コンパイル時間 417 ms
コンパイル使用メモリ 48,640 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-07 03:33:36
合計ジャッジ時間 3,026 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:83:31: warning: 'inv' may be used uninitialized [-Wmaybe-uninitialized]
   83 |                 long long int inv;
      |                               ^~~

ソースコード

diff # 

#include <stdio.h>
#include <vector>
#define MOD 1000000007

long long int power(long long int a, long long int b)
{
	long long int ans = 1;
	long long int k = a;
	while(b)
	{
		if(b%2==1) ans*=k, ans%=MOD;
		k*=k, k%=MOD;
		b/=2;
	}
	return ans;
}
long long int inv(long long int k)
{
	return power(k,MOD-2);
}

std::pair<long long int, long long int> func(long long int a, long long int b, long long int m)
{
	if(b==0) return std::make_pair(0,1);
	else
	{
		if(b%2==0)
		{
			std::pair<long long int, long long int> P1 = func(a,b/2,m);
			std::pair<long long int, long long int> P2;
			P2.first = (P1.first * P1.second + P1.first * P1.second + (P1.second*P1.second)/m) % MOD;
			P2.first += (((P1.first*P1.first)%MOD)*m) % MOD, P2.first %= MOD;
			P2.second = (P1.second*P1.second)%m;
			return P2;
		}
		else
		{
			std::pair<long long int, long long int> P1 = func(a,(b-1)/2,m);
			std::pair<long long int, long long int> P2;
			P2.first = (P1.first * P1.second + P1.first * P1.second + (P1.second*P1.second)/m) % MOD;
			P2.first += (((P1.first*P1.first)%MOD)*m) % MOD, P2.first %= MOD;
			P2.second = (P1.second*P1.second)%m;
			
			P2.first *= a, P2.first %= MOD;
			P2.second *= a, P2.first += (P2.second/m), P2.first %= MOD, P2.second%= m;
			return P2;
		}
	}
}

long long int ans[100010];
int main()
{
	long long int a,b;
	scanf("%lld%lld",&a,&b);
	
	std::pair<long long int,long long int> C = func(2,b,a);
	
	C.second--;
	if(C.second<0) C.first--, C.second+=a;
	if(C.first<0) C.first += MOD;
	
	
	long long int A = C.first, B = C.second;
	if(a%2==0)
	{
		for(int i=0;i<a;i++)
		{
			if((i+B)%2==1) ans[i] = 0;
			else
			{
				long long int t = (i+B)/2;
				t %= (a/2);
				
				if(B>=t+(a/2)) ans[i] = 2*A+2;
				else if(B>=t) ans[i] = 2*A+1;
				else ans[i] = 2*A;
			}
		}
	}
	else
	{
		long long int inv;
		for(int i=1;i<a;i++)
		{
			if((2*i)%a==1)
			{
				inv = i;
				break;
			}
		}
		for(int i=0;i<a;i++)
		{
			long long int t = ((long long int)(i+B)*inv) % a;
			if(B>=t) ans[i] = A+1;
			else ans[i] = A;
		}
	}
	
	for(int i=0;i<a;i++) printf("%lld\n",(ans[i]*inv(power(2,b)))%MOD);
}
0