題目鏈接
problem
有一個長度為\(n\)個點連成的環,每個點為黑色或白色,當一個點和與他相鄰的兩個點顏色不同時,該點的顏色就會改變,
問改變\(K\)次后每個點的顏色,
solution
發現兩個性質:
1.發現如果一個點在第一次時就不需要改變,那么他以后都不需要改變,
2.如果有個點在某次不需要改變,那么下一次他相鄰的兩個點也一定不需要改變,
所有思路就很明顯了,從不需要改變的點開始\(bfs\),得到每個點最早不需要改變的時間,然后與\(K\)取\(min\)后計算出最終顏色就行了,
code
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
typedef long long ll;
const int N = 200010;
ll read() {
ll x = 0,f = 1;char c = getchar();
while(c < '0' || c > '9') {
if(c == '-') f = -1;c = getchar();
}
while(c >= '0' && c <= '9') {
x = x * 10 + c - '0';c = getchar();
}
return x * f;
}
char s[N];
int vis[N];
queue<int>q;
int main() {
int n = read(),K = read();
scanf("%s",s);
memset(vis,-1,sizeof(vis));
for(int i = 0;i < n;++i) {
if(s[i] != s[(i - 1 + n) % n] && s[i] != s[(i + 1) % n]);
else {
q.push(i),vis[i] = 0;
}
}
while(!q.empty()) {
int u = q.front();q.pop();
if(vis[(u - 1 + n) % n] == -1) {
vis[(u - 1 + n) % n] = vis[u] + 1;q.push((u - 1 + n) % n);
}
if(vis[(u + 1) % n] == -1) {
vis[(u + 1) % n] = vis[u] + 1;q.push((u + 1) % n);
}
}
for(int i = 0;i < n;++i) {
if(vis[i] == -1 || vis[i] > K) {
if(K & 1) putchar(s[i] == 'B' ? 'W' : 'B');
else putchar(s[i]);
}
else {
if(vis[i] & 1) putchar(s[i] == 'B' ? 'W' : 'B');
else putchar(s[i]);
}
}
return 0;
}
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/96303.html
標籤:C++
