洛谷 P2512 糖果传递
这有一篇博客分析的很好,就转载一下吧。
http://hzwer.com/2656.html
程序代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
int n,a[1000001],c[1000001],ave;
long long sum,ans;
int main(){
scanf("%d",&n);
for(register int i=1;i<=n;++i) scanf("%d",&a[i]),sum+=a[i];
ave=sum/n;
for(int i=1;i<=n;++i) c[i]=c[i-1]+a[i]-ave;
sort(1+c,1+c+n);
int mid=c[(n>>1)+1];
for(int i=1;i<=n;++i) ans+=abs(c[i]-mid);
printf("%lld",ans);
return 0;
}
tyvj 1924 七夕祭
跟上面那一题差不多,只不过是二维的,但我们可以将二维转化成一维。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<map>
#include<cstdlib>
#define pa pair<int,int>
#define inf 1000000000
#define linf 9000000000000000000LL
#define ll long long
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,m,T;
int x[100005],y[100005];
int a[100005],b[100005];
ll s1[100005],s2[100005],ans;
void solve()
{
ll t;
if(T%n==0)
{
for(int i=1;i<=n;i++)
s1[i]=s1[i-1]+x[i];
sort(s1+1,s1+n+1);
t=s1[(n+1)>>1];
for(int i=1;i<=n;i++)
ans+=abs(t-s1[i]);
}
if(T%m==0)
{
for(int i=1;i<=m;i++)
s2[i]=s2[i-1]+y[i];
sort(s2+1,s2+m+1);
t=s2[(m+1)>>1];
for(int i=1;i<=m;i++)
ans+=abs(t-s2[i]);
}
}
int main()
{
freopen("romantic6.in","r",stdin);
freopen("romantic.out","w",stdout);
n=read();m=read();T=read();
for(int i=1;i<=T;i++)
{
int a=read(),b=read();
x[a]++;y[b]++;//分别转化成行,列的一维即可
}
if(T%n&&T%m)
{
puts("impossible");
return 0;
}
for(int i=1;i<=n;i++)x[i]-=T/n;
for(int i=1;i<=m;i++)y[i]-=T/m;
solve();
if(T%n==0&&T%m==0)printf("both ");
//两个都行的情况就是将上面两个所得到的之都相加即可
else if(T%n==0)printf("row ");
else printf("column ");
printf("%lld\n",ans);
return 0;
}
小结:反正我当时考试的时候想到了转一维,但是不会上面那道题,所以就按照洛谷均分纸牌(为加强版的做法做了),对的点很少(这都过了?数据太水了吧)。