Codeforces 11.27 B

B. Restoring IPv6

状态不好，看来好久才看出题意，字符处理始终是我的弱项，有空联系下。记录一下。

#include <iostream>
#include <cassert>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int MM = 110010;
#define fuck puts("fuck")
#define VI vector<int>
typedef long long int64;
#define mod 1000000007

char ch[MM];

void solve() {
    int i,j,k,len;
    scanf("%s",ch);
    len=strlen(ch);
    if(len==2&&ch[0]==':'&&ch[1]==':') {
        for(i=0;i<7;i++) printf("0000:"); puts("0000");
    }
    else {
        int pos[MM]={0},all=0;
        for(i=0;ch[i];i++) if(ch[i]==':') all++;
        if(ch[len-1]==':'&&ch[len-2]==':') pos[len-2]=1;
        for(i=0;i<len-2;i++) {
            if(ch[i]==':'&&ch[i+1]==':') {
                if(ch[i-1]=='0'&&ch[i+2]=='0') pos[i]=2;
                else pos[i]=3;
            }
        }
//        for(i=0;i<len;i++) printf("%d %d\n",i,pos[i]);
        int zero=9-all;
        for(i=0;i<len;i++) {
            for(j=i;j<=i+4;j++) if(ch[j]==':'||j==(len)) break;
            for(k=0;k<(-j+i+4);k++) printf("0");
            for(k=i;k<j;k++) printf("%c",ch[k]);
            if(ch[j]==':') {
            if(pos[j]==0)  printf(":"),i=j;
            else if(pos[j]==1) {
               for(k=0;k<zero;k++)printf(":0000");i=j+1;
            }
//            else if(pos[j]==2) printf(":0000:"),i=j+1;
            else {
                for(i=1;i<zero;i++) printf(":0000");
                printf(":");
                i=j+1;
            }
            }
            else i=j+1;
        }
        puts("");
    }
}

int main() {
    int ca; scanf("%d",&ca);
    while(ca--) solve();
    return 0;
}