結果
問題 | No.1458 Segment Function |
ユーザー |
![]() |
提出日時 | 2021-03-31 22:35:25 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 6 ms / 2,000 ms |
コード長 | 1,614 bytes |
コンパイル時間 | 4,180 ms |
コンパイル使用メモリ | 250,588 KB |
最終ジャッジ日時 | 2025-01-20 04:05:26 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 32 |
ソースコード
#include<bits/stdc++.h> #include<atcoder/all> using namespace std; using namespace atcoder; #define rep(i,n)for (int i = 0; i < int(n); ++i) #define rrep(i,n)for (int i = int(n)-1; i >= 0; --i) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() template<class T> void chmax(T& a, const T& b) {a = max(a, b);} template<class T> void chmin(T& a, const T& b) {a = min(a, b);} using ll = long long; using P = pair<int,int>; using VI = vector<int>; using VVI = vector<VI>; using VL = vector<ll>; using VVL = vector<VL>; int F[10] = {6, 2, 5, 5, 4, 5, 6, 4, 7, 6}; int f(int p) { int np = 0; while(p) { np += F[p % 10]; p /= 10; } return np; } int main() { ios::sync_with_stdio(false); cin.tie(0); string p_str, n_str; cin >> p_str >> n_str; if (n_str == "0") { cout << p_str << '\n'; return 0; } int p = 0; { int i = 0; if (p_str[i] == '-') p++, i++; int psz = p_str.size(); for(; i < psz; i++) { p += F[p_str[i] - '0']; } } if (n_str.size() <= 2) { int n = stoi(n_str) - 1; rep(_, n) p = f(p); cout << p << '\n'; return 0; } rep(_, 99) p = f(p); bool visited[10] = {}; int loop_len = 0; int nxt = p; while(!visited[nxt]) { loop_len++; visited[nxt] = true; nxt = F[nxt]; } int n = 0; for(char x: n_str) { n = (n * 10 + x - '0') % loop_len; } n = ((n - 100) % loop_len + loop_len) % loop_len; rep(_, n) p = F[p]; cout << p << '\n'; }