[POI2014]Beads

题目大意:
  有$n(nleq10^6)$种颜色,第$i$种颜色有$c_i(sum c_ileq10^6)$个,指定第一个颜色为$a$,最后一个颜色为$b$,问对于一个长度为$m=sum c_i$的序列,是否能构造出一个染色方案满足相邻的颜色不相同。如果能,试构造出一种方案。

思路:
  贪心。如果序列中有多个元素,开头结尾颜色相同而这种颜色只有一个,则显然不存在合法方案。每次选取当前数量最多的不同于前一个颜色的颜色,如果有一样多的,就尽量取和最后一个位置的颜色相同的颜色。如果没有可以取的元素或者倒数第二个元素颜色只能选和最后一个元素相同的颜色就说明不存在合法方案。洛谷和BZOJ随便AC,然而SZKOpuł上怎么卡常都是T。

正常代码:

 1 #include<queue>
 2 #include<cstdio>
 3 #include<cctype>
 4 #include<algorithm>
 5 inline int getint() {
 6     register char ch;
 7     while(!isdigit(ch=getchar()));
 8     register int x=ch^'0';
 9     while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
10     return x;
11 }
12 const int N=1e6+1;
13 int a[N],c[N];
14 std::priority_queue<std::pair<int,int> > q;
15 int main() {
16     const int n=getint();
17     a[1]=getint(),a[2]=getint();
18     int m=0;
19     for(register int i=1;i<=n;i++) {
20         m+=c[i]=getint();
21     }
22     std::swap(a[2],a[m]);
23     c[a[1]]--;
24     if(m!=1) c[a[m]]--;
25     if(c[a[1]]<0) {
26         putchar('0');
27         return 0;
28     }
29     for(register int i=1;i<=n;i++) {
30         if(c[i]) q.push(std::make_pair(c[i],i==a[m]?i+n:i));
31     }
32     for(register int i=2;i<m;i++) {
33         int x=q.top().second%n;q.pop();
34         if(!x) x+=n;
35         if(x==a[i-1]) {
36             if(q.empty()) {
37                 putchar('0');
38                 return 0;
39             }
40             int y=q.top().second%n;q.pop();
41             if(!y) y+=n;
42             q.push(std::make_pair(c[x],x==a[m]?x+n:x));
43             a[i]=y;
44             if(--c[y]) q.push(std::make_pair(c[y],y==a[m]?y+n:y));
45         } else {
46             a[i]=x;
47             if(--c[x]) q.push(std::make_pair(c[x],x==a[m]?x+n:x));
48         }
49     }
50     if(a[m-1]==a[m]) {
51         putchar('0');
52         return 0;
53     }
54     for(register int i=1;i<=m;i++) {
55         printf("%d%c",a[i]," 
"[i==m]);
56     }
57     return 0;
58 }
View Code

卡常代码:

  1 #pragma GCC optimize(3)
  2 #pragma GCC optimize("unroll-loops")
  3 #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
  4 #include<cstdio>
  5 #include<climits>
  6 #include<algorithm>
  7 #include<sys/mman.h>
  8 #include<sys/stat.h>
  9 typedef long long int64;
 10 class BufferedInputStream {
 11     private:
 12         char *buf,*p;
 13         int size;
 14     public:
 15         BufferedInputStream() {
 16             register int fd=fileno(stdin);
 17             struct stat sb;
 18             fstat(fd,&sb);
 19             size=sb.st_size;
 20             p=buf=reinterpret_cast<char*>(mmap(0,size,PROT_READ,MAP_PRIVATE,fileno(stdin),0));
 21         }
 22         char getchar() {
 23             return (p==buf+size||*p==EOF)?EOF:*p++;
 24         }
 25 };
 26 BufferedInputStream in;
 27 inline int getint() {
 28     register char ch;
 29     while(!__builtin_isdigit(ch=in.getchar()));
 30     register int x=ch^'0';
 31     while(__builtin_isdigit(ch=in.getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
 32     return x;
 33 }
 34 const int N=1e6+1;
 35 int a[N],c[N];
 36 class priority_queue {
 37     #define _left <<1
 38     #define _right <<1|1
 39     #define _par >>1
 40     private:
 41         int size;
 42         int64 val[N];
 43     public:
 44         priority_queue() {
 45             val[0]=LLONG_MAX;
 46         }
 47         void push(const int64 &x) {
 48             register int p=++size;
 49             for(;val[p _par]<x;p=p _par) {
 50                 val[p]=val[p _par];
 51             }
 52             val[p]=x;
 53         }
 54         int64 top() const {
 55             return val[1];
 56         }
 57         void pop() {
 58             register int p=1;
 59             const int64 x=val[size--];
 60             for(;p _left<=size&&(x<val[p _left]||x<val[p _right]);p=val[p _left]>val[p _right]?p _left:p _right) {
 61                 val[p]=std::max(val[p _left],val[p _right]);
 62             }
 63             val[p]=x;
 64         }
 65         bool empty() const {
 66             return !size;
 67         }
 68     #undef _left
 69     #undef _right
 70     #undef _par
 71 };
 72 inline int64 make_pair(const int &x,const int &y) {
 73     return (int64)x<<32|y;
 74 }
 75 inline int second(const int &x) {
 76     return x&0x7fffffff;
 77 }
 78 priority_queue q;
 79 int main() {
 80     const int n=getint();
 81     a[1]=getint(),a[2]=getint();
 82     int m=0;
 83     for(register int i=1;i<=n;i++) {
 84         m+=c[i]=getint();
 85     }
 86     std::swap(a[2],a[m]);
 87     c[a[1]]--;
 88     if(__builtin_expect(m!=1,1)) c[a[m]]--;
 89     if(__builtin_expect(c[a[1]]<0,0)) {
 90         __builtin_puts("0");
 91         __builtin_exit(0);
 92     }
 93     for(register int i=1;i<=n;i++) {
 94         if(__builtin_expect(c[i],1)) q.push(make_pair(c[i],i==a[m]?i+n:i));
 95     }
 96     for(register int i=2;i<m;i++) {
 97         int x=second(q.top());q.pop();
 98         if(__builtin_expect(x>n,0)) x-=n;
 99         if(__builtin_expect(x==a[i-1],0)) {
100             if(__builtin_expect(q.empty(),0)) {
101                 __builtin_puts("0");
102                 __builtin_exit(0);
103             }
104             int y=second(q.top());q.pop();
105             if(__builtin_expect(y>n,0)) y-=n;
106             q.push(make_pair(c[x],x==a[m]?x+n:x));
107             if(__builtin_expect(--c[a[i]=y],1)) q.push(make_pair(c[y],y==a[m]?y+n:y));
108         } else {
109             if(__builtin_expect(--c[a[i]=x],1)) q.push(make_pair(c[x],x==a[m]?x+n:x));
110         }
111     }
112     if(__builtin_expect(a[m-1]==a[m],0)) {
113         __builtin_puts("0");
114         __builtin_exit(0);
115     }
116     for(register int i=1;i<=m;i++) {
117         __builtin_printf("%d%c",a[i]," 
"[i==m]);
118     }
119     __builtin_exit(0);
120 }
View Code
原文地址:https://www.cnblogs.com/skylee03/p/8659946.html