在做七夕祭之前先做两个基础题型帮助理解:均分纸牌->糖果传递->七夕祭(注:一定要按顺序看!)
参考自链接: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;
}