题目
给定每个位置的最长回文半径s,第i个数是s[i],
求是否存在正数数字串,满足每个位置的最长回文半径是s[i](此时要么s[i]+1越界,要么不相等)
不存在的话输出No,否则输出字典序最小的串
思路来源
dreammoon讲解
题解
魔改manacher,
manacher的半径p,一开始会根据既定结果预先确定一个值,
这个值我们保留,如果这个值已经超过了输入中给定的值,显然无解
后面扩展的时候,需要在串上满足对应位置相等,
这就是代表并查集需要在一个集合里,把它们在并查集上合并起来,
对于最长回文半径是s[i]的情况,需要钦定i-s[i]-1和i+s[i]+1字符集不相等,
这个互斥关系也预先记录下来
先合并并查集,最后合并成若干个等价类,
然后先判互斥关系,如果互斥关系的两个点已经在同一集合了,显然无解
否则,从前到后贪心,填满足互斥关系条件下的字典序最小即可
代码
// Problem: G - Palindrome Construction
// Contest: AtCoder - AtCoder Beginner Contest 349
// URL: https://atcoder.jp/contests/abc349/tasks/abc349_g
// Memory Limit: 1024 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<ll,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define scll(a) scanf("%lld",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=2e5+10;
int n,mx,pos,p[N],a[N],par[N],ans[N];
vector<P>no;
vector<int>ban[N];
int find(int x){
return par[x]==x?x:par[x]=find(par[x]);
}
void NO(){
puts("No");
exit(0);
}
void un(int x,int y){
x=find(x),y=find(y);
if(x==y)return;
par[y]=x;
}
int main(){
sci(n);
rep(i,1,n){
sci(a[i]);
a[i]++;
par[i]=i;
}
rep(i,1,n){
if(i<mx){
p[i]=min(p[2*pos-i],mx-i);
}
else{
p[i]=1;
}
if(p[i]>a[i])NO();
while(p[i]<a[i]){
un(i-p[i],i+p[i]);
p[i]++;
}
if(i+p[i]>mx){
pos=i;
mx=i+p[i];
}
if(i-a[i]>=1 && i+a[i]<=n){
no.pb(P(i-a[i],i+a[i]));
}
}
for(auto &x:no){
int u=x.fi,v=x.se;
u=find(u),v=find(v);
if(u==v)NO();
ban[u].pb(v);
ban[v].pb(u);
}
puts("Yes");
rep(i,1,n){
int f=find(i);
if(!ans[f]){
set<int>q;
for(auto &v:ban[f]){
q.insert(ans[v]);
}
ans[f]=1;
while(q.count(ans[f]))ans[f]++;
}
printf("%d%c",ans[f]," \n"[i==n]);
}
return 0;
}