CodeForces 758B Blown Garland
原创
©著作权归作者所有:来自51CTO博客作者Richie_LL的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目链接:http://codeforces.com/contest/758/problem/B
题意:有n个灯泡,四种颜色和坏掉的情况,问你坏掉的里面有几个是红色,几个是黄色,几个是绿色,几个是蓝色,原本灯泡全亮的时候,每四盏灯都是不同颜色的
解析:我今天晚上一定是脑子出了点问题,一直研究各种枚举的写法,虽然觉得可以不暴力,可是脑子就是不想转弯,反正自己写起来很奇怪,看了别人代码,才发现自己姿势十分有问题
我的方法是四种颜色,一定是按照四种颜色铺上去的,所以我枚举了四种灯泡的排列,然后判断是否合法,唉:-(
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>
#include <map>
using namespace std;
const int maxn = 1e6+100;
const int inf = 0x7ffffff;
char cc[] = "RBYG";
string a;
int vis[10];
int ans[4];
int main(void)
{
cin>>a;
int len = a.length();
int cnt = 0;
for(int i=0;i<len;i++)
{
if(a[i]=='!')
cnt++;
}
for(int i=0;i<4;i++)
vis[i] = i;
do
{
int flag = 1;
memset(ans,0,sizeof(ans));
for(int i=0;i<len;i++)
{
int tt = vis[i%4];
if(a[i]=='!')
ans[tt]++;
else if(cc[tt]!=a[i])
{
flag = 0;
break;
}
}
if(flag)
break;
}while(next_permutation(vis,vis+4));
printf("%d %d %d %d\n",ans[0],ans[1],ans[2],ans[3]);
return 0;
}
正确思路
其实我的想法和正确思路应该很像了,只是今天晚上脑子就是不肯转弯,由于四种颜色出现的顺序是一定的,所以可以以4为单位判断坏掉的灯泡为哪一种灯泡,记录一下,输出就好了
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>
#include <map>
using namespace std;
const int maxn = 1e6+100;
const int inf = 0x7ffffff;
char cc[] = "RBYG";
string a;
int ans[4];
int main(void)
{
cin>>a;
int len = a.length();
int r = 0,g = 0,y = 0,b = 0;
for(int i=0;i<len;i++)
{
if(a[i]=='!')
ans[i%4]++;
if(a[i]=='R')
r = i%4;
if(a[i]=='G')
g = i%4;
if(a[i]=='B')
b = i%4;
if(a[i]=='Y')
y = i%4;
}
printf("%d %d %d %d\n",ans[r],ans[b],ans[y],ans[g]);
return 0;
}