【ACwing】一、快排延伸：1536. 均分纸牌+122. 糖果传递+105. 七夕祭

在做七夕祭之前先做两个基础题型帮助理解：均分纸牌->糖果传递->七夕祭（注：一定要按顺序看！）

参考自链接：https://blog.csdn.net/qq_38930523/article/details/89888915

1536. 均分纸牌(NOIP2002提高组)

输入样例：

4
9 8 17 6

输出样例：

3

# include <iostream>
using namespace std;
const int N = 10010;
int a[N];
int main()
{
    int n,avg=0,ans=0;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        avg+=a[i];
    }
    avg/=n;
    for(int i=0;i<n;i++)
    {
        if(a[i]!=avg)
        {
            a[i+1]+=a[i]-avg;
            ans++;
        }
    }
    cout<<ans<<endl;
}

122. 糖果传递(环形均分纸牌)

参考自：https://www.acwing.com/solution/content/955/
这道题是对均分纸牌的一个扩展：最后一个堆可以和第一个堆交换牌，这样所有的堆之间都形成了相邻关系，即形成了一个环，问此时的最小交换代价（与交换次数不同）

# include <iostream>
# include <algorithm>
# define LL long long 
using namespace std;


const int N = 1e6+10;

int a[N],s[N];

int main()
{
    int n;
    LL ans=0,avg=0;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        avg+=a[i];
    }
    avg/=n;//平均值

    for(int i=2;i<=n;i++)
        s[i]=s[i-1]+a[i]-avg;
    
    sort(s+1,s+n+1);
    int x1=s[(n+1)>>1];//s的中位数
    
    for(int i=1;i<=n;i++)
        ans+=abs(x1-s[i]);
    printf("%lld",ans);
}

快排延伸——105. 七夕祭

输入样例：

2 3 4
1 3
2 1
2 2
2 3

输出样例：

row 1

由于列之间的交换并不会影响行的结果，相似的，行之间的交换并不会影响列的结果；所以相当于做了两次糖果交换问题。

# include <iostream>
# include <algorithm>

using namespace std;

typedef long long ll;

const int N = 100010;

int row[N],col[N];
int s[N],c[N];
ll work(int n,int a[])
{
    for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];//求原序列的前缀和
    if(s[n]%n) return -1; //不能整除，肯定不能均分
    ll avg=s[n]/n;
    c[1]=0;
    for(int i=2;i<=n;i++) c[i]=s[i-1]-(i-1)*avg;
    sort(c+1,c+1+n);
    ll res=0;
    //x1=c[(n+1)/2]
    for(int i=1;i<=n;i++) res+=abs(c[i]-c[(n+1)/2]);
    return res;
}
int main()
{
    int n,m,q;
    cin>>n>>m>>q;
    while(q--)
    {
        int x,y;
        cin>>x>>y;
        row[x]++;
        col[y]++;
    }
    //r c的值表示行或列是否能达到avg
    ll r=work(n,row);
    ll c=work(m,col);
    if(r!=-1&&c!=-1) printf("both %lld\n",r+c);
    else if(r!=-1) printf("row %lld\n",r);
    else if(c!=-1) printf("column %lld\n",c);
    else  printf("impossible\n");
    return 0;
}