Codeforces Round #398 (Div. 2)

Codeforces Round #398 (Div. 2)

A.Snacktower

模拟

我和官方题解的命名神相似...$has$

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=1e5+;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-; c=getchar();}
    while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
    return x*f;
}
int n,has[N];
int main(){
    //freopen("in","r",stdin);
    n=read();
    int now=n;
    for(int i=;i<=n;i++){
        has[read()]=;
        while(has[now]) printf("%d ",now),now--;
        puts("");
    }
}

---

B.The Queue

比赛时花了1h然后弃疗

感觉好恶心...

---

C.Garland

题意：一颗二叉树切断两条边分成三部分使得三部分权值和相等

感觉比B简单(好写)多了，可以发现每部分权值和就是总权值的$\frac{1}{3}$

然后我用了两个树形DP，$d[i]$表示$i$子树权值和，$f[i]$表示$i$子树里有几个权值和等于$\frac{1}{3}$的

分类讨论选的两棵子树是否互相包含，比赛时各种$debug$，最后$5min$通过了...感动

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=1e6+;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-; c=getchar();}
    while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
    return x*f;
}
int n,d[N],fa,w[N],root;
struct edge{
    int v,ne;
}e[N<<];
int h[N],cnt=;
inline void ins(int u,int v){
    cnt++;
    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;
}
void dp(int u){
    d[u]=w[u];
    for(int i=h[u];i;i=e[i].ne)
        dp(e[i].v),d[u]+=d[e[i].v];
}
int f[N],val;
void dp2(int u){
    for(int i=h[u];i;i=e[i].ne){
        int v=e[i].v;
        dp2(v);
        if(f[v]==) f[u]+=d[v]==val;
        else f[u]+=f[v];
    }
}
int a[N],p;
int main(){
    //freopen("in","r",stdin);
    n=read();
    for(int i=;i<=n;i++){
        fa=read();w[i]=read();
        if(fa==) root=i;
        else ins(fa,i);
    }
    dp(root);
    if(d[root]%!=) puts("-1");
    else{
        val=d[root]/;
        dp2(root);
        //for(int i=1;i<=n;i++) printf("%d %d %d\n",i,d[i],f[i]);
        if(f[root]==) puts("-1");
        else if(f[root]>=){
            for(int i=;i<=n;i++) if(i!=root&&d[i]==val&&f[i]==) a[++p]=i;
            printf("%d %d",a[],a[]);
        }else{
            int flag=,a,b;
            for(int i=;i<=n;i++) if(i!=root&&d[root]-d[i]==val&&f[i]==){
                flag=;a=i;break;
            }
            for(int i=;i<=n;i++) if(i!=root&&d[i]==val&&f[i]==) {b=i;break;}
            if(flag) printf("%d %d",a,b);
            else puts("-1");
        }
    }
}

---

D.Cartons of milk

赛后补题.....

一开始贪心乱搞失败，然后发现直接二分答案就好了.....然而是$O(n log^{2}n)$.....

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=2e6+;
inline int read(){
    char c=getchar();int x=;
    while(c<''||c>''){c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x;
}
int n,m,k,a[N];
struct data{
    int id,b;
    bool operator <(const data &r) const{
        return b>r.b;
    }
}b[N];
int p,c[N];
bool check(int x){//printf("check %x\n",x);
    ll now=;
    p=;
    for(int i=;i<=n;i++) c[++p]=a[i];
    for(int i=;i<=x;i++) c[++p]=b[i].b;
    sort(c+,c++p);
    //for(int i=1;i<=p;i++) printf("%d ",c[i]);puts("");
    for(int i=;i<=p;i++){
        now++;//printf("now %d %d %d\n",i,c[i],now);
        if(now>(ll)k*c[i]) return false;
    }
    return true;
}
int main(){
    //freopen("in","r",stdin);
    n=read();m=read();k=read();
    for(int i=;i<=n;i++) a[i]=read()+;
    for(int i=;i<=m;i++) b[i].b=read()+,b[i].id=i;
    sort(a+,a++n);
    sort(b+,b++m);
    int l=,r=m,ans=-;
    while(l<=r){
        int mid=(l+r)>>;
        if(check(mid)) ans=mid,l=mid+;
        else r=mid-;
    }
    printf("%d\n",ans);
    if(ans!=-){
        memset(c,,sizeof(c));
        for(int i=;i<=ans;i++) c[b[i].id]=;
        for(int i=;i<=m;i++) if(c[i]) printf("%d ",i);
    }
}
 

看了一个别人的代码发现贪心也能做并且很快

$f[i]$记录时间i选了几个，贪心选择放在最晚到期，然后从后往前$>k$的给前面就行了，$O(n)$，时间只有二分答案的$\frac{1}{3}$

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=2e6+,M=1e7+;
inline int read(){
    char c=getchar();int x=;
    while(c<''||c>''){c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x;
}
int n,m,k,a[N];
struct data{
    int id,b;
    bool operator <(const data &r) const{
        return b>r.b;
    }
}b[N];
int f[M],mx;
bool vis[N];
int main(){
    //freopen("in","r",stdin);
    n=read();m=read();k=read();
    for(int i=;i<=n;i++) a[i]=read()+,f[a[i]]++,mx=max(mx,a[i]);
    for(int i=;i<=m;i++) b[i].b=read()+,b[i].id=i,mx=max(mx,b[i].b);
    sort(a+,a++n);
    for(int i=mx;i>=;i--) if(f[i]>k) f[i-]+=f[i]-k,f[i]=k;
    if(f[]) puts("-1");
    else{
        sort(b+,b++m);
        for(int i=;i<=m;i++) f[b[i].b]++;
        for(int i=mx;i>=;i--) if(f[i]>k) f[i-]+=f[i]-k,f[i]=k;
        int p=m-f[];
        printf("%d\n",p);
        for(int i=;i<=p;i++) vis[b[i].id]=;
        for(int i=;i<=m;i++) if(vis[i]) printf("%d ",i);
    }
}