Uva 1510 - Neon Sign 解题报告（计数）

    原题的网页我看不到题目了，然后直接下载了PDF，访问依旧很慢。题目传上了CSDN：http://download.csdn.net/detail/kbdwo/7124291

    题目要求求出1000点构成的图中同色边构成的三角形有多少个。枚举是不可能的，1000*1000*1000，复杂度太高。

    我没想到什么好办法，使用bitset保存某个颜色边，枚举两个点，统计他们共同连接的同色边数。1.5秒左右，个人觉得是水过。

    看Statistics里有36ms的……我得去学习下高级的方法了。本人代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <bitset>
using namespace std;

typedef bitset<1001> Bool;
Bool red[1001], blue[1001];

void work()
{
    int n;
    scanf("%d",&n);

    for(int i=0;i<n;i++)
    {
        red[i].reset();
        blue[i].reset();
    }

    for(int i=0;i<n;i++)
    {
        for(int j=i+1; j<n; j++)
        {
            int color;
            scanf("%d",&color);
            if(color)
            {
                red[i].set(j);
                red[j].set(i);
            }
            else
            {
                blue[i].set(j);
                blue[j].set(i);
            }
        }
    }

    long long ans=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++) if(i!=j)
        {
            if(red[i][j])
                ans+=(red[i]&red[j]).count();
            else
                ans+=(blue[i]&blue[j]).count();
        }
    }

    printf("%lld\n", ans/6);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
        work();
}

    果然，有更好的方法。一共有C(n, 3)个三角形，不同色的三角形一定是两条同色的边，对于不同色的三角形的三点，一定有两点连接的是不同色的边。按照这个原理，我们对每个点进行计数，统计红色边和绿色边的积，最终的结果除二，就是最终不同色三角形的数量。代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <bitset>
using namespace std;

int red[1001], blue[1001];

void work()
{
    int n;
    scanf("%d",&n);

    memset(red, 0, sizeof(red));
    memset(blue, 0, sizeof(blue));

    for(int i=0;i<n;i++)
    {
        for(int j=i+1; j<n; j++)
        {
            int color;
            scanf("%d",&color);
            if(color)
                red[i]++,red[j]++;
            else
                blue[i]++,blue[j]++;
        }
    }

    int sum=0;
    for(int i=0;i<n;i++)
        sum+=red[i]*blue[i];

    printf("%d\n", n*(n-1)*(n-2)/6 - sum/2);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
        work();
}

    上面的代码300MS不到搞定。思维啊思维。