题意:将编号为1~n的球放到k个盒子中,要求满足,如果球x在盒子i中,盒子i+1中要有编号为2*x的球,盒子i-1中要有编号为x/2的球。求第一个盒子最多有多少种。
开始的时候我和wiking大神想的是所有的奇数肯定放第一个盒子里(满足x*2^(k-1)<=n的最大奇数x),然后对于每个以奇数为首的链,求长度除以k的和,结果发现这尼玛,完全动不了啊,时间复杂度太高。
事实证明少想了一步。。。。
我们是横着求链的长度,除以k,再相加。。。。
其实可以竖着求,一段一段的往右移
例如k=3的时候可以看如下图:
用C++交,CE了,看来自己留存的那份大数模板不怎么行。
用G++交,TLE了,看来自己留存的那份大数模板真的不怎么行。
用Java终于过了。。。。
import java.math.*;
import java.util.*;
public class Main{
public static void main(String args[]){
BigInteger n,ans,tmp,cnt;
int k,t;
Scanner cin=new Scanner(System.in);
t=cin.nextInt();
while(t-->0){
n=cin.nextBigInteger();
k=cin.nextInt();
ans=BigInteger.ZERO;
tmp=BigInteger.valueOf(1<<(k-1));
while(true){
n=n.divide(tmp);
cnt=n.subtract(n.divide(BigInteger.valueOf(2)));
if(cnt.equals(BigInteger.ZERO)) break;
ans=ans.add(cnt);
n=n.divide(BigInteger.valueOf(2));
}
System.out.println(ans);
}
}
}