[SWEA] 6026 성수의 비밀번호 공격

Time Lapse :2hour 59min 53sec

6026.c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <stdio.h>
#include <memory.h>
#define MOD 1000000007
int N, M, choose[100];
long answer, pactorial[101], LOOKUP_TABLE[100][100][2];
long pow(long x,int n){
long ret = 1;
while(n){
if(n&1)
ret = (ret*x)%MOD;
x = (x*x)%MOD;
n>>=1;
}
return ret;
}
void DFS(int depth, int need){
if(!((N-depth)^need)) answer = (answer + pactorial[need])%MOD;
else if(!(need)) answer = (answer + pow(M,N-depth))%MOD;
else {
for (int i = 0; i < M; i++) {
if (choose[i]) {
if (LOOKUP_TABLE[depth][need][0] == -1) {
LOOKUP_TABLE[depth][need][0] = answer; choose[i] = 0;
DFS(depth + 1, need - 1);
LOOKUP_TABLE[depth][need][0] = (answer - LOOKUP_TABLE[depth][need][0] + MOD) % MOD; choose[i] = 1;
}
else answer = (answer + LOOKUP_TABLE[depth][need][0]) % MOD;
}
else {
if (LOOKUP_TABLE[depth][need][1] == -1) {
LOOKUP_TABLE[depth][need][1] = answer;
DFS(depth + 1, need);
LOOKUP_TABLE[depth][need][1] = (answer - LOOKUP_TABLE[depth][need][1] + MOD) % MOD;
}
else answer = (answer + LOOKUP_TABLE[depth][need][1]) % MOD;
}
}
}
}

int main(int argc, char** argv) {
int T, tc;
memset(choose, 1, sizeof(choose));
choose[0] = 0;
scanf("%d", &T);
pactorial[1] = 1;
for (int i = 2; i <= 100; i++) pactorial[i] = (pactorial[i - 1] * i) % MOD;
for(tc = 1; tc<=T;tc++) {
scanf("%d %d", &M, &N);
if(M==N) printf("#%d %d\n",tc,pactorial[M]);
else {
answer = 0;
memset(LOOKUP_TABLE,-1,sizeof(LOOKUP_TABLE));
DFS(1, M-1 );
printf("#%d %d\n", tc,(answer*M)%MOD);
}
}
return 0; //정상종료시 반드시 0을 리턴해야 합니다.
}
Author: Song Hayoung
Link: https://songhayoung.github.io/2020/08/04/PS/SWEA/6026/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.