이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define pll pair<long long, long long>
#define x first
#define y second
using namespace std;
const int N = 2e6 + 5;
pll add(pll a, pll b) { return pll(a.x + b.x, a.y + b.y); }
pll sub(pll a, pll b) { return pll(a.x - b.x, a.y - b.y); }
pll mul(pll a, pll b) { return pll(a.x * b.x, a.y * b.y); }
pll prm = pll(69LL, 420LL);
pll pwr[N];
int n, mid;
pll dp[N];
char A[N];
set<pll> ans;
int main() {
pwr[0] = pll(1LL, 1LL);
for(int i = 1; i < N; i++) pwr[i] = mul(pwr[i - 1], prm);
scanf("%d", &n);
if(!(n & 1)) return !printf("NOT POSSIBLE");
mid = n >> 1;
scanf("%s", A + 1);
for(int i = 1; i <= n; i++) dp[i] = add(mul(dp[i - 1], prm), pll(A[i], A[i]));
for(int i = 1; i <= n; i++) {
pll lhs, rhs;
if(i <= mid) {
lhs = add(mul(dp[i - 1], pwr[mid - i + 1]), sub(dp[mid + 1], mul(dp[i], pwr[mid - i + 1])));
rhs = sub(dp[n], mul(dp[mid + 1], pwr[mid]));
if(lhs == rhs) ans.emplace(mid + 2, n);
} else {
lhs = dp[mid];
rhs = add(mul(sub(dp[i - 1], mul(dp[mid], pwr[i - mid - 1])), pwr[n - i]), sub(dp[n], mul(dp[i], pwr[n - i])));
if(lhs == rhs) ans.emplace(1, mid);
}
}
if(!ans.size()) return !printf("NOT IMPOSSIBLE");
else if(ans.size() == 1) for(int i = ans.begin()->x; i <= ans.begin()->y; i++) printf("%c", A[i]);
else if(ans.size() > 1) return !printf("NOT UNIQUE");
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
friends.cpp: In function 'int main()':
friends.cpp:27:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &n);
~~~~~^~~~~~~~~~
friends.cpp:30:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%s", A + 1);
~~~~~^~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |