UVA11538 Chess Queen

最新推荐文章于 2019-02-15 06:16:35 发布

*ACoder*

最新推荐文章于 2019-02-15 06:16:35 发布

阅读量156

点赞数

分类专栏： # 排列组合

本文链接：https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/82667010

版权

排列组合专栏收录该内容

54 篇文章 0 订阅

订阅专栏

链接

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=2533

题解1

这题就是不停地推式子，第一次我推出的是这个东西：
这里写图片描述
这个东西是 $O(TNM)$ 的，继续优化
把 $j$ 去掉之后，得到这个东西：

这个是 $O(TN)$ 的，抱着侥幸心理交了一发，结果 $TLE$ 了
然后继续优化，就得到了这个 $O(T)$ 的算法，其中 $f(x)=\sum_{i=1}^xi,g(x)=\sum_{i=1}^xi^2$
这里写图片描述
挺丧心病狂的，是不是？

代码1

//数学题
#include <bits/stdc++.h>
#define maxn 1000010
#define ll long long
using namespace std;
ll f[maxn], g[maxn], ans, N, M;
void preprocess()
{
    ll i;
    for(i=1;i<maxn;i++)f[i]=f[i-1]+i, g[i]=g[i-1]+i*i;
}
ll read(ll x=0)
{
    ll c, f=1;
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+c-48;
    return f*x; 
}
void calc()
{
    ll i, t;
    ans=-3*N*M+M*N*(N+1)/2+N*M*(M+1)/2;
    t=min(N,M);
    ans+=(g[t]+f[t])/2+M*f[t]-g[t];
    if(M<N)ans+=(N-M)*(M)*(M+1)/2;
    t=max(0ll,min(N,N-M+1));
    ans+=t*M*(M-1)/2;
    if(M>1)t=max(1ll,N-M+2), ans+=(g[N-t]+f[N-t])/2+(M-1)*f[N-t]-g[N-t];
    printf("%lld\n",ans<<1);
}
int main()
{
    preprocess();
    while(N=read(),M=read(),N)calc();
    return 0;
}

题解2

$Rujia Liu$ 巨佬真的是强
巨佬的思路是分别考虑两个皇后在一行、一列、一条对角线的情况，然后再用组合数求解
一行的情况就是 $mn(n-1)$ ，一列就是 $nm(m-1)$
假设 $n\leq m$ ，对角线的长度是 $1,2,3...,n,n,n,n,...,n,n,n,n-1,n-2,...2,1$
设 $f(x)=x(x-1)$ ，答案就是 $2[f(1)+f(2)+...+f(n)+...+f(n)+f(n-1)+...+f(2)+f(1)]$
所以我对 f <script type="math/tex" id="MathJax-Element-787">f</script>求个前缀和就行了

代码2

//数学题
#include <bits/stdc++.h>
#define maxn 1000010
#define ll long long
using namespace std;
ll f[maxn], N, M;
void preprocess()
{
    ll i;
    for(i=1;i<maxn;i++)f[i]=f[i-1]+i*(i-1);
}
ll read(ll x=0)
{
    ll c, f=1;
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+c-48;
    return f*x; 
}
int main()
{
    preprocess();
    while(N=read(),M=read(),N)
    {
        if(N>M)swap(N,M);
        printf("%lld\n",M*N*(N-1)+N*M*(M-1)+(f[N]+f[N-1]+(M-N)*N*(N-1))*2);
    }
    return 0;
}