Submission #164964

# Submission time Handle Problem Language Result Execution time Memory
164964 2019-11-24T14:18:03 Z coloredrabbit Ω (kriii4_P3) C++17
0 / 100
2 ms 376 KB
#include <cstdio>
const int MOD = 1e9+7;
using ll = long long;
ll _pow(ll a, int n) {
	if (n <= 1) return n % 2 ? a : 1;
	ll h = _pow(a, n >> 1);
	return ((h * h) % MOD) * (n % 2 ? a : 1) % MOD;
}
ll rev(ll x) { return _pow(x, MOD - 2); }
int main() {
	int P, Q, N, K;
	ll v, ans = 0;
	scanf("%d%d%d%d", &P, &Q, &N, &K);
	if (K == 0 || Q == P) puts("0");
	else if (K == N || Q == 0) puts("1");
	else {
		if(2*Q == P)
			ans = K * rev(N) % MOD;
		else {
			v = Q * rev(P - Q);
			//ans = ((1 - _pow(v, K) + MOD) % MOD) * (rev(1 - _pow(v, N) + MOD) % MOD) % MOD;
			ans = (_pow(P-Q,N) - _pow(P-Q,N-K) * _pow(Q, K) % MOD + MOD)
				* rev(_pow(P - Q, N) - _pow(Q, N) + MOD) % MOD;
		}
		printf("%lld", ans);
	}
	
	return 0;
}

Compilation message

P3.cpp: In function 'int main()':
P3.cpp:12:5: warning: variable 'v' set but not used [-Wunused-but-set-variable]
  ll v, ans = 0;
     ^
P3.cpp:13:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d%d%d", &P, &Q, &N, &K);
  ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB Output is correct
2 Correct 2 ms 376 KB Output is correct
3 Correct 2 ms 256 KB Output is correct
4 Incorrect 2 ms 256 KB Output isn't correct
5 Halted 0 ms 0 KB -