CodeForces#207Div.2 - B. Flag Day

来源：互联网发布：房子平面图设计软件编辑：程序博客网时间：2024/05/20 03:44

初始化所有人都没有跳过舞。

判断一组里面第一个出现的之前跳过舞的人，让其他两个人的颜色为这个跳过舞的人的颜色后两位（在1,2,3之间循环）。若加和完之后大于3，自减3 。若全部人之前都没有跳过舞，直接赋值1,2,3 。

AC代码：

#include <stdio.h>#include <string.h>#include <stdlib.h>#include <math.h>int color[100005];int main() {    int n, m;    while(scanf("%d%d", &n, &m)!=EOF) {        int a, b, c;        memset(color, 0, sizeof(color));        for(int i = 1; i <= m; i++) {            scanf("%d%d%d", &a, &b, &c);            if(color[a] != 0) {                color[b] = color[a] + 1;                color[c] = color[a] + 2;                if(color[b] > 3)                    color[b] -= 3;                if(color[c] > 3)                    color[c] -= 3;            }            else if(color[b] != 0) {                color[a] = color[b] + 2;                color[c] = color[b] + 1;                if(color[a] > 3)                    color[a] -= 3;                if(color[c] > 3)                    color[c] -= 3;            }            else if(color[c] != 0) {                color[a] = color[c] + 1;                color[b] = color[c] + 2;                if(color[a] > 3)                    color[a] -= 3;                if(color[b] > 3)                    color[b] -= 3;            }            else {                color[a] = 1;                color[b] = 2;                color[c] = 3;            }        }        for(int i = 1; i <= n; i++) {            if(i == 1)                printf("%d", color[i]);            else                printf(" %d", color[i]);        }        printf("\n");    }    return 0;}

简化一下代码量：

#include <cstdio>#include <cstdlib>#include <cstring>using namespace std;int c[100005];int n,m,x,y,z;int main(){    while(~scanf("%d%d",&n,&m))    {        memset(c,0,sizeof(c));        while(m--)        {            scanf("%d%d%d",&x,&y,&z);            if(c[x]!=0)            {                c[y]=c[x]%3+1;                c[z]=c[y]%3+1;            }            else if(c[y]!=0)            {                c[z]=c[y]%3+1;                c[x]=c[z]%3+1;            }            else if(c[z]!=0)            {                c[x]=c[z]%3+1;                c[y]=c[x]%3+1;            }            else            {                c[x]=1;                c[y]=2;                c[z]=3;            }        }        for(int i=1;i<=n;i++)        printf("%d ",c[i]);puts("");    }    return 0;}