SRM 150 DIV1 Easy - InterestingDigits (復習×○)

問題


http://community.topcoder.com/stat?c=problem_statement&pm=1523&rd=4555

3の倍数はその各桁の数の和も同じく3の倍数になる。

上記に対し、x進数のときにも同じ法則が成り立つ倍数を求める。

解き方


x進数を考えた時、x^4まで考えればよく、xは最大で30であるので
計算量は30^4 *30 =30^5 =3*10^5となる。

条件を満たさない場合はbreak,continue等で打ち切りを行えば
十分全探索で間に合う。

コード


using namespace std;

#define all(c) (c).begin(),(c).end()
#define FORE(i,d,e) for(int i=d;i<e;i++)
#define FOR(i,s,e) for (int i = int(s); i != int(e); i++)
#define FORIT(i,c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); i++)
#define ISEQ(c) (c).begin(), (c).end()

class InterestingDigits {

public: vector<int> digits(int base) {
vector<int> ans;

FORE(i,2,base){
int valid=1;
FORE(j,1,pow(base,4)){
if(j%i!=0)continue;
int tmp=j,num=0;
while(tmp>0){
num+=tmp%base;
tmp/=base;
}
if(num%i!=0){
valid=0;
break;
}
}
if(valid)ans.push_back(i);
}

return ans;
}

};
このエントリーをはてなブックマークに追加