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

【Codeforces】879D. Teams Formation 思维+模拟

时间:2017-11-16 18:38:28      阅读:168      评论:0      收藏:0      [点我收藏+]

标签:scanf   序列   div   开始   source   c++   return   多少   format   

题意

给定$n$个数,重复拼接$m$次,相邻$k$个重复的可消除,问最后序列中有多少个数


首先可以发现当$k>=n$时,如果要使$n$个数可以被消除,那么$n$个数必须一样,否则$n$个数不能被消除

当$k<n$时,首先对序列能够消除的消除,用栈记录每个数和这个数的相同的相邻个数$f$,将序列压缩,考虑一次拼接,记$L$为第二个序列的开始元素,$R$为第一个序列的最后一个元素,如果$L!=R$那么所有序列将不能压缩,如果$L==R,f_L+f_R!=k$,那么这两个序列拼接之后会合并成一个不可被消去的元素,减去$m-1$次的贡献得到答案,如果$L==R,f_L+f_R==k$,那么这两个可以完全被消除,那么按照相同方式计算$L+1,R-1$对应的贡献,最后如果整个序列都能合并到最后,如果$n$为奇数,表示最后会剩下一个元素,计算这个元素对应答案,如果$n$为偶数,那么答案为$0$

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int st[100100], f[100100];
int n, k, m, a, flag = 0, top = 1;
LL ans = 0;
int main() {
    scanf("%d%d%d", &n, &k, &m);
    st[0] = 0;
    for(int i = 1; i <= n; ++i) {
        scanf("%d", &a);
        if(st[top - 1] != a) st[top] = a, f[top++] = 1;
        else  {
            if(f[top - 1] + 1 < k) f[top - 1] += 1;
            else if(f[top - 1] + 1 == k)  top--;
        }
    }
    if(k >= n) {
        for(int i = 1; i < top - 1; ++i) if(st[i] != st[i + 1]) flag = 1;
        if(!flag) cout << 1LL * n * m % k << endl; else cout << 1LL * n * m << endl;
        return 0;
    }
    for(int i = 1; i < top; ++i) ans += f[i];
    if(m == 1 || ans == 0) {cout << ans << endl; return 0;}
    ans = 1LL * m * ans;
    int L = 1, R = top - 1;
    while(L < R) {
        if(st[L] != st[R]) {cout << ans << endl; return 0;}
        if(f[L] + f[R] == k) {L++; R--;ans -= 1LL * (m - 1) * k; continue;}
        if(f[L] + f[R] > k) ans -= 1LL * (m - 1) * k;
        cout << ans << endl; return 0;
    }
    if(L > R) {cout << 0 << endl; return 0;}
    if(L == R) {
        LL x = 1LL * f[L] * m;
        if(x % k == 0) {cout << 0 << endl;}
        else {cout << ans - x / (LL)k * (LL)k << endl;}
    }
    return 0;
}

【Codeforces】879D. Teams Formation 思维+模拟

标签:scanf   序列   div   开始   source   c++   return   多少   format   

原文地址:http://www.cnblogs.com/ogiso-setsuna/p/7845417.html

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