E. Equality
题意:
将原数组中最小值视为 1,其余值视为 0,每次操作选定长度 k 且含有 1 的区间置为全 1,问题转化为最少操作次数使得数组全 1
思路:
求出最小值,从最左的0向右遍历,从这个 0 开始,往右使用若干段长度 k 的区间相扣,直到某一段长度 k 的区间内含有 1 时才停止,更新i位置
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+6;
int t;
int a[N],n,k;
int main(){
cin>>t;
while (t--) {
cin >> n >> k;
int mi = 3000000;
for (int i = 0; i < n; ++i) {
cin >> a[i];
mi = min(mi, a[i]);
}
int cnt=0;
for (int i = 0; i <n ; ++i) {
if(a[i]==mi)cnt++;
}
if(k==1&&cnt!=n){
cout<<"-1\n";
continue;
}
if(cnt==n&&k==1){
cout<<'0'<<endl;
continue;
}
k--;
int f=0,res=0;
for (int i = 0; i <n ; ++i) {
if(a[i]!=mi){
res++;
for (int j = i; j <min(n,i+k) ; ++j)
if(a[j]==mi)
f=1;
if(f) i+=min(n-i,k);
else{
i+=min(n-i,k-1);
}
}
f=0;
}
cout<<res<<endl;
}
}
F. Future Vision
题意:
一个迷宫中,寻找一个可动的剑,每一秒剑的位置可能不同,问第几秒可以最早寻到剑,若寻不到则输出NO
思路:
通过 BFS 求出从起点到每个位置的最短时间 tx,y 按时间顺序考虑剑在 i 时刻出现的位置 (xi , yi),找到其中最早的 t ,yi ≤ i 的位置即可
代码:
#include<bits/stdc++.h>
using namespace std;
#define PII pair<int,int>
char g[110][110];
int dis[110][110];
int t,n,m;
int x,y;
int dx[]={-1,0,0,1},dy[]={0,1,-1,0};
void bfs()
{
queue<PII> q;
memset(dis,-1,sizeof dis);
dis[x][y]=0;
q.push({x,y});
while(q.size())
{ int a,b;
a=q.front().first,b=q.front().second;
q.pop();
for(int i=0;i<4;i++)
{
int xx=a+dx[i],yy=b+dy[i];
if(xx>=1&&xx<=n&&yy>=1&&yy<=m&&g[xx][yy]!='#'&&dis[xx][yy]==-1)
{
dis[xx][yy]=dis[a][b]+1;
q.push({xx,yy});
}
}
}
}
int main()
{
cin>>t;
while(t--)
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++){
cin>>g[i][j];
if(g[i][j]=='H') x=i,y=j;
}
}
bfs();
int k;
cin>>k;
int sj=9999999;
for(int i=0;i<k;i++)
{
int xx,yy;
cin>>xx>>yy;
if(dis[xx][yy]!=-1&&dis[xx][yy]<=i)
{
sj=min(i,sj);
}
}
if(sj==9999999) cout<<"NO"<<endl;
else cout<<"YES "<<sj<<endl;
}
return 0;
}
K. Klee and Bomb
题意:
有n个炸弹,m个(x,y)号相互连接,引爆一个,就可以将相互连接(可以不相邻)且颜色相同的一起引爆,可对每一个炸弹的颜色进行任意更改,问更改一次后可使炸弹爆炸最多是多少?
思路:
使用并查集维护出在不改变颜色的情况下,所有同色联通块的大小 枚举所有点,列举出所有与其相连的联通快,将其中同色块的大小 累加到一起,再带上枚举点本身既是该点染为对应颜色的贡献 在所有贡献中取最大值即可
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
vector<int> ed[N];
int n,m,ans=1;
int fa[N],sz[N];
int c[N];
int num[N];
bool vis[N];
int find(int a){
if(fa[a]==a) return a;
return fa[a]=find(fa[a]);
}
void unity(int x,int y){
int r1=find(x),r2=find(y);
if(r1==r2) return;
fa[r1]=r2;
sz[r2]+=sz[r1];
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>c[i];
for(int i=1;i<=n;i++) fa[i]=i,sz[i]=1;
for(int i=1,x,y;i<=m;i++){
cin>>x>>y;
if(c[x]==c[y]) unity(x,y);
ed[x].push_back(y);
ed[y].push_back(x);
}
for(int i=1;i<=n;i++){
for(auto v:ed[i]){
if(!vis[find(v)]){
num[c[v]]+=sz[find(v)];
vis[find(v)]=true;
}
}
for(auto v:ed[i]) ans=max(ans,num[c[v]]+(c[i]!=c[v]));
for(auto v:ed[i]){
num[c[v]]=0;
vis[find(v)]=false;
}
}
printf("%d",ans);
return 0;
}