K尾相等数

Problem 105: K尾相等数


Time Limit:3 Ms| Memory Limit:64 MB
Difficulty:1


Description
输入一个自然数K（K>1）,如果存在自然数M和N（M>N），使得K^M和K^N均大于等于1000，且他们的末尾三位数相等，则称M和N是一对“K尾相等数”。下面请编程求出M+N最小的K尾相等数。
Input
第一行包含一个正整数T，T<10000，表示有T组数据；
随后有N行，每行包括一个整数K（K<2*10^10);
Output
对于输入的每个整数K，输出对应的M+N的最小值；
Sample Input
1
2


Sample Output
120

思路：开一个1000的数组，在k大于1000的情况下，每乘一次k，就把它后三位标记到数组里面，用当前次方数来标记。当找到一个后三位已经找过的，当前次方数加上已经找过的次方数，就是最小的m + n。若k不大于1000，记录几次乘k之后可以达到三位以上，然后用它当计数器。

#include<stdio.h>
#include<string.h>

int visit[1005];

int pow(int k);
int main(){
	int t, k, n;
	
	scanf("%d", &t);
	while(t--){
		scanf("%d", &k);
		memset(visit, 0, sizeof(visit));
		
		printf("%d\n", pow(k));
	}
	return 0;
}

int pow(int k){
	int c, s, r;
	s = 1;
	r = k;
	while(r < 1000){
		r *= k;
		s++;
	}
	c = 0;
	r = 1;
	while(1){
		r = r * k % 1000;
		c++;
		s--;
		if(s <= 0){
			if(visit[r]){
				return visit[r] + c;
			}
			else
				visit[r] = c;
		}
	}
}