【Codeforces 755 F. PolandBall and Gifts】多重背包 + 二进制优化

Codeforces 755 F

• n个人，相互送礼物
• 第i个人会送给第a[i]个人礼物
• a为1~n的一个排列
• 一个人能收到礼物当且仅当他送出了礼物且有人送给他礼物
• 现知道必定有K个人会忘记带礼物
• 求最少及最多多少人不能收到礼物
• 1 ≤ n, K ≤ 1e6

这题首先要知道 1 - n 全排列 那么肯定是一个或几个环构成
现在有两个人 s 和 t 如果 s 没带礼物 那么 s 肯定影响自己 和 t
但是送 s 礼物的人却能送礼物
也就是说一个人最多影响两个人

首先看最多不能收到礼物 我们贪心的找到偶环 和奇环 k个人数的偶环只需要k/2个人就可以破坏
但是k个人的奇环却需要 k/2+1 的人数去破坏
所以我们考虑+1的时候 能不能放下一个奇环里面去破坏两个人 而不是拿来破坏这个奇环 因为2贡献大于 1
所以最多的时候贪心策略是 破坏偶环 > 破坏奇环中偶数部分 > 实在不行把奇环中贡献为1的也破坏了

那么看最少不能收到礼物 其实就是说几个环能不能正好凑成 k 如果不能 答案就是 k + 1 (因为假如环不能破坏 你必须放另一个环中破坏两个人
那么同长度的环可能有多个 所以是一个多重背包 二进制优化一下 就过了

/*
    if you can't see the repay
    Why not just work step by step
    rubbish is relaxed
    to ljq
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <algorithm>
using namespace std;

#define dbg(x) cout<<#x<<" = "<< (x)<< endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))

typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod =  (int)1e9+7;

ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);

/*namespace sgt
{
    #define mid ((l+r)>>1)

    #undef mid
}*/
const int MAX_N = 1000025;
int cnt,belong[MAX_N],Size[MAX_N],arr[MAX_N],v[MAX_N],w[MAX_N];
bool dp[MAX_N];
void dfs(int x)
{
    belong[x] = cnt;
    if(belong[arr[x]]!=-1) return;
    dfs(arr[x]);
    Size[cnt]++;
}
vector<int> vt,vt_;
map<int ,int > mp;
set<int > st;
bool cmp(int a,int b)
{
    return a >b;
}

int main()
{
    //ios::sync_with_stdio(false);
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    int n,m,k,bc = 0;
    cnt = 0;
    scanf("%d%d",&n,&m);
    k=  m;
    for(int i = 1;i<=n;++i) scanf("%d",&arr[i]);
    memset(belong,-1,sizeof(belong));
    memset(dp,false,sizeof(dp));
    dp[0] = true;
    for(int i = 1;i<=n;++i)
    {
        if(belong[i]==-1)
        {
            cnt++;
            dfs(i);
            Size[cnt]++;
        }
    }
    for(int i = 1;i<=cnt;++i)
    {
        st.insert(Size[i]);
        mp[Size[i]]++;
        if(Size[i]%2==0) vt.push_back(Size[i]);
        else vt_.push_back(Size[i]);
    }
    sort(vt.begin(),vt.end(),cmp);
    sort(vt_.begin(),vt_.end(),cmp);
    int ans = 0,ans_ = 0,sz = vt.size(),sz_ = vt_.size();
        for(int i = 0;i<sz;++i)
        {
            if(m>=vt[i]/2)
            {
                m-=vt[i]/2;
                ans_+=vt[i];
            }
            else if(m==0) break;
            else
            {
                ans_+=m*2;
                m = 0;
                break;
            }
        }
        for(int i = 0;i<sz_;++i)
        {
            if(m>=vt_[i]/2)
            {
                m-=vt_[i]/2;
                ans_+=vt_[i]-1;
                bc++;
            }
            else if(m==0) break;
            else
            {
                ans_+=m*2;
                m = 0;
                break;
            }
    }
    ans_+=min(m,bc);
    m = k;
    int now = 0;
    set<int>::iterator it ;
    for(it = st.begin();it!=st.end();++it)
    {
        v[++now] = *it;
        w[now] = mp[*it];
    }
    for(int i = 1;i<=now;++i)
    {
        if(v[i]*w[i]>=m)
        {
            for(int j = v[i];j<=m;++j)
                dp[j] = dp[j-v[i]]|dp[j];
        }
        else
        {
            for(int k = 0;k<=30;k++)
            {
                ll tmp = (1ll<<k);
                if(w[i]>=tmp)
                {
                    w[i]-=tmp;
                    for(int j = m;j>=tmp*v[i];--j)
                        dp[j] = dp[j-tmp*v[i]] |dp[j];
                    if(w[i]==0) break;
                }
                else
                {
                    for(int j = m;j>=w[i]*v[i];--j)
                        dp[j] = dp[j-w[i]*v[i]] |dp[j];
                    w[i] = 0;
                    break;
                }
            }
        }
    }
    if(dp[k]) ans = k;
    else ans = k + 1;
    printf("%d %d\n",ans,ans_);
    //fclose(stdin);
    //fclose(stdout);
    //cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}