#302095 (C++17) No.775 tatyamと素数大富豪(hard)

提出ソース
結果

問題	No.775 tatyamと素数大富豪(hard)
コンテスト
ユーザー	👑 tatyam
提出日時	2018-12-06 19:06:24
言語	C++17 (gcc 13.3.0 + boost 1.87.0)
結果	WA
実行時間	-
コード長	6,931 bytes
コンパイル時間	7,792 ms
コンパイル使用メモリ	417,780 KB
最終ジャッジ日時	2025-01-06 18:29:31
ジャッジサーバーID （参考情報）	judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン	結果
sample	WA * 5
other	WA * 8 TLE * 1 MLE * 1
権限があれば一括ダウンロードができます
ソースコード

raw source code
#include <bits/stdc++.h>
using namespace std;
using namespace std::chrono;
using ll=long long;
using pii=pair<int,int>;
using pll=pair<ll,ll>;
using pdd=pair<double,double>;
using pdl=pair<double,ll>;
const ll LINF=0x3fffffffffffffff;
const ll MOD=1000000007;
const ll MODD=0x3b800001;
const int INF=0x3fffffff;
const double DINF=numeric_limits<double>::infinity();
const vector<pii> four={{1,0},{0,1},{-1,0},{0,-1}};
#define _overload4(_1,_2,_3,_4,name,...) name
#define _overload3(_1,_2,_3,name,...) name
#define _rep1(n) _rep2(i,n)
#define _rep2(i,n) _rep3(i,0,n)
#define _rep3(i,a,b) for(ll i=a;i<b;++i)
#define _rep4(i,a,b,c) for(ll i=a;i<b;i+=c)
#define rep(...) _overload4(__VA_ARGS__,_rep4,_rep3,_rep2,_rep1)(__VA_ARGS__)
#define _rrep1(n) _rrep2(i,n)
#define _rrep2(i,n) _rrep3(i,0,n)
#define _rrep3(i,a,b) for(ll i=b-1;i>=a;i--)
#define _rrep4(i,a,b,c) for(ll i=a+(b-a-1)/c*c;i>=a;i-=c)
#define rrep(...) _overload4(__VA_ARGS__,_rrep4,_rrep3,_rrep2,_rrep1)(__VA_ARGS__)
#define each(i,a) for(auto&& i:a)
#define sum(...) accumulate(range(__VA_ARGS__),0LL)
#define _range(i) (i).begin(),(i).end()
#define _range2(i,k) (i).begin(),(i).begin()+k
#define _range3(i,a,b) (i).begin()+a,(i).begin()+b
#define range(...) _overload3(__VA_ARGS__,_range3,_range2,_range)(__VA_ARGS__)
#define Yes(i) out(i?"Yes":"No")
#define YES(i) out(i?"YES":"NO")
//#define START auto start=system_clock::now()
//#define END auto end=system_clock::now();cerr<<duration_cast<milliseconds>(end-start).count()<<" ms\n"
#define elif else if
#define unless(a) if(!(a))
#define INT(...) int __VA_ARGS__;in(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;in(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;in(__VA_ARGS__)
#define CHR(...) char __VA_ARGS__;in(__VA_ARGS__)
#define DBL(...) double __VA_ARGS__;in(__VA_ARGS__)
#define vec(type,name,...) vector<type> name(__VA_ARGS__)
#define VEC(type,name,size) vector<type> name(size);in(name)
#define vv(type,name,h,...) vector<vector<type>>name(h,vector<type>(__VA_ARGS__))
#define VV(type,name,h,...) vector<vector<type>>name(h,vector<type>(__VA_ARGS__));in(name)
__attribute__((constructor)) void SETTINGS(){cin.tie(0); cout.tie(0); ios::sync_with_stdio(0); cout<<fixed<<setprecision(15);};
template<class T>
inline constexpr T gcd (T a,T b) {if(a==b)return a;else return gcd(b,(a-1)%b+1);}
template<class T>
inline constexpr T lcm (T a,T b) {a*b/gcd(a,b);}
template<class T>
inline constexpr T min(vector<T>& v){return *min_element(range(v));}
inline char min(string& v){return *min_element(range(v));}
template<class T>
inline constexpr T max(vector<T>& v){return *max_element(range(v));}
inline char max(string& v){return *max_element(range(v));}
template<typename T>
inline bool update_min(T& mn,const T& cnt){if(mn>cnt){mn=cnt;return 1;}else return 0;}
template<typename T>
inline bool update_max(T& mx,const T& cnt){if(mx<cnt){mx=cnt;return 1;}else return 0;}
inline void in() {}
template<class T>
istream& operator >> (istream& is, vector<T>& vec);
template<class T,size_t size>
istream& operator >> (istream& is, array<T,size>& vec);
template<class T,class L>
istream& operator >> (istream& is, pair<T,L>& p);
template<class T>
ostream& operator << (ostream& os, vector<T>& vec);
template<class T,class L>
ostream& operator << (ostream& os, pair<T,L>& p);
template<class T>
istream& operator >> (istream& is, vector<T>& vec){for(T& x: vec) is >> x;return is;}
template<class T,class L>
istream& operator >> (istream& is, pair<T,L>& p){is >> p.first;is >> p.second;return is;}
template<class T>
ostream& operator << (ostream& os, vector<T>& vec){os << vec[0];rep(i,1,vec.size()){os << ' ' << vec[i];}return os;}
template<class T>
ostream& operator << (ostream& os, deque<T>& deq){os << deq[0];rep(i,1,deq.size()){os << ' ' << deq[i];}return os;}
template<class T,class L>
ostream& operator << (ostream& os, pair<T,L>& p){os << p.first << " " << p.second;return os;}
template <class Head, class... Tail>
inline void in(Head&& head,Tail&&... tail){cin>>head;in(move(tail)...);}
template <class T>
inline void out(T t){cout<<t<<'\n';}
inline void out(){cout<<'\n';}
template <class Head, class... Tail>
inline void out(Head head,Tail... tail){cout<<head<<' ';out(move(tail)...);}
template <class T>
inline void err(T t){cerr<<t<<'\n';}
inline void err(){cerr<<'\n';}
template <class Head, class... Tail>
inline void err(Head head,Tail... tail){cerr<<head<<' ';out(move(tail)...);}


#undef vv
#include <boost/multiprecision/miller_rabin.hpp>
#include <boost/multiprecision/cpp_int.hpp>
using namespace boost::multiprecision;
int n,k;
vector<uint128_t>primes;
set<tuple<int,string,vector<int>,int>>s;
bool permsearch(int n,string num,vector<int>&card,int last=0){
    if(s.count({n,num,card,last}))return 0;
    s.insert({n,num,card,last});
    if(n==0){
        if(miller_rabin_test(uint128_t(num),40)){
            primes.push_back(uint128_t(num));
            k--;
            if(!k)return 1;
        }
        return 0;
    }
    bool b=1;
    rep(i,1,100,2)if(i%5&&card[i]){
        b=0;
        break;
    }
    if(b)return 0;
    b=1;
    rep(i,1,10)if(card[i]){
        b=0;
        break;
    }
    if(b){
        int c=int(uint128_t(num)%11);
        rep(i,10,100)c+=card[i]*i;
        if(c%11==0)return 0;
    }
    rrep(i,10,100){
        if(i%11==0&&card[i/11]){
            if(last==i)continue;
            int card1=card[i/11],card11=card[i],m=n;
            string num2=num;
            card[i/11]=0;
            card[i]=0;
            m-=card1+card11;
            rep(j,card11)num2=num2+to_string(i);
            rep(j,card1)num2=num2+to_string(i/11);
            while(card[i/11]!=card1||card[i]!=card11){
                if(permsearch(m,num2,card,i))return 1;
                if(card[i/11]!=card1){
                    card[i/11]++;
                    m++;
                    num2.pop_back();
                }
                else{
                    if(card1){
                        card[i]++;
                        card[i/11]--;
                        num2.pop_back();
                    }
                    else{
                        card[i]++;
                        m++;
                        num2.pop_back();
                        num2.pop_back();
                    }
                }
            }
            card[i/11]=card1;
            card[i]=card11;
        }
        elif(card[i]){
            card[i]--;
            if(permsearch(n-1,num+to_string(i),card))return 1;
            card[i]++;
        }
    }
    return 0;
}
signed main(){
    in(n,k);
    if(n==1){
        INT(a);
        if(miller_rabin_test(a,10))out(a);
        return puts("-1")&0;
    }
    int l=k;
    k+=100;
    VEC(int,a,n);
    if(sum(a)%3==0)return puts("-1")&0;
    vector<int>card(100);
    each(i,a)card[i]++;
    permsearch(n,"",card);
    sort(range(primes),greater<uint128_t>());
    rep(min(l,int(primes.size())))cout<<primes[i]<<endl;
    out(-1);
}
yukicoder

結果

ソースコード
