結果

問題 No.1458 Segment Function
コンテスト
ユーザー 沙耶花
提出日時 2021-03-31 21:28:28
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 1,255 bytes
コンパイル時間 6,365 ms
コンパイル使用メモリ 251,984 KB
最終ジャッジ日時 2025-01-20 02:54:36
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000000

void substruct(string &N){
	int n = N.size();
	for(int i=n-1;i>=0;i--){
		if(N[i]!='0'){
			N[i]--;
			break;
		}
		N[i] = '9';
	}
	while(N.size()>1&&N[0]=='0')N.erase(N.begin());
}

long long get(char c){
	vector<int> d = {6,2,5,5,4,5,6,4,7,6};
	if(isdigit(c))return d[c-'0'];
	if(c=='-')return 1;
	return 0;
}

long long get(string P){
	long long ret =0;
	rep(i,P.size()){
		ret += get(P[i]);
	}
	return ret;
}

long long get(long long P){
	return get(to_string(P));
}

int main(){
	
	string P,N;
	cin>>P>>N;
	
	if(N=="0"){
		cout<<P<<endl;
		return 0;
	}
	
	substruct(N);
	
	long long X = get(P);
	
	vector<int> A(1,X);
	
	while(true){
		if(N=="0"){
			cout<<X<<endl;
			return 0;
		}
		substruct(N);
		
		int t = get(X);
		bool f = true;
		rep(i,A.size()){
			if(t==A[i]){
				f=false;
				A.erase(A.begin(),A.begin()+i);
				break;
			}
		}
		if(!f)break;
		X = t;
		A.push_back(X);
	}
	
	int Mod = 0;
	rep(i,N.size()){
		Mod *= 10;
		Mod %= A.size();
		Mod += N[i]-'0';
		Mod %= A.size();
	}
	
	cout<<A[Mod]<<endl;
	
	
    return 0;
}
0