码迷,mamicode.com
首页 > 其他好文 > 详细

USACO 1.1 Broken Necklace(USACO官方)

时间:2014-11-19 01:58:42      阅读:186      评论:0      收藏:0      [点我收藏+]

标签:beads   usaco   broken necklace   

本题我不会写,囧,看了官方的代码,算法复杂度为O(n^2),学习下:

/*
  ID:twd30651
  PROG:beads
  LANG:C++
*/
#include<iostream>
#include<fstream>
#include<string.h>
using namespace std;
int N;
char s[400];
int len;
int mod(int n,int m)
{
    while(n<0)n+=m;
    return n%m;//值得学习,跑一圈又回来
}
int nbreak(int p,int dir)
{
    int i;
    int n;
    if(dir>0)
        i=p;
    else
        i=mod(p-1,len);
    char color='w';//用color来保存颜色
    for(n=0;n<len;i=mod(i+dir,len))
    {
        if(color=='w'&&s[i]!='w')
            color=s[i];//处理收集珠子时颜色的变换
        else if(color!=s[i]&&s[i]!='w')
            break;
        n++;
    }
    return n;
}
int main(int argc,char *argv[])
{
    freopen("beads.in","r",stdin);
    freopen("beads.out","w",stdout);
    int i=0;
    scanf("%d",&N);
    scanf("%s",s);
    len=strlen(s);
    int n;
    int m=0;
    for(i=0;i<len;++i)
    {
        n=nbreak(i , 1)+nbreak(i,-1);//兵分两路
        if(n>m)
            m=n;
    }
    if(m>len)//考虑转了一圈,又回来了,类似于"rrrrrr"这样的字符串
        m=len;
    printf("%d\n",m);
    return 0;
}

下面一种方法是dp,时间复杂度为O(n)

/*
  ID:twd30651
  PROG:beads
  LANG:C++
*/
#include<iostream>
#include<fstream>
#include<stdlib.h>
#include<string.h>
using namespace std;
/*
  copy:
  用数组bl,br,rl,rr分别记录在项链i处向左向右收集的蓝色红色珠子数。
  项链是环形的,但我们只要把两个同样的项链放在一块,就把它转换成线性的了。
  我们只要求出bl,br,rl,rr,那么结果就是max(max(bl[i],rl[i])+max(br[i+1],rr[i+1])) (0<=i<=2*n-1)。
  我们以求bl,rl为例:
  初始时bl[0]=rl[0]=0
  我们从左往右计算
  如果necklace[i]='r',rl[i]=rl[i-1]+1,bl[i]=0;
  如果necklace[i]='b', bl[i]=bl[i-1]+1,rl[i]=0;
  如果necklace[i]='w',bl[i]=bl[i-1]+1,rl[i]=rl[i-1]+1。
  同理可以求出br,rr。
*/
typedef struct node
{
    int lr;//向左最多红色珠子
    int lb;//向左最多蓝色珠子
    int rr;//向右最多红色珠子
    int rb;//向右最多蓝色珠子
}node;
node necklace[800];
char s[800];
char tmp[400];
int N;
int main(int argc,char *argv[])
{
    freopen("beads.in","r",stdin);
    freopen("beads.out","w",stdout);
    int i=0;
    scanf("%d",&N);
    scanf("%s",s);
    strcpy(tmp, s);
    //也是珠子加上它自己的一个副本
    strcat(s, tmp);
    necklace[0].lr=0;
    necklace[0].lb=0;
    for(int i=1;i<=2*N;++i)
    {
        if(s[i-1]=='r')
            necklace[i].lr=necklace[i-1].lr+1,necklace[i].lb=0;
        else if(s[i-1]=='b')
            necklace[i].lb=necklace[i-1].lb+1,necklace[i].lr=0;
        else
            necklace[i].lb=necklace[i-1].lb+1,
                necklace[i].lr=necklace[i-1].lr+1;
    }
    necklace[2*N].rr=0;
    necklace[2*N].rb=0;
    for(int i=2*N-1;i>=0;--i)
    {
        if(s[i]=='r')
            necklace[i].rr=necklace[i+1].rr+1,necklace[i].rb=0;
        else if(s[i]=='b')
            necklace[i].rb=necklace[i+1].rb+1,necklace[i].rr=0;
        else
            necklace[i].rb=necklace[i+1].rb+1,
                necklace[i].rr=necklace[i+1].rr+1;
    }
    int m=0;
    for(int i=0;i<2*N;++i)
        m=max(m,max(necklace[i].lr,necklace[i].lb)
              +
              max(necklace[i].rr, necklace[i].rb));
    if(m>N)
        m=N;
    printf("%d\n",m);
    return 0;
}


USACO 1.1 Broken Necklace(USACO官方)

标签:beads   usaco   broken necklace   

原文地址:http://blog.csdn.net/wdkirchhoff/article/details/41246395

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!